ybelkada HF staff commited on
Commit
a62250e
1 Parent(s): bc628a7

Update coco_utils.py

Browse files
Files changed (1) hide show
  1. 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}