minor fixes for training, fix float/double initializer in mxnet

zhreshold · Nov 16, 2016 · acce1cf · acce1cf
1 parent 09cf204
commit acce1cf
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/config/config.py b/config/config.py
@@ -12,8 +12,9 @@
     RandCropper(min_scale=.3, min_aspect_ratio=.5, max_aspect_ratio=2., min_overlap=.3),
     RandCropper(min_scale=.3, min_aspect_ratio=.5, max_aspect_ratio=2., min_overlap=.5),
     RandCropper(min_scale=.3, min_aspect_ratio=.5, max_aspect_ratio=2., min_overlap=.7),
-    RandCropper(min_scale=.3, min_aspect_ratio=.5, max_aspect_ratio=2., min_overlap=.9),
-    RandCropper(min_scale=.3, min_aspect_ratio=.5, max_aspect_ratio=2., min_overlap=1.0)]
+    RandPadder(max_scale=2., min_aspect_ratio=.5, max_aspect_ratio=2., min_gt_scale=.05),
+    RandPadder(max_scale=3., min_aspect_ratio=.5, max_aspect_ratio=2., min_gt_scale=.05),
+    RandPadder(max_scale=4., min_aspect_ratio=.5, max_aspect_ratio=2., min_gt_scale=.05),]
 # cfg.TRAIN.RAND_SAMPLERS = []
 cfg.TRAIN.RAND_MIRROR = True
 cfg.TRAIN.INIT_SHUFFLE = True
@@ -25,7 +26,7 @@
 # validation
 cfg.VALID = edict()
 cfg.VALID.RAND_SAMPLERS = []
-cfg.VALID.RAND_MIRROR = True
-cfg.VALID.INIT_SHUFFLE = True
-cfg.VALID.EPOCH_SHUFFLE = True
+cfg.VALID.RAND_MIRROR = False
+cfg.VALID.INIT_SHUFFLE = False
+cfg.VALID.EPOCH_SHUFFLE = False
 cfg.VALID.RAND_SEED = None
diff --git a/dataset/iterator.py b/dataset/iterator.py
@@ -167,7 +167,7 @@ def _data_augmentation(self, data, label):
                     offset_x = 0 - xmin
                     offset_y = 0 - ymin
                     data_bak = data
-                    data = np.full((new_height, new_width, 3), 255)
+                    data = np.full((new_height, new_width, 3), 128.)
                     data[offset_y:offset_y+height, offset_x:offset_x + width, :] = data_bak
                 label = rand_crops[index][1]
 

diff --git a/mxnet b/mxnet
+1 −1		src/operator/multibox_detection-inl.h
+1 −1		src/operator/multibox_target-inl.h