demo realtime

Alex-bd · Aug 5, 2019 · a98f945 · a98f945
1 parent 6b1e290
commit a98f945
Show file tree

Hide file tree

Showing 6 changed files with 328 additions and 324 deletions.
diff --git a/config/st_gcn/kinetics-skeleton/camera.yaml → ..._gcn/kinetics-skeleton/demo_realtime.yaml b/config/st_gcn/kinetics-skeleton/camera.yaml → ..._gcn/kinetics-skeleton/demo_realtime.yaml
@@ -1,4 +1,5 @@
 weights: ./models/st_gcn.kinetics.pt
+model_fps: 30
 
 # model
 model: net.st_gcn.Model

diff --git a/main.py b/main.py
@@ -14,7 +14,7 @@
     processors = dict()
     processors['recognition'] = import_class('processor.recognition.REC_Processor')
     processors['demo'] = import_class('processor.demo.Demo')
-    processors['camera'] = import_class('processor.demo_camera.DemoCamera')
+    processors['camera'] = import_class('processor.demo_realtime.DemoRealtime')
     #endregion yapf: enable
 
     # add sub-parser

diff --git a/processor/demo.py b/processor/demo.py
@@ -118,8 +118,7 @@ def get_parser(add_help=False):
             help='Path to save results')
         parser.add_argument('--height',
             default=1080,
-            type=int,
-            help='Path to save results')
+            type=int)
         parser.set_defaults(config='./config/st_gcn/kinetics-skeleton/demo.yaml')
         parser.set_defaults(print_log=False)
         # endregion yapf: enable

diff --git a/processor/demo_camera.py b/processor/demo_camera.py