fix

zhreshold · Aug 17, 2017 · 3ac9d2b · 3ac9d2b
1 parent cc37628
commit 3ac9d2b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/demo.py b/demo.py
@@ -36,7 +36,7 @@ def get_detector(net, prefix, epoch, data_shape, mean_pixels, ctx,
     """
     sys.path.append(os.path.join(os.getcwd(), 'symbol'))
     if net is not None:
-        prefix = prefix + "_" + net.strip('_yolo') + '_' + str(data_shape)
+        prefix = prefix + "_" + net.strip('_yolo') + '_' + str(416)
         net = importlib.import_module("symbol_" + net) \
             .get_symbol(len(CLASSES), nms_thresh, force_nms)
     detector = Detector(net, prefix, epoch, \

diff --git a/symbol/symbol_darknet19_yolo.py b/symbol/symbol_darknet19_yolo.py
@@ -12,7 +12,7 @@ def get_symbol(num_classes=20, nms_thresh=0.5, force_nms=False, **kwargs):
     conv5_5 = bone.get_internals()["conv5_5_output"]
     conv6_5 = bone.get_internals()["conv6_5_output"]
     # anchors
-    anchors = [0.4, 0.4, 
+    anchors = [
                1.3221, 1.73145,
                3.19275, 4.00944,
                5.05587, 8.09892,

diff --git a/tools/find_mxnet.py b/tools/find_mxnet.py
@@ -1,5 +1,5 @@
 try:
-    import mxnet as mx
+    import mxnet0 as mx
 except ImportError:
     import os, sys
     curr_path = os.path.abspath(os.path.dirname(__file__))