Skip to content

Commit a75335f

Browse files
committed
chore: update readme
1 parent 8681e91 commit a75335f

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

README.md

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,11 +82,23 @@ pip install rapid-table-det-paddle (默认安装gpu版本,可以自行覆盖
8282
```
8383
```python
8484
from rapid_table_det_paddle.inference import TableDetector
85+
86+
img_path = f"tests/test_files/chip.jpg"
87+
8588
table_det = TableDetector(
8689
obj_model_path="models/obj_det_paddle",
8790
edge_model_path="models/edge_det_paddle",
8891
cls_model_path="models/cls_det_paddle",
92+
use_obj_det=True,
93+
use_edge_det=True,
94+
use_cls_det=True,
8995
)
96+
result, elapse = table_det(img_path)
97+
obj_det_elapse, edge_elapse, rotate_det_elapse = elapse
98+
print(
99+
f"obj_det_elapse:{obj_det_elapse}, edge_elapse={edge_elapse}, rotate_det_elapse={rotate_det_elapse}"
100+
)
101+
# 一张图片中可能有多个表格
90102
# img = img_loader(img_path)
91103
# file_name_with_ext = os.path.basename(img_path)
92104
# file_name, file_ext = os.path.splitext(file_name_with_ext)
@@ -103,6 +115,7 @@ table_det = TableDetector(
103115
# wrapped_img = extract_table_img(extract_img.copy(), lt, rt, rb, lb)
104116
# cv2.imwrite(f"{out_dir}/{file_name}-extract-{i}.jpg", wrapped_img)
105117
# cv2.imwrite(f"{out_dir}/{file_name}-visualize.jpg", img)
118+
106119
```
107120
#### 参数说明
108121
mode: str 模式,onnx包默认使用onnx_tiny,可选 onnx, paddle包唯一使用paddle \

0 commit comments

Comments
 (0)