Skip to content

GitLab

Commit 87f0aeed authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix gap between train_model predict and inference

parent 3ce97f18
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
tools/infer/predict_det.py
View file @ 87f0aeed
...@@ -39,10 +39,7 @@ class TextDetector(object): ...@@ -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],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment