Commit Graph

986 Commits (main)

Author SHA1 Message Date
zhangyubo0722 c8eb175db5
fix pdx_model_name ()
Co-authored-by: zhangyubo0722 <zangyubo0722@163.com>
2025-05-07 19:33:00 +08:00
zhangyubo0722 948d521bce
uniform export format with pdx () 2025-05-07 00:24:02 +08:00
zhangyubo0722 a80d2c89e5
fix det for hpi config () 2025-04-22 16:22:00 +08:00
zhangyubo0722 5d120f8fe9
fix rec hpi config () 2025-04-20 14:46:30 +08:00
Tingquan Gao b0ce52f729
save the inference model in json format by default () 2025-04-17 20:17:34 +08:00
liuhongen1234567 1ccf688ca2
fix static train in formula () 2025-03-08 00:24:37 +08:00
zhangyubo0722 f95280d52d
fix bs to 1 in trt dy shape config for some formule rec model and tabel rec models () 2025-03-06 11:34:08 +08:00
co63oc 534e66506f
Fix typos in several files ()
* Fix typos in several files

* Fix
2025-03-06 10:34:46 +08:00
co63oc e061055808
Fix typos () 2025-03-05 19:09:08 +08:00
zhuyipin 9e7a1f4cc1
remove usused mem in latex_ocr head ()
* adapt to npu

* add device judgement
2025-03-05 11:19:15 +08:00
co63oc 78ec762aac
Fix typos () 2025-03-04 14:20:13 +08:00
co63oc de12ece0aa
Fix () 2025-03-04 11:04:41 +08:00
co63oc f35dfef921
Fix () 2025-03-04 09:38:37 +08:00
Wang Xin 37f44372b1
unlock albumentations version () 2025-02-25 16:30:08 +08:00
zhangyubo0722 2b7b76310b
fix formula rec models hpi_config ()
1. for formula rec models, the channel of input data is 1;
2. for latex_ocr_rec models, fix min/max size of dynamic shape.
2025-02-25 14:59:49 +08:00
mauryaland dc34f9b45a
use the env variable PADDLE_OCR_BASE_DIR if it exists to download models ()
* use the env variable PADDLE_OCR_BASE_DIR to download models

* use PADDLE_OCR_BASE_DIR env variable to download models
2025-02-15 07:44:23 +08:00
Tingquan Gao 17fff8cca4
fix dy shapes of trt for rec models () 2025-02-11 11:17:23 +08:00
liuhongen1234567 2c0c4beb06
repair bleu score computation () 2025-02-06 16:02:59 +08:00
Thanajade Dechananthachai c685537e64
Add Thai character dictionary for OCR recognition ()
* Add Thai character dictionary for OCR recognition

* Update Thai character dictionary with empty new line at end of file
2025-02-05 16:09:49 +08:00
liuhongen1234567 cf4c0591ba
repair bug in latexocr cpu infer and typo () 2025-01-16 15:56:13 +08:00
Liu Jiaxuan 52bc8f0eab
fix slanext export bug ()
* add slanext models

* refine codes

* refine codes

* refine codes

* fix export SLANeXt

* fix export bugs
2025-01-09 11:49:23 +08:00
zhangyubo0722 bf2b73f0f0
add version control for export and modify hpi config () 2025-01-08 17:29:52 +08:00
Liu Jiaxuan a6b96bbfb1
fix SLANeXt export bug ()
* add slanext models

* refine codes

* refine codes

* refine codes

* fix export SLANeXt
2025-01-07 19:21:34 +08:00
vivienfanghuagood 359ab6cb76
fix latex_ocr inference ()
* add

* update

* add

* add
2025-01-07 11:26:03 +08:00
liuhongen1234567 ed6fe285a8
add ppocrv4_doc dict () 2025-01-06 15:55:59 +08:00
zhangyubo0722 e314510319
import encryption for aistudio & fix sync bn 2025-01-03 15:34:29 +08:00
Sunflower7788 4f7476d7b8
fix_server_v4_det_output () 2025-01-03 11:59:22 +08:00
zhangyubo0722 2f0a29ed3a
modify export with pir () 2024-12-30 17:00:42 +08:00
liuhongen1234567 0d41ffc91d
repair formula bug when export () 2024-12-24 17:44:31 +08:00
liuhongen1234567 d523388ed1
Add pp formulanet ()
* add ppformulanet

* rename loss

* modify doc

* add export code

* modify yaml for global ref
2024-12-23 13:14:33 +08:00
zhangyubo0722 0697d248f8
support export with pir and no pir () 2024-12-19 20:16:26 +08:00
liuhongen1234567 04c989b7fe
repair type bug for ppocrv3 () 2024-12-16 14:19:57 +08:00
Liu Jiaxuan ae67d96f3e
add slanext models ()
* add slanext models

* refine codes

* refine codes

* refine codes
2024-12-13 13:39:19 +08:00
wanghuancoder f49dec92d6
fix shape64 () 2024-12-12 16:02:56 +08:00
liuhongen1234567 78e7184022
add unimernet model ()
* add unimernet model

* add commate and single test

* repair pytest

* delete export and infer

* delete [ file
2024-12-12 14:17:24 +08:00
fangfangzk 2672be5763
fix:calcute the left_center_pt and right_center_pt from min_area_quad () 2024-12-11 21:56:34 +08:00
wanghuancoder 9c01b43301
paddle.shape return int64 tensor () 2024-12-04 14:26:57 +08:00
liuhongen1234567 6d2bc9f573
add d2s_train_image_shape for static train () 2024-12-02 20:03:12 +08:00
liuhongen1234567 0018cbd2b6
support latexocr static train () 2024-11-29 17:44:53 +08:00
liuhongen1234567 8fdc409edf
change support list () 2024-11-29 15:18:54 +08:00
Wang Xin 500381c940
fix benchmark det_r50_vd_pse_v2_0 train error () 2024-11-16 10:44:12 +08:00
zhangyubo0722 1d4e7a80a0
rename train result () 2024-11-13 15:49:52 +08:00
Christian Clauss 9b92a1c661
Remove Python 2 compatibility dependency six ()
* Remove Python 2 compatibility dependency

* Remove Python 2 compatibility dependency six

* Update operators.py

* Remove Python 2 compatibility dependency six
2024-11-12 11:01:20 +08:00
zhangyubo0722 b153f10d97
update hpi config () 2024-11-08 17:38:32 +08:00
Wang Xin 15fb82dfa1
upgrade to numpy 2.0 and remove imgaug ()
* upgrade to numpy 2.0 and remove imgaug

* fix bug

* fix bug

* fix bug

* fix bug

* fix bug

* add license
2024-11-06 12:09:01 +08:00
wangna11BD 661cda1289
fix nan in ppocrv4 for benchmark ()
* fix nan in ppocrv4 for benchmark

* fix config
2024-10-23 11:55:43 +08:00
Wang Xin 7541776021
fix isnan_v2 is not supported in paddle2onnx () 2024-10-22 09:18:50 +08:00
zhangyubo0722 de457325cd
reset latex ocr () 2024-10-18 19:51:32 +08:00
wangna11BD 349a604951
fix nan in dp16 () 2024-10-18 18:00:08 +08:00
wanghuancoder e621d034b5
fix a pir while bug () 2024-10-16 16:24:59 +08:00