diff --git a/mmdet/core/evaluation/eval_hooks.py b/mmdet/core/evaluation/eval_hooks.py index c7099581952cd525eef1b9a6454aadf9d3ed70c2..99bd2d8d6ccd2b57cc6d0c76ec4f535dddf01377 100644 --- a/mmdet/core/evaluation/eval_hooks.py +++ b/mmdet/core/evaluation/eval_hooks.py @@ -145,7 +145,11 @@ class CocoDistEvalmAPHook(DistEvalHook): cocoGt = self.dataset.coco imgIds = cocoGt.getImgIds() for res_type in res_types: - cocoDt = cocoGt.loadRes(result_files[res_type]) + try: + cocoDt = cocoGt.loadRes(result_files[res_type]) + except IndexError: + print('No prediction found.') + break iou_type = res_type cocoEval = COCOeval(cocoGt, cocoDt, iou_type) cocoEval.params.imgIds = imgIds