Merge pull request #2788 from LDOUBLEV/fix_21

add params to control att steps

deploy/hubserving/ocr_det/params.py

@@ -24,11 +24,6 @@ def read_params():
     cfg.use_dilation = False
     cfg.det_db_score_mode = "fast"
 
-    # #EAST parmas
-    # cfg.det_east_score_thresh = 0.8
-    # cfg.det_east_cover_thresh = 0.1
-    # cfg.det_east_nms_thresh = 0.2
-
     cfg.use_pdserving = False
     cfg.use_tensorrt = False
 

tools/infer/predict_det.py

@@ -65,7 +65,9 @@ class TextDetector(object):
             postprocess_params["max_candidates"] = 1000
             postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio
             postprocess_params["use_dilation"] = args.use_dilation
+            if hasattr(args, "det_db_score_mode"):
                 postprocess_params["score_mode"] = args.det_db_score_mode
+
         elif self.det_algorithm == "EAST":
             postprocess_params['name'] = 'EASTPostProcess'
             postprocess_params["score_thresh"] = args.det_east_score_thresh