Commit Graph

89 Commits

Author SHA1 Message Date
三洋三洋
36a2680d28 Update model config 2024-06-22 22:08:08 +08:00
三洋三洋
9638c0030d Support onnx runtime 2024-06-22 22:00:05 +08:00
三洋三洋
76eeb18b83 Add formula detection service 2024-06-17 21:23:55 +08:00
三洋三洋
624f9531b4 Update server.py
1. Change the default host address to 0.0.0.0.
2. Convert the output to KaTeX.
2024-06-07 12:26:24 +00:00
三洋三洋
354833aac8 Modify the names of options in the web.py
Formula only       -> Formula recognition
Text formula mixed -> Paragraph recognition

Improved display during mixed inference
2024-06-06 13:06:16 +00:00
三洋三洋
760bd78c10 Refine mix_inference
1. Add the formula number back to the isolated formula and merge multiple tag.
2. remove bold effect from inline formuals
3. change split environment into aligned
2024-06-06 13:06:11 +00:00
三洋三洋
c0e730f697 Bugfix: to_katex.py
1. Added `change_all` function to fix a bug where some LaTeX formulas with the same wrapper were causing issues.
2. Removed some unnecessary formatting commands.

Bugfix: to_katex.py
2024-06-06 08:25:50 +00:00
三洋三洋
89aa396cbb Change the model configuration to trocr 2024-05-28 07:50:09 +00:00
三洋三洋
9b11689f22 Using paddleocr with onnxruntime
Deleted the code for test time.
2024-05-28 07:50:09 +00:00
三洋三洋
85d558f772 Added mixed recognition
change suryaocr to paddleocr
2024-05-28 07:50:08 +00:00
三洋三洋
2af1e067c1 Added ONNX file for PaddleOCR model 2024-05-28 07:50:08 +00:00
三洋三洋
e193fe3798 Added code for PaddleOCR inference 2024-05-28 07:50:08 +00:00
三洋三洋
714fef4def Eliminated dependency on paddleocr
Change to trocr
2024-05-28 07:50:08 +00:00
OleehyO
1b8f6ba0b6 bugfix: ocr_aug.py
Change "lhy_custom" in ink_swap_color to "random"
2024-05-28 07:49:55 +00:00
三洋三洋
a27cf716ee bugfix: missing filter_fn and inference/train transform 2024-05-12 07:49:04 +00:00
TonyLee1256
83da4262fd Update mix_inference.py
替换文本OCR模型为paddleocr
2024-05-09 00:23:02 +08:00
TonyLee1256
bd2aaa3e00 Update inference.py
替换文本OCR模型为paddleocr
2024-05-09 00:22:01 +08:00
TonyLee1256
fe7e4a7af0 Update inference.py
增加了计时功能
2024-05-09 00:20:32 +08:00
TonyLee1256
48043d11e3 Update infer_det.py
增加使用gpu进行onnx模型推理的功能
2024-05-09 00:19:39 +08:00
三洋三洋
e495640690 bugfix 2024-05-08 14:34:01 +00:00
三洋三洋
84fa43321f Added Language option in mixed mode 2024-05-07 07:44:24 +00:00
三洋三洋
85b22ff9c7 bugfix 2024-05-07 07:11:34 +00:00
三洋三洋
42959cd6a5 Add train_config.yaml 2024-05-07 07:11:05 +00:00
TonyLee1256
d2c1e5e10f bugfix inference.py 2024-05-07 13:28:07 +08:00
TonyLee1256
8c85575260 bugfix inference.py 2024-05-07 13:19:43 +08:00
TonyLee1256
1a7939190f Update rec_infer_from_crop_imgs.py 2024-04-22 00:08:36 +08:00
TonyLee1256
0bb11bebfc Update infer_det.py 2024-04-22 00:07:41 +08:00
三洋三洋
188ab88e07 Merge branch 'dev' into pre_release 2024-04-21 13:14:49 +00:00
三洋三洋
5cbbfb38d6 1) 修复了to_katex.py的bug; 2)把Box.py中的转化结果写在logs 2024-04-21 12:09:26 +00:00
三洋三洋
11df230200 merge dev后调整了项目结构 2024-04-21 00:48:24 +08:00
三洋三洋
e6dca76123 merge dev后删除了resizer 2024-04-21 00:13:21 +08:00
三洋三洋
185b2e3db6 1) 实现了文本-公式混排识别; 2) 重构了项目结构 2024-04-21 00:05:14 +08:00
三洋三洋
eab6e4c85d update infer_det.py 2024-04-18 00:06:05 +08:00
三洋三洋
48f778eeda 为了支持mixed inference, 重构了目录 2024-04-17 15:24:06 +00:00
三洋三洋
7883d3c07f 修复了merge pre_release分支后导致参数名不一致的bug 2024-04-17 14:47:58 +00:00
三洋三洋
a064b7dbb0 Merge branch 'pre_release' into dev 2024-04-17 10:32:22 +00:00
三洋三洋
f81a31a8c9 checkpoint 2024-04-17 10:20:15 +00:00
三洋三洋
3cebc2eb2a 前端更新, inference.py更新
1) 前端支持剪贴板粘贴图片.
2) 前端支持模型配置.
3) 修改了inference.py的接口.
4) 删除了不必要的文件
2024-04-17 09:36:40 +00:00
三洋三洋
7d1d8ddd77 work in progress 2024-04-12 03:20:04 +00:00
TonyLee1256
9e8ac666b0 新增公式检测模块 2024-04-11 16:44:19 +00:00
三洋三洋
1538cb73f8 修改了transforms.py中inference_transform的bug: 在训练的eval阶段没有把png图片转化为np.ndarray 2024-04-11 07:04:58 +00:00
三洋三洋
762012be1f 优化了transform.py中的trim_white_border 2024-04-10 16:09:13 +00:00
三洋三洋
1589fb3217 增加了数据增强的概率 2024-04-09 13:50:35 +00:00
三洋三洋
1db514bdbf inference.py支持katex语法 2024-04-06 12:06:08 +00:00
三洋三洋
840be6b843 update README.md 2024-04-06 11:57:50 +00:00
三洋三洋
93fc22adf5 inference.py支持katex 2024-04-06 11:38:59 +00:00
三洋三洋
ecd5481bea web demo支持katex, 不再需要本地安装xelatex渲染器 2024-04-06 07:28:46 +00:00
三洋三洋
b5f7166e58 web demo加入了katex支持, 不再需要本地安装xelatex渲染器 2024-04-06 07:18:40 +00:00
三洋三洋
c9c15d27bd inference_transform bugfix 2024-04-06 05:09:50 +00:00
三洋三洋
70dce92e19 Merge remote-tracking branch 'origin/dev' into dev 2024-04-05 07:52:40 +00:00