fix: col slice

ejuarezg · Feb 6, 2018 · 0572464 · 0572464
1 parent 6f5ffad
commit 0572464
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/ssd.py b/model/ssd.py
@@ -208,7 +208,7 @@ def non_maximum_suppression(self, candidates, overlap_threshold):
         """
 
         label = candidates[:,4]
-        boxes = candidates[label<classes]
+        boxes = candidates[label<classes-1]
 
         if len(boxes) == 0:
             return []