models

Warden7 · Jun 5, 2018 · a6f59f9 · a6f59f9
1 parent 3773feb
commit a6f59f9
Show file tree

Hide file tree

Showing 6 changed files with 3,014 additions and 10 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,7 +6,9 @@ config_v0/*
 backup/*
 .vscode
 model/*
-models/*
+#models/*
+*.pt
+*.caffemodel
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

diff --git a/config/st_gcn/kinetics_skeleton/test.yaml b/config/st_gcn/kinetics_skeleton/test.yaml
@@ -0,0 +1,22 @@
+weights: ./models/kinetics-st_gcn.pt
+
+# feeder
+feeder: feeder.feeder.Feeder
+test_feeder_args:
+  data_path: ./data/Kinetics/kinetics-skeleton/val_data.npy
+  label_path: ./data/Kinetics/kinetics-skeleton/val_label.pkl
+
+# model
+model: net.st_gcn.Model
+model_args:
+  in_channels: 3
+  num_class: 400
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'openpose'
+    strategy: 'spatial'
+
+# test 
+phase: test
+device: 0
+test_batch_size: 64
diff --git a/config/st_gcn/kinetics/train.full.yaml → config/st_gcn/kinetics_skeleton/train.yaml b/config/st_gcn/kinetics/train.full.yaml → config/st_gcn/kinetics_skeleton/train.yaml