two_stream config

lingtengqiu · Feb 25, 2019 · c1680c4 · c1680c4
1 parent 7018731
commit c1680c4
Show file tree

Hide file tree

Showing 2 changed files with 70 additions and 0 deletions.
diff --git a/config/st_gcn.twostream/ntu-xsub/train.yaml b/config/st_gcn.twostream/ntu-xsub/train.yaml
@@ -0,0 +1,35 @@
+work_dir: ./work_dir/recognition/ntu-xsub/ST_GCN_TWO_STREAM
+
+# feeder
+feeder: feeder.feeder.Feeder
+train_feeder_args:
+  data_path: ./data/NTU-RGB-D/xsub/train_data.npy
+  label_path: ./data/NTU-RGB-D/xsub/train_label.pkl
+test_feeder_args:
+  data_path: ./data/NTU-RGB-D/xsub/val_data.npy
+  label_path: ./data/NTU-RGB-D/xsub/val_label.pkl
+
+# model
+model: net.st_gcn_twostream.Model
+model_args:
+  in_channels: 3
+  num_class: 60
+  dropout: 0.5
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'ntu-rgb+d'
+    strategy: 'spatial'
+
+#optim
+weight_decay: 0.0001
+base_lr: 0.1
+step: [10, 50]
+
+# training
+device: [0,1,2,3]
+batch_size: 32
+test_batch_size: 32
+num_epoch: 80
+
+
+
diff --git a/config/st_gcn.twostream/ntu-xview/train.yaml b/config/st_gcn.twostream/ntu-xview/train.yaml
@@ -0,0 +1,35 @@
+work_dir: ./work_dir/recognition/ntu-xview/ST_GCN_TWO_STREAM
+
+# feeder
+feeder: feeder.feeder.Feeder
+train_feeder_args:
+  data_path: ./data/NTU-RGB-D/xview/train_data.npy
+  label_path: ./data/NTU-RGB-D/xview/train_label.pkl
+test_feeder_args:
+  data_path: ./data/NTU-RGB-D/xview/val_data.npy
+  label_path: ./data/NTU-RGB-D/xview/val_label.pkl
+
+# model
+model: net.st_gcn_twostream.Model
+model_args:
+  in_channels: 3
+  num_class: 60
+  dropout: 0.5
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'ntu-rgb+d'
+    strategy: 'spatial'
+
+#optim
+weight_decay: 0.0001
+base_lr: 0.1
+step: [10, 50]
+
+# training
+device: [0,1,2,3]
+batch_size: 32 
+test_batch_size: 32
+num_epoch: 80
+
+
+