-
Notifications
You must be signed in to change notification settings - Fork 60
Fix problems of psenet-ctw1500 training and others #664
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
b7cb963
to
09e9aee
Compare
deploy/py_infer/infer.py
Outdated
@@ -12,7 +12,7 @@ def main(): | |||
args = infer_args.get_args() | |||
parallel_pipeline = ParallelPipeline(args) | |||
parallel_pipeline.start_pipeline() | |||
parallel_pipeline.infer_for_images(args.input_images_dir) | |||
parallel_pipeline.infer_for_images(args.input_images_dir, 0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
此处入参宜改为task_id=0,增强可读性。
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已修改,并同步到0.3.1
@@ -72,9 +72,12 @@ train: | |||
- RandomColorAdjust: | |||
brightness: 0.1255 # 32.0 / 255 | |||
saturation: 0.5 | |||
- IaaAugment: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
mindocr\data\transforms\transforms_factory.py中的IaaAugment调用是否也需要同步修改替换?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已修改,并同步到0.3.1
Thank you for your contribution to the MindOCR repo.
Before submitting this PR, please make sure:
Motivation
infer.py
和模型导出工具bug(Write your motivation for proposed changes here.)
Test Plan
(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)
Related Issues and PRs
(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)