Spaces:
Runtime error
Runtime error
Update coco_utils.py
Browse files- coco_utils.py +2 -3
coco_utils.py
CHANGED
|
@@ -89,13 +89,12 @@ def get_coco_api_from_dataset(dataset):
|
|
| 89 |
class CocoEvaluator(object):
|
| 90 |
def __init__(self, coco_gt, iou_types):
|
| 91 |
assert isinstance(iou_types, (list, tuple))
|
| 92 |
-
coco_gt = copy.deepcopy(coco_gt)
|
| 93 |
-
self.coco_gt = coco_gt
|
| 94 |
|
| 95 |
self.iou_types = iou_types
|
| 96 |
self.coco_eval = {}
|
| 97 |
for iou_type in iou_types:
|
| 98 |
-
self.coco_eval[iou_type] = COCOeval(coco_gt, iouType=iou_type)
|
| 99 |
|
| 100 |
self.img_ids = []
|
| 101 |
self.eval_imgs = {k: [] for k in iou_types}
|
|
|
|
| 89 |
class CocoEvaluator(object):
|
| 90 |
def __init__(self, coco_gt, iou_types):
|
| 91 |
assert isinstance(iou_types, (list, tuple))
|
| 92 |
+
self.coco_gt = copy.deepcopy(coco_gt)
|
|
|
|
| 93 |
|
| 94 |
self.iou_types = iou_types
|
| 95 |
self.coco_eval = {}
|
| 96 |
for iou_type in iou_types:
|
| 97 |
+
self.coco_eval[iou_type] = COCOeval(self.coco_gt, iouType=iou_type)
|
| 98 |
|
| 99 |
self.img_ids = []
|
| 100 |
self.eval_imgs = {k: [] for k in iou_types}
|