浏览代码

fix a bug in RandomIoUCrop

yjh0410 1 年之前
父节点
当前提交
d390715fe3
共有 1 个文件被更改,包括 1 次插入1 次删除
  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,