From 8d010d7de9c2643e715aaf6033ff7fd5c60ebdc2 Mon Sep 17 00:00:00 2001
From: Cao Yuhang <yhcao6@gmail.com>
Date: Tue, 10 Sep 2019 10:31:59 +0800
Subject: [PATCH] Fix bug when using coco_eval evaluate (#1367)
* fix bug when using coco_eval evaluate
* raise TypeError
---
mmdet/core/evaluation/coco_utils.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/mmdet/core/evaluation/coco_utils.py b/mmdet/core/evaluation/coco_utils.py
index 3022ad0..325d396 100644
--- a/mmdet/core/evaluation/coco_utils.py
+++ b/mmdet/core/evaluation/coco_utils.py
@@ -23,7 +23,12 @@ def coco_eval(result_files, result_types, coco, max_dets=(100, 300, 1000)):
return
for res_type in result_types:
- result_file = result_files[res_type]
+ if isinstance(result_files, str):
+ result_file = result_files
+ elif isinstance(result_files, dict):
+ result_file = result_files[res_type]
+ else:
+ assert TypeError('result_files must be a str or dict')
assert result_file.endswith('.json')
coco_dets = coco.loadRes(result_file)
--
GitLab