rename stgcn backbone

kamiQiu · Oct 9, 2019 · dbb00d8 · dbb00d8
1 parent 551e3b4
commit dbb00d8
Show file tree

Hide file tree

Showing 12 changed files with 15 additions and 14 deletions.
diff --git a/configs/recognition/st_gcn/dataset_example/test.yaml b/configs/recognition/st_gcn/dataset_example/test.yaml
@@ -15,7 +15,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: "models.backbones.ST_GCN_AAAI18"
+    type: "models.backbones.ST_GCN_18"
     in_channels: 3
     num_class: 3
     edge_importance_weighting: True

diff --git a/configs/recognition/st_gcn/kinetics-skeleton-from-openpose/test.yaml b/configs/recognition/st_gcn/kinetics-skeleton-from-openpose/test.yaml
@@ -16,7 +16,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: "models.backbones.ST_GCN_AAAI18"
+    type: "models.backbones.ST_GCN_18"
     in_channels: 3
     num_class: 400
     edge_importance_weighting: True

diff --git a/configs/recognition/st_gcn/kinetics-skeleton-from-openpose/train.yaml b/configs/recognition/st_gcn/kinetics-skeleton-from-openpose/train.yaml
@@ -19,7 +19,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: 'models.backbones.ST_GCN_AAAI18'
+    type: 'models.backbones.ST_GCN_18'
     in_channels: 3
     num_class: 400
     edge_importance_weighting: True

diff --git a/configs/recognition/st_gcn_aaai18/kinetics-skeleton/test.yaml b/configs/recognition/st_gcn_aaai18/kinetics-skeleton/test.yaml
@@ -15,7 +15,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: "models.backbones.ST_GCN_AAAI18"
+    type: "models.backbones.ST_GCN_18"
     in_channels: 3
     num_class: 400
     edge_importance_weighting: True

diff --git a/configs/recognition/st_gcn_aaai18/kinetics-skeleton/train.yaml b/configs/recognition/st_gcn_aaai18/kinetics-skeleton/train.yaml
@@ -5,7 +5,7 @@ argparse_cfg:
   work_dir:
     bind_to: processor_cfg.work_dir
     help: the dir to save logs and models
-    default: ./work_dir/recognition/st_gcn_aaai18/kinetics-skeleton
+    default: ./work_dir/recognition/ST_GCN_18/kinetics-skeleton
   batch_size:
     bind_to: processor_cfg.batch_size
   resume_from:
@@ -19,7 +19,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: 'models.backbones.ST_GCN_AAAI18'
+    type: 'models.backbones.ST_GCN_18'
     in_channels: 3
     num_class: 400
     edge_importance_weighting: True

diff --git a/configs/recognition/st_gcn_aaai18/ntu-rgbd-xsub/test.yaml b/configs/recognition/st_gcn_aaai18/ntu-rgbd-xsub/test.yaml
@@ -15,7 +15,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: "models.backbones.ST_GCN_AAAI18"
+    type: "models.backbones.ST_GCN_18"
     in_channels: 3
     num_class: 60
     dropout: 0.5

diff --git a/configs/recognition/st_gcn_aaai18/ntu-rgbd-xsub/train.yaml b/configs/recognition/st_gcn_aaai18/ntu-rgbd-xsub/train.yaml
@@ -5,7 +5,7 @@ argparse_cfg:
   work_dir:
     bind_to: processor_cfg.work_dir
     help: the dir to save logs and models
-    default: ./work_dir/recognition/st_gcn_aaai18/ntu-rgbd-xsub
+    default: ./work_dir/recognition/ST_GCN_18/ntu-rgbd-xsub
   batch_size:
     bind_to: processor_cfg.batch_size
   resume_from:
@@ -19,7 +19,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: 'models.backbones.ST_GCN_AAAI18'
+    type: 'models.backbones.ST_GCN_18'
     in_channels: 3
     num_class: 60
     dropout: 0.5

diff --git a/configs/recognition/st_gcn_aaai18/ntu-rgbd-xview/test.yaml b/configs/recognition/st_gcn_aaai18/ntu-rgbd-xview/test.yaml
@@ -15,7 +15,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: "models.backbones.ST_GCN_AAAI18"
+    type: "models.backbones.ST_GCN_18"
     in_channels: 3
     num_class: 60
     dropout: 0.5

diff --git a/configs/recognition/st_gcn_aaai18/ntu-rgbd-xview/train.yaml b/configs/recognition/st_gcn_aaai18/ntu-rgbd-xview/train.yaml
@@ -5,7 +5,7 @@ argparse_cfg:
   work_dir:
     bind_to: processor_cfg.work_dir
     help: the dir to save logs and models
-    default: ./work_dir/recognition/st_gcn_aaai18/ntu-rgbd-xview
+    default: ./work_dir/recognition/ST_GCN_18/ntu-rgbd-xview
   batch_size:
     bind_to: processor_cfg.batch_size
   resume_from:
@@ -19,7 +19,7 @@ processor_cfg:
 
   # model setting
   model_cfg:
-    type: 'models.backbones.ST_GCN_AAAI18'
+    type: 'models.backbones.ST_GCN_18'
     in_channels: 3
     num_class: 60
     dropout: 0.5

diff --git a/configs/utils/build_kinetics_skeleton_from_hrnet.yaml b/configs/utils/build_kinetics_skeleton_from_hrnet.yaml
@@ -27,6 +27,7 @@ processor_cfg:
         - 0.224
         - 0.225
       post_process: true
+  tracker_cfg: null
 
 argparse_cfg:
   gpus:

diff --git a/mmskeleton/models/backbones/__init__.py b/mmskeleton/models/backbones/__init__.py
@@ -1,2 +1,2 @@
-from .st_gcn_aaai18 import ST_GCN_AAAI18
+from .st_gcn_aaai18 import ST_GCN_18
 from .hrnet import HRNet
diff --git a/mmskeleton/models/backbones/st_gcn_aaai18.py b/mmskeleton/models/backbones/st_gcn_aaai18.py
@@ -6,7 +6,7 @@
 from mmskeleton.ops.st_gcn import ConvTemporalGraphical, Graph
 
 
-class ST_GCN_AAAI18(nn.Module):
+class ST_GCN_18(nn.Module):
     r"""Spatial temporal graph convolutional networks.
 
     Args:
-Original file line number
+Diff line change
@@ Expand Up / @@ -27,6 +27,7 @@ processor_cfg: @@
             - 0.224
             - 0.225
           post_process: true
+      tracker_cfg: null
     argparse_cfg:
       gpus:
@@ Expand Down @@