Add exceptions handling while validating (#975)

* Add exceptions handling while validating * fix lint * add Exception type

Add exceptions handling while validating (#975)
* Add exceptions handling while validating * fix lint * add Exception type
424299cc · Tzu-Chan (Alice) Chuang · Kai Chen · ca7b5216 · 424299cc
Commit 424299cc authored Jul 24, 2019 by Tzu-Chan (Alice) Chuang Committed by Kai Chen Jul 25, 2019
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

mmdet/core/evaluation/eval_hooks.py mmdet/core/evaluation/eval_hooks.py +5 -1

No files found.
--- 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:
+            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