91 Commits

Author SHA1 Message Date
三洋三洋
e1046ba3fa Change to better import dependency 2024-08-07 01:19:26 +08:00
三洋三洋
7438dee7ac Change setting name 2024-07-11 20:33:51 +08:00
三洋三洋
6793142557 Update model config 2024-06-22 22:08:08 +08:00
三洋三洋
cd519d8e99 Support onnx runtime 2024-06-22 22:00:05 +08:00
三洋三洋
d8659cd3a9 Add formula detection service 2024-06-17 21:23:55 +08:00
三洋三洋
a1c2b5b1ef 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
三洋三洋
2920b753a8 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
三洋三洋
dbbec511ef 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
三洋三洋
29e626c984 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
三洋三洋
f509b8c94a Change the model configuration to trocr 2024-05-28 07:50:09 +00:00
三洋三洋
2ac159bfa2 Using paddleocr with onnxruntime
Deleted the code for test time.
2024-05-28 07:50:09 +00:00
三洋三洋
226c1e1f76 Added mixed recognition
change suryaocr to paddleocr
2024-05-28 07:50:08 +00:00
三洋三洋
a24ccd53ae Added ONNX file for PaddleOCR model 2024-05-28 07:50:08 +00:00
三洋三洋
e2bf22dac8 Added code for PaddleOCR inference 2024-05-28 07:50:08 +00:00
三洋三洋
5c9cff2125 Eliminated dependency on paddleocr
Change to trocr
2024-05-28 07:50:08 +00:00
OleehyO
19827f1837 bugfix: ocr_aug.py
Change "lhy_custom" in ink_swap_color to "random"
2024-05-28 07:49:55 +00:00
三洋三洋
0a51bde1c5 bugfix: missing filter_fn and inference/train transform 2024-05-12 07:49:04 +00:00
TonyLee1256
82f3eb67b7 Update mix_inference.py
替换文本OCR模型为paddleocr
2024-05-09 00:23:02 +08:00
TonyLee1256
30fbc6dc2d Update inference.py
替换文本OCR模型为paddleocr
2024-05-09 00:22:01 +08:00
TonyLee1256
b869122dc6 Update inference.py
增加了计时功能
2024-05-09 00:20:32 +08:00
TonyLee1256
eaed8d88ca Update infer_det.py
增加使用gpu进行onnx模型推理的功能
2024-05-09 00:19:39 +08:00
三洋三洋
5d95d2e65c bugfix 2024-05-08 14:34:01 +00:00
三洋三洋
ad84fcfce8 Added Language option in mixed mode 2024-05-07 07:44:24 +00:00
三洋三洋
ff1872d067 bugfix 2024-05-07 07:11:34 +00:00
三洋三洋
ef529f9234 Add train_config.yaml 2024-05-07 07:11:05 +00:00
TonyLee1256
f3148ef32c bugfix inference.py 2024-05-07 13:28:07 +08:00
TonyLee1256
7b2b947c47 bugfix inference.py 2024-05-07 13:19:43 +08:00
TonyLee1256
9e2d4347b1 Update rec_infer_from_crop_imgs.py 2024-04-22 00:08:36 +08:00
TonyLee1256
55823256ec Update infer_det.py 2024-04-22 00:07:41 +08:00
三洋三洋
dc57872bc9 Merge branch 'dev' into pre_release 2024-04-21 13:14:49 +00:00
三洋三洋
5f62c7fbf0 1) 修复了to_katex.py的bug; 2)把Box.py中的转化结果写在logs 2024-04-21 12:09:26 +00:00
三洋三洋
9b7e392c66 merge dev后调整了项目结构 2024-04-21 00:48:24 +08:00
三洋三洋
eac7f455d6 merge dev后删除了resizer 2024-04-21 00:13:21 +08:00
三洋三洋
f84168a00b 1) 实现了文本-公式混排识别; 2) 重构了项目结构 2024-04-21 00:05:14 +08:00
三洋三洋
3746ddd427 update infer_det.py 2024-04-18 00:06:05 +08:00
三洋三洋
d5eca45fcc 为了支持mixed inference, 重构了目录 2024-04-17 15:24:06 +00:00
三洋三洋
5a9138026f 修复了merge pre_release分支后导致参数名不一致的bug 2024-04-17 14:47:58 +00:00
三洋三洋
891a9c310a Merge branch 'pre_release' into dev 2024-04-17 10:32:22 +00:00
三洋三洋
7a8491b595 checkpoint 2024-04-17 10:20:15 +00:00
三洋三洋
b4b9e8cfc4 前端更新, inference.py更新
1) 前端支持剪贴板粘贴图片.
2) 前端支持模型配置.
3) 修改了inference.py的接口.
4) 删除了不必要的文件
2024-04-17 09:36:40 +00:00
三洋三洋
7d237d820c work in progress 2024-04-12 03:20:04 +00:00
TonyLee1256
936744ea13 新增公式检测模块 2024-04-11 16:44:19 +00:00
三洋三洋
574dcc2842 修改了transforms.py中inference_transform的bug: 在训练的eval阶段没有把png图片转化为np.ndarray 2024-04-11 07:04:58 +00:00
三洋三洋
5c58b88c96 优化了transform.py中的trim_white_border 2024-04-10 16:09:13 +00:00
三洋三洋
aaee57acd2 增加了数据增强的概率 2024-04-09 13:50:35 +00:00
三洋三洋
7e163928c7 inference.py支持katex语法 2024-04-06 12:06:08 +00:00
三洋三洋
8fdaef43f9 update README.md 2024-04-06 11:57:50 +00:00
三洋三洋
35bc4e71a1 inference.py支持katex 2024-04-06 11:38:59 +00:00
三洋三洋
6179cc3226 web demo支持katex, 不再需要本地安装xelatex渲染器 2024-04-06 07:28:46 +00:00
三洋三洋
8d1e719455 web demo加入了katex支持, 不再需要本地安装xelatex渲染器 2024-04-06 07:18:40 +00:00