rebase

yingmuying · Jun 2, 2018 · 3773feb · 3773feb
1 parent 88e4148
commit 3773feb
Show file tree

Hide file tree

Showing 59 changed files with 1,294 additions and 1,957 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,6 +6,7 @@ config_v0/*
 backup/*
 .vscode
 model/*
+models/*
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

diff --git a/config/baseline/kinetics-skeleton-from-rawdata/test.yaml b/config/baseline/kinetics-skeleton-from-rawdata/test.yaml
diff --git a/config/baseline/kinetics-skeleton-from-rawdata/train.yaml b/config/baseline/kinetics-skeleton-from-rawdata/train.yaml
diff --git a/config/baseline/kinetics-skeleton/test.yaml b/config/baseline/kinetics-skeleton/test.yaml
diff --git a/config/baseline/kinetics-skeleton/train.yaml b/config/baseline/kinetics-skeleton/train.yaml
diff --git a/config/baseline/nturgbd-cross-subject/test.yaml b/config/baseline/nturgbd-cross-subject/test.yaml
diff --git a/config/baseline/nturgbd-cross-subject/train.yaml b/config/baseline/nturgbd-cross-subject/train.yaml
diff --git a/config/baseline/nturgbd-cross-view/test.yaml b/config/baseline/nturgbd-cross-view/test.yaml
diff --git a/config/baseline/nturgbd-cross-view/train.yaml b/config/baseline/nturgbd-cross-view/train.yaml
diff --git a/config/demo/demo.yaml b/config/demo/demo.yaml
@@ -0,0 +1,16 @@
+work_dir: ./work_dir/demo
+weights: ./work_dir/recognition/kinetics_skeleton/ST_GCN//epoch10_model.pt
+
+# model
+model: net.st_gcn.Model
+model_args:
+  in_channels: 3
+  num_class: 400
+  edge_importance_weighting: True
+  graph_args:
+    layout: 'openpose'
+    strategy: 'spatial'
+
+# training
+device: [0]
+
diff --git a/config/st_gcn/kinetics-skeleton-from-rawdata/test.yaml b/config/st_gcn/kinetics-skeleton-from-rawdata/test.yaml
diff --git a/config/st_gcn/kinetics-skeleton-from-rawdata/train.yaml b/config/st_gcn/kinetics-skeleton-from-rawdata/train.yaml
diff --git a/config/st_gcn/kinetics-skeleton/test.yaml b/config/st_gcn/kinetics-skeleton/test.yaml