update predict_det

41c2af49 · WenmuZhou · 2ad0ca44 · 41c2af49
Commit 41c2af49 authored Dec 16, 2020 by WenmuZhou
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tools/infer/predict_det.py tools/infer/predict_det.py +3 -1

No files found.
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -177,8 +177,10 @@ class TextDetector(object):
            preds['f_score'] = outputs[1]
            preds['f_tco'] = outputs[2]
            preds['f_tvo'] = outputs[3]
-        else:
+        elif self.det_algorithm == 'DB':
            preds['maps'] = outputs[0]
+        else:
+            raise NotImplementedError

        post_result = self.postprocess_op(preds, shape_list)
        dt_boxes = post_result[0]['points']