a-PyTorch-Tutorial-to-Object-Detection icon indicating copy to clipboard operation
a-PyTorch-Tutorial-to-Object-Detection copied to clipboard

eval bug

Open stabling opened this issue 4 years ago • 0 comments

Evaluating: 100%|██████████| 78/78 [28:25<00:00, 21.87s/it] Traceback (most recent call last): File "F:/task/a-PyTorch-Tutorial-to-Object-Detection-master - a/eval.py", line 88, in evaluate(test_loader, model) File "F:/task/a-PyTorch-Tutorial-to-Object-Detection-master - a/eval.py", line 79, in evaluate APs, mAP = calculate_mAP(det_boxes, det_labels, det_scores, true_boxes, true_labels, true_difficulties) File "F:\task\a-PyTorch-Tutorial-to-Object-Detection-master - a\utils.py", line 202, in calculate_mAP det_class_boxes = det_boxes[det_labels == c] # (n_class_detections, 4) IndexError: The shape of the mask [917585] at index 0 does not match the shape of the indexed tensor [954568, 4] at index 0

stabling avatar Jun 29 '20 15:06 stabling