Fixed an coco evaluate error while training yourself dataset

Author: onmywei

tool/tv_reference/coco_eval.py

@@ -273,6 +273,7 @@ def loadRes(self, resFile):
     elif 'bbox' in anns[0] and not anns[0]['bbox'] == []:
         res.dataset['categories'] = copy.deepcopy(self.dataset['categories'])
         for id, ann in enumerate(anns):
+            ann['bbox'] = ann['bbox'][0]
             bb = ann['bbox']
             x1, x2, y1, y2 = [bb[0], bb[0] + bb[2], bb[1], bb[1] + bb[3]]
             if 'segmentation' not in ann:

train.py

@@ -490,7 +490,7 @@ def evaluate(model, data_loader, cfg, device, logger=None, **kwargs):
         # outputs = outputs.cpu().detach().numpy()
         res = {}
         # for img, target, output in zip(images, targets, outputs):
-        for img, target, (boxes, confs) in zip(images, targets, outputs):
+        for img, target, boxes, confs in zip(images, targets, outputs[0], outputs[1]):
             img_height, img_width = img.shape[:2]
             # boxes = output[...,:4].copy()  # output boxes in yolo format
             boxes = boxes.squeeze(2).cpu().detach().numpy()