Commit Graph

610 Commits (3c9200c67161d02ac61c401f8d055e92225167d2)

Author SHA1 Message Date
xiaoting 5c664bf4f5
Merge pull request #3721 from Topdu/dygraph
add rec_nrtr
2021-08-26 15:30:52 +08:00
LDOUBLEV 6b8d6cc3b7 fix log 2021-08-26 02:22:32 +00:00
LDOUBLEV d7cbe52697 adjust the min_subgraph_size to 15 2021-08-26 02:21:22 +00:00
LDOUBLEV 71ed68c9a8 add comparion script 2021-08-25 13:38:20 +00:00
tink2123 22b57d632d fix infer shape 2021-08-25 09:41:16 +00:00
tink2123 7488bc1db5 fix infer shape 2021-08-25 07:34:23 +00:00
Double_V 68e52250f3
Merge pull request #3751 from LDOUBLEV/fix_pgnet
fix predict_e2e.py
2021-08-25 14:55:26 +08:00
andyjpaddle 8a95b3352d add_rec_sar, test=dygraph 2021-08-24 03:49:26 +00:00
LDOUBLEV 3691291ae1 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into fix_trt_res 2021-08-23 09:14:36 +00:00
LDOUBLEV 1bf24676c2 fix rec trt 2021-08-23 09:13:22 +00:00
LDOUBLEV 5f7d4b9fe1 fix trt dynamic shape 2021-08-23 08:52:42 +00:00
topduke 73058cc082
Update program.py 2021-08-23 11:45:10 +08:00
topduke 533b15c820
Update program.py 2021-08-23 11:43:45 +08:00
LDOUBLEV 8587b40c92 fix predict_e2e.py 2021-08-19 12:23:05 +00:00
Topdu 55b76dcaa5 delete blank lines and modify forward_train 2021-08-19 09:31:02 +00:00
LDOUBLEV 30b10d2933 raise error when run on cpu but set use_gpu as True 2021-08-18 07:01:13 +00:00
topduke adc62fcde4
Merge branch 'dygraph' into dygraph 2021-08-17 22:08:54 +08:00
Topdu 2e76e46b3c update nrtr metric 2021-08-17 13:44:44 +00:00
Topdu 1623c17cdc add rec_nrtr 2021-08-17 13:10:53 +00:00
Topdu b6f0a90366 add rec_nrtr 2021-08-17 12:41:01 +00:00
LDOUBLEV 49c4a3a7ed gpu_id is None when args.use_gpu as False 2021-08-17 02:46:18 +00:00
LDOUBLEV 5bb6ae05c4 add rec get_gpu_id 2021-08-17 02:45:26 +00:00
LDOUBLEV d83b1de17e mv get_gpu_id func to utility 2021-08-17 02:43:15 +00:00
LDOUBLEV 96a3f90ced add gpu check 2021-08-17 02:29:41 +00:00
WenmuZhou 55f6a634fa Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into fx_pse 2021-08-09 12:49:16 +08:00
WenmuZhou 915d06324d merge dygraph 2021-08-09 11:11:29 +08:00
littletomatodonkey af47617493 rename bool to str2bool 2021-08-07 15:43:05 +08:00
MissPenguin 5487ab7a77
Merge pull request #3333 from LDOUBLEV/test_v7
add new ci shell
2021-08-04 19:28:43 +08:00
LDOUBLEV dad8397fd3 Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into test_v7 2021-08-04 02:54:45 +00:00
fengshuai03 fdc808ba6a add dynamic_input for pact 2021-08-03 20:43:01 +08:00
WenmuZhou 22d81cebac Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into fx_pse 2021-08-03 18:59:15 +08:00
Double_V da5d7ee397
Merge pull request #3496 from fengxiaoshuai/dynameic_shape_change
change dynameic shape
2021-08-03 16:58:39 +08:00
LDOUBLEV 44826b515a cherry-pick 3505 2021-08-02 12:18:49 +00:00
fengshuai03 ef103a5956 change dynameic shape 2021-08-02 07:24:37 +00:00
WenmuZhou 026f96b657 add export model of pse 2021-07-29 19:33:47 +08:00
WenmuZhou a6a44ea124 add export model of pse 2021-07-29 19:31:30 +08:00
WenmuZhou a739abab57 merge dygraph 2021-07-27 18:14:19 +08:00
WenmuZhou fa790288f1 add psenet 2021-07-27 15:33:05 +08:00
tink2123 59cc4efdc5 add for SEED 2021-07-22 19:58:14 +08:00
LDOUBLEV 09e7211082 fix bugs and adjust warmup times from 10 to 2 to speed up 2021-07-20 06:47:18 +00:00
zhoujun 07026825ea
Merge pull request #3289 from WenmuZhou/fx_pse
fix a bug of train log not save and add filepath check to dataset
2021-07-19 17:11:55 +08:00
zhoujun 4dbbfd6739
Merge pull request #3338 from WenmuZhou/fx_bug
fix bug
2021-07-19 11:13:13 +08:00
WenmuZhou 57556882ff fix bug 2021-07-14 18:19:52 +08:00
zhoujun 180f7d0051
Merge pull request #3325 from WenmuZhou/fx_bug
fix starttime no define error
2021-07-14 18:17:43 +08:00
LDOUBLEV 9d5d552b79 delete save_log_path and set logger 2021-07-14 06:18:29 +00:00
Double_V c261d0dad0
Merge pull request #3268 from tink2123/autolog_rec
add autolog for infer_rec
2021-07-13 17:12:37 +08:00
WenmuZhou 41f73e31bc fix starttime noe define error 2021-07-13 12:23:41 +08:00
zhoujun 2062b5097c
Merge pull request #3273 from LDOUBLEV/distill
Add det distill
2021-07-12 12:21:46 +08:00
Double_V 76bb40fc3f
fix ci 2021-07-09 14:29:39 +08:00
xiaoting fddef3912f
Merge branch 'dygraph' into autolog_rec 2021-07-09 14:14:24 +08:00
Double_V 54ed8d0626
fix ci 2021-07-09 11:38:01 +08:00
Double_V 11151e51d2
Update program.py 2021-07-09 09:27:35 +08:00
Double_V 9de346b234
Update train.py 2021-07-09 09:27:07 +08:00
LDOUBLEV b8a65d4333 fix eval bug 2021-07-08 14:32:44 +00:00
tink2123 7c9a5bd68c fix args 2021-07-08 16:34:05 +08:00
Double_V 4066478ea4
fix precision 2021-07-08 11:56:52 +08:00
WenmuZhou 61e77849f2 fix a bug of train log not save 2021-07-07 18:59:52 +08:00
LDOUBLEV 0742f5c521 fix metric etc.al 2021-07-07 07:54:02 +00:00
LDOUBLEV 185d1e1f92 fix bug 2021-07-07 01:54:03 +00:00
LDOUBLEV 10e3729fd5 minsubgraph_size 3 to 10 2021-07-06 14:49:38 +08:00
littletomatodonkey 51193f8820 fix bug of inference 2021-07-06 14:03:03 +08:00
tink2123 e51e2b2bb9 add autolog for infer_rec 2021-07-06 13:11:27 +08:00
LDOUBLEV 902ee2c483 fix comment and delete shrink memory 2021-07-05 06:04:41 +00:00
LDOUBLEV 4042247c97 fix ci and output 2021-07-05 03:51:14 +00:00
LDOUBLEV 65118f3099 fix trt dynamic shape 2021-07-05 03:33:10 +00:00
MissPenguin 9c813bb332
Merge pull request #3218 from WenmuZhou/copy_paste
add copy paste
2021-07-01 15:45:00 +08:00
WenmuZhou 7d4805468e add get_rotate_crop_image 2021-07-01 14:36:33 +08:00
Double_V 5661c6865c
Merge pull request #3182 from LDOUBLEV/autolog
enhance auto_log
2021-07-01 11:52:56 +08:00
WenmuZhou f1f5a4d255 merge dygraph 2021-06-30 18:04:50 +08:00
Double_V f58ad9dd5b
Merge pull request #3203 from LDOUBLEV/fix_infer_mode
delete benchmark
2021-06-30 17:15:56 +08:00
WenmuZhou ac8c2a8946 merge dygraph 2021-06-30 10:55:05 +08:00
WenmuZhou 88a8be1297 add assert 2021-06-30 10:50:13 +08:00
WenmuZhou fdaf1f1577 Extraction get_rotate_crop_image 2021-06-30 10:45:27 +08:00
Double_V cdd98d085f
Update predict_det.py 2021-06-29 10:01:24 +08:00
LDOUBLEV 602b6c1878 Merge branch 'fix_infer_mode' of https://github.com/LDOUBLEV/PaddleOCR into fix_infer_mode 2021-06-28 21:57:47 +08:00
LDOUBLEV d87be123d2 delete benchmark_util 2021-06-28 21:56:59 +08:00
Double_V be181cb3bd
add new load dygraph func (#3088)
* add new load dygraph func

* update load_pretrain_params

* update load_dygrah_params

* Update save_load.py

* Update train.py

* Update save_load.py

* return {} when path is None

* return {} when path is None
2021-06-28 20:44:06 +08:00
Double_V ade33ef3d1
Merge branch 'dygraph' into fix_infer_mode 2021-06-28 20:35:14 +08:00
Double_V e4d49819e2
Merge branch 'dygraph' into autolog 2021-06-28 20:34:37 +08:00
LDOUBLEV 586775a945 delete benchmark 2021-06-28 20:19:10 +08:00
LDOUBLEV 729ff65f68 Merge branch 'autolog' of https://github.com/LDOUBLEV/PaddleOCR into autolog 2021-06-28 09:11:06 +00:00
LDOUBLEV b37c597d17 fix subprocess error 2021-06-28 09:10:10 +00:00
LDOUBLEV fa2d08f211 add elapse to cls 2021-06-28 13:50:29 +08:00
LDOUBLEV b88349335b delete benchmark 2021-06-28 13:47:25 +08:00
Double_V 465ef3bffd
Merge branch 'dygraph' into bm_dyg 2021-06-28 11:05:24 +08:00
xiaoting bc99998693
Merge pull request #3185 from LDOUBLEV/fix_vis
save visualized image
2021-06-25 10:13:12 +08:00
LDOUBLEV 42853ff0de save visualized image 2021-06-24 12:00:58 +00:00
Double_V f8441cffa4
Update predict_det.py 2021-06-24 17:48:58 +08:00
Double_V 4c52e3857d
Update predict_det.py 2021-06-24 17:48:36 +08:00
LDOUBLEV ea60a03487 enhance auto_log 2021-06-24 07:18:08 +00:00
WenmuZhou bf1a716855 rename structure 2 table 2021-06-23 12:28:32 +08:00
MissPenguin 7bcabe0f36 refine 2021-06-22 12:39:43 +00:00
MissPenguin 6e1cfb0525 fix cpu infer error for table structure 2021-06-22 08:39:03 +00:00
MissPenguin 3f92af453f Merge branch 'dygraph' of https://github.com/MissPenguin/PaddleOCR into dygraph 2021-06-22 08:16:32 +00:00
MissPenguin 97a668747c refine 2021-06-22 04:24:14 +00:00
MissPenguin 053cc43d82 refine 2021-06-22 04:23:27 +00:00
MissPenguin 79c0a0604c
Merge branch 'dygraph' into dygraph 2021-06-22 11:41:41 +08:00
MissPenguin c0492e02c7 refine 2021-06-22 03:32:00 +00:00
MissPenguin de6bc4458b refine 2021-06-21 12:33:19 +00:00
MissPenguin 16c247ac46 refine 2021-06-21 12:20:25 +00:00
xiaoting 8fbc59d233
Merge pull request #3061 from LDOUBLEV/warmup
add warmup
2021-06-18 16:52:31 +08:00
littletomatodonkey 0938c44a0a
add embedding and fix eval when distillation (#3112) 2021-06-17 13:29:49 +08:00
LDOUBLEV 7588fc082a delete benchmark utils 2021-06-17 02:24:00 +00:00
MissPenguin 7c8b2c8d19 add train code for table 2021-06-16 08:47:33 +00:00
LDOUBLEV 13f5ee1e93 revert mp process, test=develop 2021-06-16 10:04:27 +08:00
WenmuZhou 5cce398ebb fix structure pipeline infer bug 2021-06-10 17:12:14 +08:00
LDOUBLEV ab0b1ace8f add params min_subgraph_size 2021-06-10 14:29:56 +08:00
LDOUBLEV 96a53fb4be add params min_subgraph_size 2021-06-10 14:29:06 +08:00
WenmuZhou 037e17fc82 merge dygraph 2021-06-10 14:24:59 +08:00
LDOUBLEV 616ad6a179 fix tps and fix trt 2021-06-09 19:15:57 +08:00
zhoujun 3302a0b1ad
Revert "add table eval and predict script" (#3062) 2021-06-09 19:06:44 +08:00
Double_V 85aeae712a
Merge pull request #3002 from littletomatodonkey/dyg/add_distillation
add distillation
2021-06-09 18:48:40 +08:00
LDOUBLEV 796898e0f8 adjust the min_subgraph_size for trt 2021-06-09 18:22:29 +08:00
LDOUBLEV 7ad663119a fix save_log_path as null 2021-06-09 18:20:36 +08:00
Double_V 7f102639b6
Merge branch 'dygraph' into warmup 2021-06-09 14:54:08 +08:00
LDOUBLEV 1de1f30a97 add warmup 2021-06-09 13:27:13 +08:00
Double_V aa1d40b1d9
Merge branch 'dygraph' into bm_dyg 2021-06-09 13:09:07 +08:00
WenmuZhou 0ee6137d75 del pad 2021-06-08 15:28:40 +08:00
WenmuZhou dec76eb75d add pad for small image in det 2021-06-08 10:52:47 +08:00
Double_V b796c91649
Merge branch 'dygraph' into bm_dyg 2021-06-07 13:50:36 +08:00
WenmuZhou bc0d766425 init commit for paddlestructure 2021-06-05 22:00:17 +08:00
littletomatodonkey 48d8537959 rm load_dyg_pretrain 2021-06-05 06:52:45 +00:00
littletomatodonkey bd1820b784 fix infer rec 2021-06-05 03:58:17 +00:00
littletomatodonkey 115955f722 fix metric 2021-06-04 02:46:45 +00:00
WenmuZhou 794362481e add table eval and predict script 2021-06-03 16:39:31 +08:00
littletomatodonkey 6361a38ff5 fix export model for distillation model 2021-06-03 06:53:24 +00:00
WenmuZhou bb5c6f3b2c slim code 2021-06-02 20:10:59 +08:00
littletomatodonkey ed02b91d26 add distillation function 2021-06-02 08:31:57 +00:00
WenmuZhou f896d5afa4 merge upstream 2021-06-01 19:09:24 +08:00
WenmuZhou 359d81e585 change infer args format from dict to list 2021-05-28 17:31:48 +08:00
LDOUBLEV c8cc0fb4b7 pre-commit 2021-05-26 10:55:53 +00:00
LDOUBLEV 0fd13fa410 process else 2021-05-26 10:54:54 +00:00
LDOUBLEV b8972b3675 add python benchmark for ocr 2021-05-26 10:40:16 +00:00
WenmuZhou eaf38b9b12 combine args in paddleocr and ppocr/infer/utility 2021-05-26 17:34:47 +08:00
LDOUBLEV 0707f74348 delete debug code 2021-05-26 07:36:44 +00:00
LDOUBLEV c946b386fd add trt min max opt shape 2021-05-26 07:18:40 +00:00
LDOUBLEV 9a68a6123a fix hub server error 2021-05-26 07:09:52 +00:00
LDOUBLEV 16d70fb74b add cpu_math_library_num_threads params 2021-05-26 07:08:56 +00:00
Jane-Ding d5651dfa99 fix win train loader 2021-04-29 14:17:48 +00:00
MissPenguin 3b19311dc1
Merge pull request #2678 from WenmuZhou/fix_srn_post_process
add max_text_length to export model
2021-04-28 14:36:51 +08:00
WenmuZhou d36c19ce1a add max_text_length to export model 2021-04-28 13:41:04 +08:00
WenmuZhou b566bcbbaa add max_text_length to export model 2021-04-28 13:36:16 +08:00
littletomatodonkey 099957c907
fix eval laoder on win (#2654) 2021-04-27 10:32:17 +08:00
zhoujun c27022294e
add global_step to .states files (#2566)
Co-authored-by: littletomatodonkey <2120160898@bit.edu.cn>
2021-04-27 10:13:21 +08:00
MissPenguin 64a3580f1b
Modify the resize operation to be consistent with the python version (#2651) 2021-04-27 00:08:53 +08:00
littletomatodonkey 1b959e3eaa add save rec res 2021-04-25 12:53:29 +00:00
littletomatodonkey eafb157a05 add py impl of pr 2574 2021-04-23 08:23:24 +00:00
littletomatodonkey fb5f7f5eb4
fix loader cause it does not need to break (#2562)
* fix loader cause it does not need to be rebuild for each epoch

* fix loader for dynamic sample

* rm thin import
2021-04-20 23:47:40 +08:00
Jethong be5fdae573 fix config format 2021-04-14 15:02:54 +08:00
Jethong 84e25c556b fix dygraph infer post process 2021-04-12 18:59:05 +08:00
zhoujun 60ecbd0ba4
Merge branch 'dygraph' into fix_doc 2021-04-09 18:26:21 -05:00
MissPenguin 3563c0f91f
Merge pull request #2429 from littletomatodonkey/dev/add_thread_pred
add supprt for multi process inference
2021-04-09 23:23:38 +08:00
littletomatodonkey 3de964a4b8 fix print 2021-04-09 12:30:42 +00:00
littletomatodonkey d6eb34f65c fix infer 2021-04-09 12:28:49 +00:00
littletomatodonkey 916cec34c6 unify predict mp and sp 2021-04-09 11:50:01 +00:00
littletomatodonkey 76320bf0db Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into dev/add_thread_pred 2021-04-09 10:29:21 +00:00
littletomatodonkey e19bedf523 fix infer 2021-04-09 10:28:06 +00:00
littletomatodonkey 86f4abf7d5 add supprt for multi process inference 2021-04-09 10:19:34 +00:00
zhoujun 52f45bc357
Merge branch 'dygraph' into dygraph_rc 2021-04-09 18:12:37 +08:00
zhoujun 3e289f6acf
Merge branch 'dygraph' into fix_doc 2021-04-09 18:12:00 +08:00
Jethong a0d1f9236d add different post process 2021-04-09 16:04:02 +08:00
Jethong 0cd48c35e5 Merge branch 'pgnet-v1' of github.com:JetHong/PaddleOCR into pgnet-v1 2021-04-07 12:27:53 +08:00
Jethong 6824db26e7 fix errors and add pretrain_model 2021-04-07 12:26:05 +08:00
zhoujun f5315349e7
Merge branch 'dygraph' into dygraph_rc 2021-03-30 16:27:03 +08:00
zhoujun d354020d2d
Merge branch 'dygraph' into fix_doc 2021-03-30 16:26:12 +08:00
Double_V 7171d78b31
Merge branch 'dygraph' into pgnet-v1 2021-03-30 16:21:41 +08:00
littletomatodonkey 9e7dc2b88e fix logger info for empty log 2021-03-24 06:32:38 +00:00
WenmuZhou 683cb55649 fix typo error 2021-03-22 12:54:17 +08:00
WenmuZhou ef620c2a20 remove unused func 2021-03-22 12:49:10 +08:00
Jethong f698142542 Merge branch 'pgnet-v1' of github.com:JetHong/PaddleOCR into pgnet-v1 2021-03-19 12:06:21 +08:00
Jethong 88964dc92f add visual png 2021-03-19 11:59:35 +08:00
Double_V 93c919e644
Merge branch 'dygraph' into pgnet-v1 2021-03-15 14:18:48 +08:00
Jethong 9711111270 fix error 2021-03-15 13:58:53 +08:00
WenmuZhou a0fd1a48a1 fix srn infer error when use custom max_text_length 2021-03-12 17:15:48 +08:00
Double_V e0908dd523
Merge branch 'dygraph' into fix_mem 2021-03-12 11:07:04 +08:00
LDOUBLEV 4c9d141c6f enable memory optim 2021-03-11 16:44:24 +08:00
WenmuZhou 1271db6d7c fix mem 2021-03-11 15:24:49 +08:00
Jethong 051fe64a0d fix config add e2e_ch.md e2e_res_img623_pg 2021-03-10 14:14:21 +08:00
Jethong bb49e1a53f ADD PGNet_v2 2021-03-08 15:11:57 +08:00
Jethong 1f76f449db Add PGNet 2021-03-08 14:55:49 +08:00
Double_V 08efbb9608
Merge branch 'dygraph' into dygraph 2021-03-02 17:01:49 +08:00
LDOUBLEV 85fd65c8f0 fix typo 2021-02-24 20:20:17 +08:00
tingquangao add9faa822 Fix the log about length of valid_dataloader 2021-02-24 07:42:20 +00:00
LDOUBLEV 87f0aeed70 fix gap between train_model predict and inference 2021-02-24 15:09:37 +08:00
littletomatodonkey 3ce97f18ec
fix predict rec (#2065)
* fix predict rec

* fix lopp
2021-02-22 15:12:32 +08:00
LDOUBLEV cf550d0524 set dilation as False 2021-02-18 19:00:30 +08:00
LDOUBLEV bbd7665c47 fix commit 2021-02-18 18:53:28 +08:00
LDOUBLEV 5fc33c1213 keep dilation as False 2021-02-18 18:45:53 +08:00
LDOUBLEV bbdbf598d4 keep dilation as False 2021-02-18 18:42:47 +08:00
LDOUBLEV 6ec9aa4def Merge branch 'dygraph' of https://github.com/PaddlePaddle/PaddleOCR into trt_cpp 2021-02-18 13:44:13 +08:00
zhoujun 95a1c6176d
Merge pull request #1985 from WenmuZhou/fix_doc
fix rare export error and support rare infer
2021-02-08 14:23:35 +08:00
WenmuZhou 4eb351866c predict_rec support rare 2021-02-08 14:05:48 +08:00
tink2123 b658e10b35 fix srn for eval 2021-02-08 05:42:30 +00:00
xiaoting 895d44bc39
Merge pull request #1971 from tink2123/fix_eval_for_srn
polish code for srn eval
2021-02-07 15:32:56 +08:00
tink2123 ff6b193a61 polish code for srn eval 2021-02-07 07:31:24 +00:00
WenmuZhou 3b1703e3eb add dataset len check 2021-02-04 12:01:56 +08:00
WenmuZhou b9d3e8eeec add dataset len check 2021-02-04 11:33:48 +08:00
LDOUBLEV df1c97af80 fix bug 2021-02-02 21:27:48 +08:00
LDOUBLEV adc83ca9ae opt deploy doc 2021-02-02 21:08:13 +08:00
littletomatodonkey 077fcabcbc fix doc 2021-02-01 14:03:32 +00:00