Skip to content
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.

Sign up for GitHub

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

Jump to bottom

[Bug Fix] Modify TRT dynamic input shape for PP-OCRv2 python demo. #775

Merged
merged 89 commits into from
Dec 2, 2022
Merged

[Bug Fix] Modify TRT dynamic input shape for PP-OCRv2 python demo. #775

Changes from all commits
Commits
Show all changes
89 commits
Select commit Hold shift + click to select a range
1f1bd0c
Imporve OCR Readme
yunyaoXYY Oct 8, 2022
b23825a
Improve OCR Readme
yunyaoXYY Oct 8, 2022
22dfedb
Improve OCR Readme
yunyaoXYY Oct 8, 2022
755a19d
Improve OCR Readme
yunyaoXYY Oct 8, 2022
338ed80
Improve OCR Readme
yunyaoXYY Oct 8, 2022
c4119e7
Add Initialize function to PP-OCR
yunyaoXYY Oct 9, 2022
21da162
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 9, 2022
146d721
Add Initialize function to PP-OCR
yunyaoXYY Oct 9, 2022
3bfdf3c
Merge branch 'develop' into ocr_dev
jiangjiajun Oct 9, 2022
9dfddfd
Add Initialize function to PP-OCR
yunyaoXYY Oct 9, 2022
7d3642c
Merge branch 'ocr_dev' of github.com:yunyaoXYY/FastDeploy into ocr_dev
yunyaoXYY Oct 9, 2022
d12bb9e
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 12, 2022
16e7c93
Make all the model links come from PaddleOCR
yunyaoXYY Oct 12, 2022
468fce5
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 13, 2022
d72d36f
Improve OCR readme
yunyaoXYY Oct 13, 2022
c47c31b
Improve OCR readme
yunyaoXYY Oct 13, 2022
7c99c99
Improve OCR readme
yunyaoXYY Oct 13, 2022
b5f808e
Improve OCR readme
yunyaoXYY Oct 13, 2022
f037275
Add Readme for vision results
yunyaoXYY Oct 17, 2022
7599543
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 17, 2022
9997fed
Add Readme for vision results
yunyaoXYY Oct 17, 2022
6fd8784
Add Readme for vision results
yunyaoXYY Oct 17, 2022
c34823f
Add Readme for vision results
yunyaoXYY Oct 17, 2022
ae11b80
Add Readme for vision results
yunyaoXYY Oct 17, 2022
cef3415
Add Readme for vision results
yunyaoXYY Oct 17, 2022
532f18f
Add Readme for vision results
yunyaoXYY Oct 17, 2022
302743c
Add Readme for vision results
yunyaoXYY Oct 17, 2022
b9968f6
Add Readme for vision results
yunyaoXYY Oct 17, 2022
5be415e
Add Readme for vision results
yunyaoXYY Oct 17, 2022
a0184b9
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 18, 2022
c718cf2
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
efb445f
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
e93472f
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
2bebd5b
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
75f34a5
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
74ff599
Add check for label file in postprocess of Rec model
yunyaoXYY Oct 18, 2022
d04eaf9
Add comments to create API docs
yunyaoXYY Oct 18, 2022
9f3c59c
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 18, 2022
46eb0da
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 18, 2022
7283757
Improve OCR comments
yunyaoXYY Oct 18, 2022
cc7610d
Merge branch 'wendang' of github.com:yunyaoXYY/FastDeploy into ocr_dev
yunyaoXYY Oct 18, 2022
35e2edc
Rename OCR and add comments
yunyaoXYY Oct 18, 2022
950ec46
Make sure previous python example works
yunyaoXYY Oct 19, 2022
4d21891
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 19, 2022
aabc074
Make sure previous python example works
yunyaoXYY Oct 19, 2022
1d2ed18
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 21, 2022
7ad56b5
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Oct 24, 2022
d23af91
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 2, 2022
6efbbf4
Fix Rec model bug
yunyaoXYY Nov 2, 2022
f6b24ca
Fix Rec model bug
yunyaoXYY Nov 2, 2022
db51e74
Fix rec model bug
yunyaoXYY Nov 2, 2022
2946992
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 2, 2022
5005524
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 3, 2022
ad9ca72
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 8, 2022
ce7affe
Add SetTrtMaxBatchSize function for TensorRT
yunyaoXYY Nov 8, 2022
cfb3c82
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 8, 2022
9dc9ac6
Add SetTrtMaxBatchSize Pybind
yunyaoXYY Nov 8, 2022
b4eb322
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 8, 2022
2781e54
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 8, 2022
1737099
Add set_trt_max_batch_size python function
yunyaoXYY Nov 9, 2022
5fee52c
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 9, 2022
a2784b4
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 9, 2022
f8b82b4
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 11, 2022
7e626a3
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 11, 2022
b2daadc
Set TRT dynamic shape in PPOCR examples
yunyaoXYY Nov 11, 2022
ad7d3a3
Set TRT dynamic shape in PPOCR examples
yunyaoXYY Nov 11, 2022
266d5d6
Set TRT dynamic shape in PPOCR examples
yunyaoXYY Nov 11, 2022
daaa667
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 21, 2022
7a447e1
Fix PPOCRv2 python example
yunyaoXYY Nov 21, 2022
f7a41b1
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 21, 2022
876cc0d
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 22, 2022
571a8e7
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 22, 2022
1da1d07
Fix PPOCR dynamic input shape bug
yunyaoXYY Nov 22, 2022
06eedd6
Remove useless code
yunyaoXYY Nov 23, 2022
0ff28bc
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 24, 2022
d57b18f
Fix PPOCR bug
yunyaoXYY Nov 24, 2022
9022ed5
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 29, 2022
e2f9710
Remove useless comments in PaddleSeg example
yunyaoXYY Nov 29, 2022
82e0862
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 29, 2022
c6af423
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 29, 2022
9c6ae43
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 29, 2022
729112e
Fix quantize docs readme
yunyaoXYY Nov 29, 2022
5a2491e
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Nov 30, 2022
05cb9c7
Fix PP-OCRv2 readme
yunyaoXYY Nov 30, 2022
5987c3c
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Dec 1, 2022
5b9ab51
Modify dynamic shape in PP-OCRv2 example
yunyaoXYY Dec 1, 2022
6f43d5b
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Dec 1, 2022
6caa8bf
Modify TRT dynamic shape for PP-OCRv2
yunyaoXYY Dec 2, 2022
61ea51c
Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy …
yunyaoXYY Dec 2, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions examples/vision/ocr/PP-OCRv2/python/infer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -119,15 +119,15 @@ def build_option(args):

cls_option = runtime_option
cls_option.set_trt_input_shape("x", [1, 3, 48, 10], [10, 3, 48, 320],
[64, 3, 48, 1024])
[32, 3, 48, 1024])
# 用户可以把TRT引擎文件保存至本地
# cls_option.set_trt_cache_file(args.cls_model + "/cls_trt_cache.trt")
cls_model = fd.vision.ocr.Classifier(
cls_model_file, cls_params_file, runtime_option=cls_option)

rec_option = runtime_option
rec_option.set_trt_input_shape("x", [1, 3, 32, 10], [10, 3, 32, 320],
[64, 3, 32, 2304])
[32, 3, 32, 2304])
# 用户可以把TRT引擎文件保存至本地
# rec_option.set_trt_cache_file(args.rec_model + "/rec_trt_cache.trt")
rec_model = fd.vision.ocr.Recognizer(
Expand Down