Skip to content

Re-organize online inferene, support more models, add demo results, update docs, add test #311

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 2 commits into from
May 25, 2023

Conversation

SamitHuang
Copy link
Collaborator

@SamitHuang SamitHuang commented May 24, 2023

…pdate docs, add test

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

Motivation

  1. Add scripts to support detection inference, rec inference, and system inference individually
  2. Allow inference on a single image, with no limit to directory input, like python predict_system.py --image_dir test.jpg
  3. Align command line APIs and arguments to paddleocr inference for low-coat transfer.
  4. Remove redundant font file. font files are all placed in docs/font/xx.ttf
  5. Previous detection preprocessing only fit the icdar15 dataset (that's input shape is 720x1280), but can not work correctly for input images with arbitrary shapes. This code adopt the new DetResize to fix this problem, allowing setting limit_type as min or max to better resize for inference.
  6. Add some visualization results on EN and CH text images, which can make the doc more vivid.

Test Plan

Add system test for it.

Already test the new pipeline on icdar 15 test set. Results are shown in tools/infer/text/README.md, which is slightly better than previous version with the same det and rec algorithm. The running time is also similar, which is about 94s for inference on the whole icdar15 detection test set.

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)

@HaoyangLee HaoyangLee merged commit 73cb375 into mindspore-lab:main May 25, 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.

4 participants