[Fix] Fix Internet demo (#2717)

open-mmlab · Sep 22, 2023 · 676bbdb · 676bbdb
1 parent 26f3688
commit 676bbdb
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/demo/docs/en/3d_hand_demo.md b/demo/docs/en/3d_hand_demo.md
@@ -20,6 +20,7 @@ python demo/hand3d_internet_demo.py \
     [--device ${GPU_ID or CPU}] \
     [--kpt-thr ${KPT_THR}] \
     [--show-kpt-idx] \
+    [--show-interval] \
     [--radius ${RADIUS}] \
     [--thickness ${THICKNESS}]
 ```

diff --git a/demo/hand3d_internet_demo.py b/demo/hand3d_internet_demo.py
@@ -59,6 +59,8 @@ def parse_args():
         action='store_true',
         default=False,
         help='Whether to show the index of keypoints')
+    parser.add_argument(
+        '--show-interval', type=int, default=0, help='Sleep seconds per frame')
     parser.add_argument(
         '--radius',
         type=int,
@@ -170,7 +172,8 @@ def main():
             f'{os.path.splitext(os.path.basename(args.input))[0]}.json'
 
     # build the model from a config file and a checkpoint file
-    model = init_model(args.config, args.checkpoint, device=args.device)
+    model = init_model(
+        args.config, args.checkpoint, device=args.device.lower())
 
     # init visualizer
     model.cfg.visualizer.radius = args.radius

diff --git a/mmpose/datasets/datasets/__init__.py b/mmpose/datasets/datasets/__init__.py
@@ -6,5 +6,6 @@
 from .face import *  # noqa: F401, F403
 from .fashion import *  # noqa: F401, F403
 from .hand import *  # noqa: F401, F403
+from .hand3d import *  # noqa: F401, F403
 from .wholebody import *  # noqa: F401, F403
 from .wholebody3d import *  # noqa: F401, F403