From 41c2af492495c92d6e07cd51078291197d31e718 Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Wed, 16 Dec 2020 00:03:50 +0800
Subject: [PATCH] update predict_det

---
 tools/infer/predict_det.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index b2260662..ba0adaee 100755
--- 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']
-- 
GitLab