Merge pull request #2788 from LDOUBLEV/fix_21
add params to control att steps
This commit is contained in:
commit
a499f048d0
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
postprocess_params["score_mode"] = args.det_db_score_mode
|
||||
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
|
||||
|
|
Loading…
Reference in New Issue