fixed multi_scale_inp_size

Author: longcw

demo.py

@@ -14,7 +14,7 @@ def preprocess(fname):
     # return fname
     image = cv2.imread(fname)
     im_data = np.expand_dims(
-        yolo_utils.preprocess_test((image, None, cfg.inp_size))[0], 0)
+        yolo_utils.preprocess_test((image, None, cfg.multi_scale_inp_size), 0)[0], 0)
     return image, im_data
 
 

layers/reorg/_ext/reorg_layer/__init__.py

@@ -3,8 +3,6 @@
 from ._reorg_layer import lib as _lib, ffi as _ffi
 
 __all__ = []
-
-
 def _import_symbols(locals):
     for symbol in dir(_lib):
         fn = getattr(_lib, symbol)
@@ -14,5 +12,4 @@ def _import_symbols(locals):
             locals[symbol] = fn
         __all__.append(symbol)
 
-
 _import_symbols(locals())

layers/roi_pooling/_ext/roi_pooling/__init__.py

@@ -3,8 +3,6 @@
 from ._roi_pooling import lib as _lib, ffi as _ffi
 
 __all__ = []
-
-
 def _import_symbols(locals):
     for symbol in dir(_lib):
         fn = getattr(_lib, symbol)
@@ -14,5 +12,4 @@ def _import_symbols(locals):
             locals[symbol] = fn
         __all__.append(symbol)
 
-
 _import_symbols(locals())

test.py

@@ -15,7 +15,7 @@
 def preprocess(fname):
     # return fname
     image = cv2.imread(fname)
-    im_data = np.expand_dims(yolo_utils.preprocess_test(image, cfg.inp_size), 0)  # noqa
+    im_data = np.expand_dims(yolo_utils.preprocess_test(image, cfg.multi_scale_inp_size), 0)  # noqa
     return image, im_data