new pretrained model

yyl-bot · Feb 21, 2019 · ed11e21 · ed11e21
1 parent fb71c93
commit ed11e21
Show file tree

Hide file tree

Showing 6 changed files with 54 additions and 4 deletions.
diff --git a/config/st_gcn/kinetics-skeleton/demo.yaml b/config/st_gcn/kinetics-skeleton/demo.yaml
@@ -1,4 +1,4 @@
-weights: ./models/kinetics-st_gcn.pt
+weights: ./models/st_gcn.kinetics.pt
 
 # model
 model: net.st_gcn.Model

diff --git a/config/st_gcn/kinetics-skeleton/test.yaml b/config/st_gcn/kinetics-skeleton/test.yaml
@@ -1,4 +1,4 @@
-weights: ./models/kinetics-st_gcn.pt
+weights: ./models/st_gcn.kinetics.pt
 
 # feeder
 feeder: feeder.feeder.Feeder

diff --git a/config/st_gcn/kinetics-skeleton/train.yaml b/config/st_gcn/kinetics-skeleton/train.yaml
@@ -30,7 +30,7 @@ test_batch_size: 256
 #optim
 base_lr: 0.1
 step: [20, 30, 40, 50]
-num_epoch: 60
+num_epoch: 50
 
 
 
diff --git a/config/st_gcn/ntu-xsub/test.yaml b/config/st_gcn/ntu-xsub/test.yaml
@@ -0,0 +1,24 @@
+weights: ./models/st_gcn.ntu-xsub.pt
+
+# feeder
+feeder: feeder.feeder.Feeder
+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.Model
+model_args:
+  in_channels: 3
+  num_class: 60
+  dropout: 0.5
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'ntu-rgb+d'
+    strategy: 'spatial'
+
+# test 
+phase: test
+device: 0
+test_batch_size: 64
+
diff --git a/config/st_gcn/ntu-xview/test.yaml b/config/st_gcn/ntu-xview/test.yaml
@@ -0,0 +1,24 @@
+weights: ./models/st_gcn.ntu-xview.pt
+
+# feeder
+feeder: feeder.feeder.Feeder
+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.Model
+model_args:
+  in_channels: 3
+  num_class: 60
+  dropout: 0.5
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'ntu-rgb+d'
+    strategy: 'spatial'
+
+# test 
+phase: test
+device: 0
+test_batch_size: 64
+
diff --git a/resource/reference_model.txt b/resource/reference_model.txt
@@ -1 +1,3 @@
-kinetics-st_gcn.pt
+st_gcn.kinetics.pt
+st_gcn.ntu-xview.pt
+st_gcn.ntu-xsub.pt