fix bugs in config files

configs/recognition/st_gcn/dataset_example/test.yaml

@@ -31,11 +31,11 @@ processor_cfg:
       data_dir: ./data/dataset_example
       num_track: 2
     pipeline:
-      - name: "skeleton_process.normalize_by_resolution"
-      - name: "skeleton_process.mask_by_visibility"
-      - name: "skeleton_process.transpose"
+      - name: "datasets.skeleton.normalize_by_resolution"
+      - name: "datasets.skeleton.mask_by_visibility"
+      - name: "datasets.skeleton.transpose"
         order: [0, 2, 1, 3]
-      - name: "skeleton_process.to_tuple"
+      - name: "datasets.skeleton.to_tuple"
 
   # dataloader setting
   batch_size: 3

configs/recognition/st_gcn/dataset_example/train.yaml

@@ -38,23 +38,24 @@ processor_cfg:
         num_track: 2
         repeat: 20
       pipeline:
-        - name: "skeleton_process.normalize_by_resolution"
-        - name: "skeleton_process.mask_by_visibility"
-        - name: "skeleton_process.transpose"
+        # - name: "skeleton_process.normalize_by_resolution"
+        - name: "datasets.skeleton.normalize_by_resolution"
+        - name: "datasets.skeleton.mask_by_visibility"
+        - name: "datasets.skeleton.transpose"
           order: [0, 2, 1, 3]
-        - name: "skeleton_process.to_tuple"
+        - name: "datasets.skeleton.to_tuple"
 
     - name: "datasets.DataPipeline"
       data_source:
         name: "datasets.SkeletonLoader"
         data_dir: ./data/dataset_example
         num_track: 2
       pipeline:
-        - name: "skeleton_process.normalize_by_resolution"
-        - name: "skeleton_process.mask_by_visibility"
-        - name: "skeleton_process.transpose"
+        - name: "datasets.skeleton.normalize_by_resolution"
+        - name: "datasets.skeleton.mask_by_visibility"
+        - name: "datasets.skeleton.transpose"
           order: [0, 2, 1, 3]
-        - name: "skeleton_process.to_tuple"
+        - name: "datasets.skeleton.to_tuple"
 
   # dataloader setting
   batch_size: 3

configs/recognition/st_gcn/kinetics-skeleton-from-openpose/test.yaml

@@ -32,11 +32,11 @@ processor_cfg:
       data_dir: ./data/Kinetics/skeleton-from-openpose/val
       num_track: 2
     pipeline:
-      - name: "skeleton_process.normalize_by_resolution"
-      - name: "skeleton_process.mask_by_visibility"
-      - name: "skeleton_process.transpose"
+      - name: "datasets.skeleton.normalize_by_resolution"
+      - name: "datasets.skeleton.mask_by_visibility"
+      - name: "datasets.skeleton.transpose"
         order: [0, 2, 1, 3]
-      - name: "skeleton_process.to_tuple"
+      - name: "datasets.skeleton.to_tuple"
 
   # dataloader setting
   batch_size: 256

configs/recognition/st_gcn/kinetics-skeleton-from-openpose/train.yaml

@@ -37,26 +37,26 @@ processor_cfg:
         data_dir: ./data/Kinetics/skeleton-from-openpose/train
         num_track: 2
       pipeline:
-        - name: "skeleton_process.normalize_by_resolution"
-        - name: "skeleton_process.mask_by_visibility"
-        - name: "skeleton_process.temporal_repeat"
+        - name: "datasets.skeleton.normalize_by_resolution"
+        - name: "datasets.skeleton.mask_by_visibility"
+        - name: "datasets.skeleton.temporal_repeat"
           size: 150
-        - name: "skeleton_process.simulate_camera_moving"
-        - name: "skeleton_process.transpose"
+        - name: "datasets.skeleton.simulate_camera_moving"
+        - name: "datasets.skeleton.transpose"
           order: [0, 2, 1, 3]
-        - name: "skeleton_process.to_tuple"
+        - name: "datasets.skeleton.to_tuple"
 
     - name: "datasets.DataPipeline"
       data_source:
         name: "datasets.SkeletonLoader"
         data_dir: ./data/Kinetics/skeleton-from-openpose/val
         num_track: 2
       pipeline:
-        - name: "skeleton_process.normalize_by_resolution"
-        - name: "skeleton_process.mask_by_visibility"
-        - name: "skeleton_process.transpose"
+        - name: "datasets.skeleton.normalize_by_resolution"
+        - name: "datasets.skeleton.mask_by_visibility"
+        - name: "datasets.skeleton.transpose"
           order: [0, 2, 1, 3]
-        - name: "skeleton_process.to_tuple"
+        - name: "datasets.skeleton.to_tuple"
 
 
   # dataloader setting

mmskeleton/datasets/__init__.py

@@ -1,4 +1,3 @@
-from .skeleton_feeder import SkeletonFeeder
 from .coco import COCODataset
 from .data_pipeline import DataPipeline
-from .skeleton_dataset import SkeletonLoader
+from .skeleton import SkeletonLoader

mmskeleton/datasets/skeleton/__init__.py

@@ -1,3 +1,2 @@
 from .loader import SkeletonLoader
-from .cache_loader import SkeletonCacheLoader
 from .skeleton_process import *

mmskeleton/utils/importer.py

@@ -15,7 +15,8 @@ def import_obj(name):
     except ModuleNotFoundError:
         if name[0:11] != 'mmskeleton.':
             return import_obj('mmskeleton.' + name)
-        raise ImportError('Object {} cannot be found.'.format(class_str))
+        raise ModuleNotFoundError('Object {} cannot be found in {}.'.format(
+            class_str, mod_str))
 
 
 def call_obj(name, **kwargs):