Skip to content

Commit 668e4a9

Browse files
committed
chore: add model benchmark
1 parent 2a841d4 commit 668e4a9

File tree

1 file changed

+14
-11
lines changed

1 file changed

+14
-11
lines changed

README.md

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
💡✨ 强大且高效的表格检测,支持论文、期刊、杂志、发票、收据、签到单等各种表格。
2222

23-
🚀 支持来源于paddle和yolo的版本,平衡速度和精度下单图 CPU 推理仅需 1 秒,Paddle-GPU(V100) 仅需 0.2 秒。
23+
🚀 支持来源于paddle和yolo的版本,平衡速度和精度下单图 CPU 推理仅需 1.2 秒,onnx-GPU(V100) 最小组合仅需 0.4 秒,使用pt和paddle模型还能更快!(这个有需要后面再更新吧)
2424

2525
🛠️ 支持三个模块自由组合,独立训练调优,提供 ONNX 转换脚本和微调训练方案。
2626

@@ -60,16 +60,19 @@ obj_model_type="yolo_obj_det", \
6060
edge_model_type= "yolo_edge_det", \
6161
cls_model_type= "paddle_cls_det"
6262

63-
| `model_type` | 任务类型 | 训练来源 | 大小 | 单表格耗时 |
64-
|:---------------------|:-------|:-------------------------------------|:-------|:----------------------|
65-
| **yolo_obj_det** | 表格目标检测 | `yolo11-l` | `100m` | `cpu:500ms, gpu:0.2` |
66-
| `paddle_obj_det` | 表格目标检测 | `paddle yoloe-plus-x` | `380m` | `cpu:500ms, gpu:0.2` |
67-
| `paddle_obj_det_s` | 表格目标检测 | `paddle yoloe-plus-x + quantization` | `95m` | `cpu:1000ms, gpu:0.2` |
68-
| **yolo_edge_det** | 语义分割 | `yolo11-l-segment` | `108m` | `cpu:500ms, gpu:0.2` |
69-
| `yolo_edge_det_s` | 语义分割 | `yolo11-s-segment` | `11m` | `cpu:100ms, gpu:0.2` |
70-
| `paddle_edge_det` | 语义分割 | `paddle-dbnet` | `99m` | `cpu:600ms, gpu:0.2` |
71-
| `paddle_edge_det_s` | 语义分割 | `paddle-dbnet + quantization` | `25m` | `cpu:500ms, gpu:0.2` |
72-
| **paddle_cls_det** | 方向分类 | `paddle pplcnet` | `6.5m` | `cpu:70ms, gpu:0.2` |
63+
由于onnx使用gpu加速效果有限,还是建议直接使用yolox或安装paddle来执行模型会快很多(有需要我再补充整体流程)
64+
paddle的s模型由于量化导致反而速度降低和精度降低,但是模型大小减少很多
65+
66+
| `model_type` | 任务类型 | 训练来源 | 大小 | 单表格耗时(v100-16G,cuda12,cudnn9,ubuntu) |
67+
|:---------------------|:-------|:-------------------------------------|:-------|:-------------------------------------|
68+
| **yolo_obj_det** | 表格目标检测 | `yolo11-l` | `100m` | `cpu:570ms, gpu:400ms` |
69+
| `paddle_obj_det` | 表格目标检测 | `paddle yoloe-plus-x` | `380m` | `cpu:1000ms, gpu:300ms` |
70+
| `paddle_obj_det_s` | 表格目标检测 | `paddle yoloe-plus-x + quantization` | `95m` | `cpu:1200ms, gpu:1000ms` |
71+
| **yolo_edge_det** | 语义分割 | `yolo11-l-segment` | `108m` | `cpu:570ms, gpu:200ms` |
72+
| `yolo_edge_det_s` | 语义分割 | `yolo11-s-segment` | `11m` | `cpu:260ms, gpu:200ms` |
73+
| `paddle_edge_det` | 语义分割 | `paddle-dbnet` | `99m` | `cpu:1200ms, gpu:120ms` |
74+
| `paddle_edge_det_s` | 语义分割 | `paddle-dbnet + quantization` | `25m` | `cpu:860ms, gpu:760ms` |
75+
| **paddle_cls_det** | 方向分类 | `paddle pplcnet` | `6.5m` | `cpu:70ms, gpu:60ms` |
7376

7477

7578
执行参数

0 commit comments

Comments
 (0)