Quick fix NMS

Author: LauLauThom

MTM/NMS.py

@@ -49,10 +49,13 @@ def NMS(tableHit, scoreThreshold=0, sortAscending=False, N_object=-1, maxOverlap
         listScores = [1-score for score in listScores] # NMS expect high-score for good predictions
         scoreThreshold = 1-scoreThreshold
 
-    indexes = cv2.dnn.NMSBoxes(listBoxes, listScores, scoreThreshold, maxOverlap, top_k=N_object)
+    #indexes = cv2.dnn.NMSBoxes(listBoxes, listScores, scoreThreshold, maxOverlap, top_k=N_object) # weird result when top_k=N
+    indexes = cv2.dnn.NMSBoxes(listBoxes, listScores, scoreThreshold, maxOverlap)
+    
+    if N_object>0: indexes = indexes[:N_object] # alternative to keep the N best detections (ie after the actual NMS)
 
-    indexes  = [index[0] for index in indexes]
-    outTable = tableHit[tableHit.index.isin(indexes)]
+    indexes  = [index[0] for index in indexes]        # ordered by score
+    outTable = tableHit[tableHit.index.isin(indexes)].sort_values(by=['Score'], ascending=False)
 
     return outTable