-
Notifications
You must be signed in to change notification settings - Fork 668
More ocr models #446
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
More ocr models #446
Conversation
Codecov Report
@@ Coverage Diff @@
## dev-v0.5.0 #446 +/- ##
==============================================
- Coverage 59.18% 58.82% -0.36%
==============================================
Files 233 235 +2
Lines 7739 7820 +81
Branches 1186 1196 +10
==============================================
+ Hits 4580 4600 +20
- Misses 2835 2893 +58
- Partials 324 327 +3
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
mmdeploy/codebase/mmocr/models/text_detection/panet_trt_fp16.py
Outdated
Show resolved
Hide resolved
|
mmdeploy/codebase/mmocr/models/text_detection/panet_trt_fp16.py
Outdated
Show resolved
Hide resolved
mmdeploy/codebase/mmocr/models/text_detection/panet_trt_fp16.py
Outdated
Show resolved
Hide resolved
CRNN requires other configs with 1x32x32 specified. I'm actually not sure if we should keep the codes for fp-16 because it only works for one checkpoint. |
Hi, @AllentDan may use dev-v0.5.0 as your base branch on this PR. |
4cf8222
to
eefb093
Compare
The exported pipeline.json for PANet has both {
"type": "Resize",
"img_scale": [
3000,
640
],
"keep_ratio": true,
"size": [
736,
1333
]
} For now, only |
configs/mmocr/text-recognition/text-recognition_tensorrt-fp16_static-32x32.py
Outdated
Show resolved
Hide resolved
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.
LGTM
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.
LGTM
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.
LGTM
* rewrite sync batchnorm * export panet and psenet * resolution * align fp16 for panet * refine codes * enable satrn for trt * refine docs * docstring * docstring * add ut and refine codes * fix ut * resolve comments and move panet-fp16 to doc * remove ut * refine ut * resolve comments * use size instead of img_scale * use size of MultiScaleAug Co-authored-by: dongchunyu.vendor <[email protected]>
* rewrite sync batchnorm * export panet and psenet * resolution * align fp16 for panet * refine codes * enable satrn for trt * refine docs * docstring * docstring * add ut and refine codes * fix ut * resolve comments and move panet-fp16 to doc * remove ut * refine ut * resolve comments * use size instead of img_scale * use size of MultiScaleAug Co-authored-by: dongchunyu.vendor <[email protected]>
Motivation
Add more mmocr models. PANet, PSENet, SATRN.
Checklist