copy bug fix

1996scarlet · Oct 25, 2020 · b2064a0 · b2064a0
1 parent a355c1d
commit b2064a0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/face_detector.py b/face_detector.py
@@ -51,7 +51,6 @@ def find_biggest_box(dets):
     @staticmethod
     def non_maximum_suppression(dets, threshold):
         ''' ##### Author [email protected]
-        merged_non_maximum_suppression
         Greedily select boxes with high confidence and overlap with threshold.
         If the boxes' overlap > threshold, we consider they are the same one.
 
@@ -82,7 +81,7 @@ def non_maximum_suppression(dets, threshold):
         while order.size > 0:
             keep, others = order[0], order[1:]
 
-            yield dets[keep]
+            yield np.copy(dets[keep])
 
             xx1 = maximum(x1[keep], x1[others])
             yy1 = maximum(y1[keep], y1[others])

diff --git a/weights/model_update.py b/weights/model_update.py
@@ -1,3 +1,15 @@
+import numpy as np
+
+order = np.array([3,5,2,4,1,0])
+
+ovr = 
+
+a = np.arange(9)
+inds = np.where(a > 5)[0]
+
+print(inds + 1)
+exit(0)
+
 import mxnet as mx
 from mxnet.contrib import onnx as onnx_mxnet
 
@@ -27,4 +39,4 @@
     #   "op": "Softmax", 
     #   "axis": "1", 
 
-# mmtoir -f mxnet -n M25-symbol.json -w M25-0000.params -d resnet152 --inputShape 3,112,112
+# mmtoir -f mxnet -n M25-symbol.json -w M25-0000.params -d resnet152 --inputShape 3,112,112