demo

Alex-bd · Jun 6, 2018 · d64b382 · d64b382
1 parent 9f66f3c
commit d64b382
Show file tree

Hide file tree

Showing 8 changed files with 3 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -8,7 +8,6 @@ backup/*
 model/*
 *.pt
 *.caffemodel
-resource/media
 cache/
 
 # Byte-compiled / optimized / DLL files

diff --git a/README.md b/README.md
@@ -87,11 +87,10 @@ To visualize how ST-GCN exploit local correlation and local pattern, we compute
 
 Run the demo by this command:
 ```
-python main.py demo --openpose <path to openpose build directory> [--video <path to your video>]
+python main.py demo --openpose <path to openpose build directory> [--video <path to your video> --device <gpu0> <gpu1>]
 ```
 A video as above will be generated.
 
-
 ## Data Preparation
 
 We experimented on two skeleton-based action recognition datasts: **Kinetics-skeleton** and **NTU RGB+D**. The experiments on NTU RGB+D

diff --git a/resource/media/BQkGMJD39c8.mp4 b/resource/media/BQkGMJD39c8.mp4
diff --git a/resource/media/Kop0sDqOn-c.mp4 b/resource/media/Kop0sDqOn-c.mp4
diff --git a/resource/media/skateboarding.mp4 b/resource/media/skateboarding.mp4
diff --git a/resource/media/ta_chi.mp4 b/resource/media/ta_chi.mp4
diff --git a/resource/media/zZ9Zb41KNR4.mp4 b/resource/media/zZ9Zb41KNR4.mp4
diff --git a/tools/utils/openpose.py b/tools/utils/openpose.py
@@ -6,7 +6,8 @@ def json_pack(snippets_dir, video_name, frame_width, frame_height, label='unknow
     p = Path(snippets_dir)
     for path in p.glob(video_name+'*.json'):
         json_path = str(path)
-        frame_id = int(path.stem.split('_')[1])
+        print(path)
+        frame_id = int(path.stem.split('_')[-2])
         frame_data = {'frame_index': frame_id}
         data = json.load(open(json_path))
         skeletons = []
-Original file line number
+Diff line change
@@ Expand Up / @@ -8,7 +8,6 @@ backup/* @@
     model/*
     *.pt
     *.caffemodel
-    resource/media
     cache/
     # Byte-compiled / optimized / DLL files
@@ Expand Down @@