forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2091 from LDOUBLEV/trt_cpp 

fix gap between train_model predict and inference
This commit is contained in:
Double_V 2021-03-01 13:55:29 +08:00 committed by GitHub
parent 6d7be3e03b ac436d750b
commit f1cd0c35e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/infer/predict_det.py
View File

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

2
tools/infer_det.py
View File

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