Skip to content

Commit 9f8d960

Browse files
committed
chore: fix demo img use
1 parent d4c266a commit 9f8d960

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

demo_onnx.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
11
from rapid_table_det.inference import TableDetector
22

3-
img_path = f"tests/test_files/WechatIMG943.jpg"
3+
img_path = f"tests/test_files/chip2.jpg"
44
table_det = TableDetector(
55
obj_model_path="rapid_table_det/models/obj_det.onnx",
66
edge_model_path="rapid_table_det/models/edge_det.onnx",
7-
use_edge_det=False, use_cls_det=False
87
)
98
result, elapse = table_det(img_path)
109
obj_det_elapse, edge_elapse, rotate_det_elapse = elapse

0 commit comments

Comments
 (0)