Skip to content

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

Merged
merged 4 commits into from
Feb 4, 2024

Conversation

horcham
Copy link
Collaborator

@horcham horcham commented Jan 31, 2024

Thank you for your contribution to the MindOCR repo.
Before submitting this PR, please make sure:

Motivation

  • 修复issues 630:issue 630 。db++改为dbpp。
  • 修复psenet在ctw1500训练错误
  • 修复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)

@horcham horcham force-pushed the main branch 2 times, most recently from b7cb963 to 09e9aee Compare February 1, 2024 02:49
@horcham horcham changed the title [WIP] Fix problems of psenet-ctw1500 training and others Fix problems of psenet-ctw1500 training and others Feb 2, 2024
@@ -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)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

此处入参宜改为task_id=0,增强可读性。

Copy link
Collaborator Author

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:
Copy link
Collaborator

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调用是否也需要同步修改替换?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

已修改,并同步到0.3.1

@panshaowu panshaowu merged commit 9f06e51 into mindspore-lab:main Feb 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants