Explorar el Código

fix a bug in RandomIoUCrop

yjh0410 hace 1 año
padre
commit
d390715fe3
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      evaluator/voc_evaluator.py

+ 1 - 1
evaluator/voc_evaluator.py

@@ -271,7 +271,7 @@ class VOCAPIEvaluator():
         for imagename in imagenames:
             R = [obj for obj in recs[imagename] if obj['name'] == classname]
             bbox = np.array([x['bbox'] for x in R])
-            difficult = np.array([x['difficult'] for x in R]).astype(np.bool)
+            difficult = np.array([x['difficult'] for x in R]).astype(np.bool_)
             det = [False] * len(R)
             npos = npos + sum(~difficult)
             class_recs[imagename] = {'bbox': bbox,