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}
|