fix ssd demo (apache#4890)

iblislin · Feb 7, 2017 · 2f222cf · 2f222cf
1 parent e9da515
commit 2f222cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/example/ssd/demo.py b/example/ssd/demo.py
@@ -91,7 +91,7 @@ def parse_args():
 
     detector = get_detector(args.network, args.prefix, args.epoch,
                             args.data_shape,
-                            mx.nd.array((args.mean_r, args.mean_g, args.mean_b)).reshape((3,1,1)),
+                            (args.mean_r, args.mean_g, args.mean_b),
                             ctx, args.nms_thresh, args.force_nms)
     # run detection
     detector.detect_and_visualize(image_list, args.dir, args.extension,

diff --git a/example/ssd/detect/detector.py b/example/ssd/detect/detector.py
@@ -37,7 +37,7 @@ def __init__(self, symbol, model_prefix, epoch, data_shape, mean_pixels, \
         self.mod.bind(data_shapes=[('data', (batch_size, 3, data_shape, data_shape))])
         self.mod.set_params(args, auxs)
         self.data_shape = data_shape
-        self.mean_pixels = mx.nd.array(mean_pixels).reshape((3,1,1))
+        self.mean_pixels = mean_pixels
 
     def detect(self, det_iter, show_timer=False):
         """