Skip to content

fix data_sink_mode freeze, adjust DBNet augmentation pipeline #178

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 1 commit into from
Apr 13, 2023

Conversation

hadipash
Copy link
Collaborator

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

Motivation

Passing images to the RandomColorAdjust MindSpore's transform as np.ndarray causes training freeze when the data_sink_mode is set to True. This fix converts images to the PIL format first and converts them back to np.ndarray after color adjustment.

@hadipash hadipash requested a review from HaoyangLee April 13, 2023 04:12
adjust DBNet augmentation pipeline
@HaoyangLee HaoyangLee merged commit bff7f26 into mindspore-lab:main Apr 13, 2023
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.

2 participants