u

JesseYang · May 7, 2018 · 4d9a193 · 4d9a193
1 parent f0a72bf
commit 4d9a193
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -9,3 +9,4 @@ coco
 *.jpg
 *.npy
 *.json
+fitness_videos
diff --git a/cfgs/config_mobilenetv2.py b/cfgs/config_mobilenetv2.py
@@ -62,8 +62,8 @@
     _.limb_seq.append([_.limb_from[idx], _.limb_to[idx]])
     _.map_idx.append([2 * idx, 2 * idx + 1])
 
-# _.stride = 8
-_.stride = 4
+_.stride = 8
+# _.stride = 4
 
 _.sigma = 7
 

diff --git a/train.py b/train.py
@@ -12,8 +12,8 @@
 if cfg.backbone == 'vgg19':
     from modules import VGGBlock as Backbone, Stage1Block, StageTBlock
 else:
-    # from modules import Mobilenetv2Block as Backbone, Stage1DepthBlock as Stage1Block, StageTDepthBlock as StageTBlock
-    from modules import Mobilenetv2Block as Backbone, Stage1Block, StageTBlock
+    from modules import Mobilenetv2Block as Backbone, Stage1DepthBlock as Stage1Block, StageTDepthBlock as StageTBlock
+    # from modules import Mobilenetv2Block as Backbone, Stage1Block, StageTBlock
 from reader import Data
 
 def apply_mask(t, mask):
-Original file line number
+Diff line change
@@ Expand Up / @@ -9,3 +9,4 @@ coco @@
     *.jpg
     *.npy
     *.json
+    fitness_videos