This commit is contained in:
LDOUBLEV 2021-02-24 20:38:07 +08:00
parent 2c7c60bec2
commit cf2bd52989
2 changed files with 2 additions and 5 deletions

View File

@ -39,10 +39,7 @@ class TextDetector(object):
self.args = args
self.det_algorithm = args.det_algorithm
pre_process_list = [{
'DetResizeForTest': {
'limit_side_len': args.det_limit_side_len,
'limit_type': args.det_limit_type
}
'DetResizeForTest': None
}, {
'NormalizeImage': {
'std': [0.229, 0.224, 0.225],

View File

@ -97,7 +97,7 @@ def main():
preds = model(images)
post_result = post_process_class(preds, shape_list)
boxes = post_result[0]['points']
# write resule
# write result
dt_boxes_json = []
for box in boxes:
tmp_json = {"transcription": ""}