Commit Graph

856 Commits (main)

Author SHA1 Message Date
zhangyubo0722 4ba32bc91c
support eval pre epoch (#11003) 2023-09-26 18:50:42 +08:00
zhangyubo0722 e49e491417
add svtr large model (#10937)
* add svtr large model

* [WIP]add svtr large model
2023-09-26 14:38:29 +08:00
topduke 8a52c99ad8
[New] add rec CPPD model (#10990)
* fix gris_sample data type bug when use fp16

* fix gris_sample data type bug when use fp16

* fix v4rec batchsize

* fix bug of hang when multi gpus training(sampler)

* add rec algorithm cppd

* delete cppd useless code

* update cppd bug

* add rec algorithm cppd

* update cppd trainedmodel url

* add cppd en doc
2023-09-25 15:43:45 +08:00
xlg-go ebc67db25b
rec_r45_abinet for export model (#10892)
* When exporting the inference model for ABINet, adapt to the 'image_shape' of ABINetRecResizeImg.

* restore h
2023-09-21 14:50:58 +08:00
ToddBear 75d16610f4
Add new recognition method "ParseQ" (#10836)
* Update PP-OCRv4_introduction.md

* Update PP-OCRv4_introduction.md (#10616)

* Update PP-OCRv4_introduction.md

* Update PP-OCRv4_introduction.md

* Update PP-OCRv4_introduction.md

* Update README.md

* Cherrypicking GH-10217 and GH-10216 to PaddlePaddle:Release/2.7 (#10655)

* Don't break overall processing on a bad image

* Add preprocessing common to OCR tasks
Add preprocessing to options

* Update requirements.txt (#10656)

added missing pyyaml library

* [TIPC]update xpu tipc script (#10658)

* fix-typo (#10642)

Co-authored-by: Dennis <dvorst@users.noreply.github.com>
Co-authored-by: shiyutang <34859558+shiyutang@users.noreply.github.com>

* 修改数据增强导致的DSR报错 (#10662) (#10681)

* 修改数据增强导致的DSR报错

* 错误修改回滚

* Update algorithm_overview_en.md (#10670)

Fixed simple spelling errors.

* Implement recoginition method ParseQ

* Document update for new recognition method ParseQ

* add prediction for parseq

* Update rec_vit_parseq.yml

* Update rec_r31_sar.yml

* Update rec_r31_sar.yml

* Update rec_r50_fpn_srn.yml

* Update rec_vit_parseq.py

* Update rec_vit_parseq.yml

* Update rec_parseq_head.py

* Update rec_img_aug.py

* Update rec_vit_parseq.yml

* Update __init__.py

* Update predict_rec.py

* Update paddleocr.py

* Update requirements.txt

* Update utility.py

* Update utility.py

---------

Co-authored-by: xiaoting <31891223+tink2123@users.noreply.github.com>
Co-authored-by: topduke <784990967@qq.com>
Co-authored-by: dyning <dyning.2003@163.com>
Co-authored-by: UserUnknownFactor <63057995+UserUnknownFactor@users.noreply.github.com>
Co-authored-by: itasli <ilyas.tasli@outlook.fr>
Co-authored-by: Kai Song <50285351+USTCKAY@users.noreply.github.com>
Co-authored-by: dvorst <87502756+dvorst@users.noreply.github.com>
Co-authored-by: Dennis <dvorst@users.noreply.github.com>
Co-authored-by: shiyutang <34859558+shiyutang@users.noreply.github.com>
Co-authored-by: Dec20B <1192152456@qq.com>
Co-authored-by: ncoffman <51147417+ncoffman@users.noreply.github.com>
2023-09-07 16:36:47 +08:00
xlg-go 63f5787bfa
ch_PP-OCRv4_rec_distill.yml, fix KeyError: 'NRTRLabelDecode' (#10761) 2023-08-31 14:25:43 +08:00
itasli 66461c3325
Cherrypicking GH-10251 & GH-10181 to PaddleOCR:dygraph (#10710)
* fix undefined save_model_dir when using wandb (#10251)

* fix undefined save_model_dir when using wand

* Update program.py

* Fix fitz camelCase deprecation and .PDF not being recognized as pdf file (#10181)

* Fix fitz camelCase deprecation and .PDF not being recognized as pdf file

* refactor get_image_file_list function
2023-08-30 10:21:29 +08:00
UserUnknownFactor b3912fcf7a
Cherrypicking GH-10217 and GH-10216 to PaddlePaddle:dygraph (#10654)
* Don't break overall processing on a bad image

* Add preprocessing common to OCR tasks
Add preprocessing to options
2023-08-21 16:33:03 +08:00
zhoujun dbf35bb714
onnxruntime support gpu (#10668)
* Update ch_PP-OCRv3_rec.yml

* Update ch_PP-OCRv3_rec_distillation.yml

* Update en_PP-OCRv3_rec.yml

* Update arabic_PP-OCRv3_rec.yml

* Update chinese_cht_PP-OCRv3_rec.yml

* Update cyrillic_PP-OCRv3_rec.yml

* Update devanagari_PP-OCRv3_rec.yml

* Update japan_PP-OCRv3_rec.yml

* Update ka_PP-OCRv3_rec.yml

* Update korean_PP-OCRv3_rec.yml

* Update latin_PP-OCRv3_rec.yml

* Update ta_PP-OCRv3_rec.yml

* Update te_PP-OCRv3_rec.yml

* Update utility.py
2023-08-17 18:26:20 +08:00
RuohengMa 2a4d551547
Fix seed passing issue of build_dataloader (#10614) 2023-08-11 14:11:13 +08:00
shiyutang 4a91a21245
compat_pillow (#10596) 2023-08-10 16:41:35 +08:00
ToddBear b3f9f681d9
CV套件建设专项活动 - 文字识别返回单字识别坐标 (#10515) (#10537)
* modification of return word box

* update_implements

* Update rec_postprocess.py

* Update utility.py
2023-08-10 15:12:01 +08:00
Lin Manhui fec014e92d
Suppress scalar warnings (#10509) 2023-07-31 15:13:11 +08:00
livingbody 49d1a59572
upgrade pillow to 10.0.0 (#10405) 2023-07-17 14:42:30 +08:00
xiaoting b0e3c3f2c7
clear fluid (#10333) 2023-07-10 14:12:40 +08:00
Lin Manhui 2d44a71b20
Toward Devkit Consistency (#10150)
* Accommodate UAPI

* Fix signal handler

* Save model.pdopt

* Change variable name

* Update vdl dir
2023-06-14 14:31:45 +08:00
Zhang Ting 6949448558
improve amp training (#10119) 2023-06-08 15:50:37 +08:00
xiaoting 096fd27121
fix sr_telescope (#10004) 2023-05-23 16:54:52 +08:00
topduke 425166434c
Fix grid_sample data type bug when use fp16 (#9930)
* fix gris_sample data type bug when use fp16

* fix gris_sample data type bug when use fp16

* fix v4rec batchsize
2023-05-15 17:03:53 +08:00
topduke 2a98d40b10
Add v4rec hgnet (#9768)
* v4rec code

* v4rec add nrtrloss

* Add V4rec backbone file

* Add V4Rec config file.

* Fix V4rec reparameters when export_model

* convert lvnetv3

* fix codestyle

* fix infer_rec v4rec

* add v4rec hgnet

* add v4rec hgnet config

* add svtr_hgnet

* fix bugs in infer_rec and hgnet
2023-04-21 12:34:48 +08:00
duanyanhui ce65745755
update npu inference api (#9765) 2023-04-20 10:26:26 +08:00
Zhang Ting d496c8e8d1
fix performance (#9772) 2023-04-20 09:53:31 +08:00
topduke 43abe2fa4b
V4Rec code pr (#9725)
* v4rec code

* v4rec add nrtrloss

* Add V4rec backbone file

* Add V4Rec config file.

* Fix V4rec reparameters when export_model

* convert lvnetv3

* fix codestyle

* fix infer_rec v4rec
2023-04-19 16:04:23 +08:00
duanyanhui 2d6f3a56a4
update npu api (#9688) 2023-04-11 09:56:08 +08:00
xiaoting dd02918bb4
rename PP-OCRv3 algorithm to SVTR_LCNet (#9025)
* rename PP-OCRv3 algorithm to SVTR_LCNet

* rename PP-OCRv3 algorithm to SVTR_LCNet

* update multi-lang config for v3
2023-02-13 11:17:58 +08:00
zhiminzhang0830 30201ef954
add satrn (#8433)
* add satrn

* 修复satrn导出问题

* 规范satrn config文件

* 删除SATRNRecResizeImg

---------

Co-authored-by: zhiminzhang0830 <zhangzhimin04@baidu.com>
2023-02-08 11:25:50 +08:00
Double_V 3629d5a365
support set gpu_id when inference (#8926)
* support min_area_rect crop

* add check_install

* fix requirement.txt

* fix check_install

* add lanms-neo for drrg

* fix

* fix doc

* fix

* support set gpu_id when inference

* fix #8855

* fix #8855
2023-01-30 18:46:41 +08:00
xiaoting ff947d8960
fix onnx for dynamic shape (#8812) 2023-01-10 10:48:42 +08:00
Zhang Ting 4f735dbde1
matmul use fp32 compute_type (#8733) 2022-12-29 13:42:12 +08:00
Double_V 5cac747656
[bug] fix issues about sort_function and DB Head (#8580)
* support min_area_rect crop

* add check_install

* fix requirement.txt

* fix check_install

* add lanms-neo for drrg

* fix

* fix doc

* fix
2022-12-13 16:26:42 +08:00
zhoujun 29a33c547b
fix re inference error (#8473) 2022-12-01 09:52:03 +08:00
Double_V d314008fa7
support min_area_rect crop (#8127) 2022-11-01 11:30:34 +08:00
WenmuZhou 6120094ac2 add Telescope to support alg 2022-10-25 05:08:58 +00:00
dorren f875556541 update can transform method and add copyright info for new file 2022-10-17 17:18:14 +08:00
dorren c57effb84f update can data loading method and tipc configs, revert precommit config 2022-10-17 15:04:42 +08:00
dorren f3f473d3f9 update CAN model 2022-10-15 20:27:05 +08:00
zhoujun ce72023ce1
Merge pull request #7832 from zhiminzhang0830/drrg_branch
add text detection algorithm DRRG
2022-10-13 13:19:13 +08:00
Double_V 9df7730eeb
Merge pull request #7840 from LDOUBLEV/dygraph
add polygon params
2022-10-12 10:14:16 +08:00
zhiminzhang0830 3ee146fa99 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into drrg_branch 2022-10-10 15:56:08 +08:00
MissPenguin a7069086c3
Merge pull request #7835 from qipengh/add_mlu_dbnet
[MLU]adapt mlu device for running dbnet network
2022-10-10 15:35:20 +08:00
LDOUBLEV 3558d57681 fix ci 2022-10-09 14:15:47 +08:00
LDOUBLEV 119196f11d add det_box_type params 2022-10-09 11:28:32 +08:00
LDOUBLEV 9919cb71d1 delete debug 2022-10-09 10:10:05 +08:00
zhiminzhang0830 c25eec882a Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into rfl_branch 2022-10-09 10:07:46 +08:00
LDOUBLEV 81e4c3d821 add polygon params 2022-10-09 09:58:52 +08:00
huangqipeng 7851977157 [MLU]adapt mlu device for running dbnet network 2022-10-08 16:46:41 +08:00
littletomatodonkey eeef62b3c6
fix PREN export and infer (#7833) 2022-10-08 16:37:12 +08:00
zhiminzhang0830 03802c7f93 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into drrg_branch 2022-10-08 12:00:17 +08:00
zhiminzhang0830 1f9400dd73 add drrg 2022-10-08 11:56:45 +08:00
zhiminzhang0830 c459b72565 添加RFL CNT分支infer支持 2022-10-08 11:20:36 +08:00
zhiminzhang0830 3f8602c169 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into rfl_branch 2022-09-29 16:40:10 +08:00
andyjpaddle 6dd8da089f format code 2022-09-29 07:51:13 +00:00
andyjpaddle e61f40ef41 support infer pdf file 2022-09-29 07:31:45 +00:00
zhiminzhang0830 0002349df3 add text recognition algorithm rflearning 2022-09-27 10:54:31 +08:00
WenmuZhou 06194524ca add re predict 2022-09-20 22:13:27 +08:00
Double_V 8fb6fea192
Update utility.py 2022-09-19 17:35:04 +08:00
LDOUBLEV bb54e9b8eb fix 2022-09-19 14:03:15 +08:00
LDOUBLEV a8c6173347 add txt 2022-09-19 13:33:47 +08:00
LDOUBLEV 9881156261 delete shape_info_name 2022-09-19 13:26:18 +08:00
LDOUBLEV 1b07e42912 fix trt 2022-09-16 20:11:18 +08:00
huangjun12 b5268dc3a0 add centripetal text model 2022-09-15 11:08:16 +00:00
qili93 149e118474 [TIPC] add scripts for NPU and XPU, test=develop 2022-09-13 14:11:21 +08:00
Double_V d022b26903
Merge pull request #7420 from wangjingyeye/dyg_db
update pgnet
2022-09-05 18:45:44 +08:00
zhangxinnan 51d771ce19 import random放到脚本 开始 2022-09-04 12:40:45 +08:00
zhangxinnan 2925aeb4e4 去掉旧版本的draw_ocr_box_txt 2022-09-04 12:38:18 +08:00
zhangxinnan 960a0ca607 增加draw_ocr_box_txt2:解决文本框倾斜时文字画在外边的情况 2022-09-04 10:01:16 +08:00
wangjingyeye 929b4f4557 update pgnet 2022-08-30 05:58:39 +00:00
andyjpaddle 96a8e0f88e Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into dygraph 2022-08-23 11:53:53 +00:00
andyjpaddle 29268eaaf0 add args en doc 2022-08-23 11:52:42 +00:00
xiaoting 5a524807f9
fix infer_rec for other input (#7286) 2022-08-23 18:05:03 +08:00
WenmuZhou 2b1885f159 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into tipc 2022-08-22 11:33:51 +00:00
WenmuZhou c3924a959b add amp eval 2022-08-22 11:32:37 +00:00
littletomatodonkey aedeb28a56
fix re bug (#7288) 2022-08-22 16:31:54 +08:00
Evezerest 7e4e87dd6d
cherry pick PRs from community (#7273)
* Merge pull request #6824 from ChenNima/release/2.5-kie-save-res

[kie]add write_kie_result to kie infer tool

* Merge pull request #6677 from TonyJiangWJ/release/2.5

修复内存泄露问题

* Update native.cpp (#6650)

fix issue 6640

* Merge pull request #6625 from ynjang/ynjang

update sorted_boxes

* fix DeprecationWarning, (#6604)

DeprecationWarning: The binary mode of fromstring is deprecated, as it behaves surprisingly on unicode inputs. Use frombuffer instead

* Merge pull request #6585 from maxbachmann/release/2.5

replace GPL licensed components

* Merge pull request #6575 from Eling486/release/2.5

update win doc

* Merge pull request #6477 from MikoyChinese/fix-copy-paste

Fix copy_paste no texts augment.

* Merge pull request #6361 from mohamadmansourX/patch-9

Update README_en.md

Co-authored-by: Double_V <liuvv0203@163.com>
Co-authored-by: shawn <1021362695@qq.com>
Co-authored-by: paopjian <672034519@qq.com>
2022-08-21 18:03:57 +08:00
littletomatodonkey 8c7c45420f
polish kie doc and code (#7255)
* add fapiao kie

* fix readme

* fix fanli

* add readme

* add how to do kie en

* add algo kie

* add algo overview en

* rename vqa to kie

* fix read gif
2022-08-21 10:55:49 +08:00
zhoujun aec5dbf8f7
update model size (#7269)
* update model size

* update layout dict in whl
2022-08-20 16:10:45 +08:00
zhoujun 94710ae319
Merge pull request #7266 from WenmuZhou/tipc
add PP-OCRv2 det amp custom_black_list
2022-08-19 17:02:24 +08:00
WenmuZhou 207969039b support config black_list in yml 2022-08-19 08:29:19 +00:00
xiaoting c731d54e18
fix sr typo (#7262)
* fix sr typo

* rm eval visual
2022-08-19 14:49:35 +08:00
WenmuZhou 7c5fbd46dc Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into tipc 2022-08-19 13:26:49 +08:00
WenmuZhou 1b93e92ccb fix amp bug 2022-08-19 13:26:02 +08:00
zhoujun a485740fa7
Merge pull request #6842 from smilelite/robustscanner_branch
添加robustscanner(第三次)
2022-08-18 12:11:21 +08:00
littletomatodonkey f68571b2f6
rm dyg shape for trt (#7221) 2022-08-17 23:24:47 +08:00
smilelite 40c45e2ccc
Merge branch 'dygraph' into robustscanner_branch 2022-08-17 21:49:11 +08:00
andyjpaddle 13f7b1c8b3
Merge pull request #7231 from tink2123/revert_imgh
rever rec input h
2022-08-17 18:59:29 +08:00
tink2123 c49d000389 rever rec input h 2022-08-17 17:13:06 +08:00
zhoujun 9611770e1e
Merge pull request #7137 from WenmuZhou/table_pr
add SLANet
2022-08-17 15:07:50 +08:00
WenmuZhou 9aa7593c6f rm print model 2022-08-16 14:23:37 +00:00
andyjpaddle 6e89ec8d09 fix sar export 2022-08-16 09:34:48 +00:00
WenmuZhou 22c13af65b Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into table_pr 2022-08-16 03:35:46 +00:00
smilelite a8ba8963e0 merge cpnflict 0815 2022-08-15 23:03:21 +08:00
andyjpaddle e5e787018e fix vl 2022-08-15 09:05:32 +00:00
WenmuZhou 7efee821bc Support variable length input 2022-08-15 09:00:35 +00:00
WenmuZhou 444080a0cd update vis 2022-08-15 08:51:07 +00:00
andyjpaddle 89f8b1d49f
Merge pull request #7196 from andyjpaddle/fix_sar
fix sar infer shape
2022-08-15 14:07:43 +08:00
andyjpaddle b7ca8c8070 fix sar infer shape 2022-08-15 03:40:29 +00:00
littletomatodonkey 78871cfefb
kie doc (#7154)
* kie doc

* fix xlm model export

* fix doc

* add wildreceipt dataset

* fix doc

* fix doc
2022-08-15 11:39:11 +08:00
WenmuZhou ba284a2b74 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into table_pr 2022-08-14 09:07:19 +00:00
WenmuZhou ce321153e7 rm unused code 2022-08-14 09:01:49 +00:00
xiaoting 7054013004
Submit SR model (#6933)
* add sr model

* update for eval

* submit sr

* polish code

* polish code

* polish code

* update sr model

* update doc

* update doc

* update doc

* fix typo

* format code

* update metric

* fix export
2022-08-12 10:49:54 +08:00