File tree Expand file tree Collapse file tree 1 file changed +21
-17
lines changed
Expand file tree Collapse file tree 1 file changed +21
-17
lines changed Original file line number Diff line number Diff line change 4242# 建议使用清华源安装 https://pypi.tuna.tsinghua.edu.cn/simple
4343pip install rapid- table- det
4444```
45+
46+ #### 参数说明
47+ cpu和gpu的初始化完全一致
48+ ```
49+ table_det = TableDetector(
50+ # 目标检测表格模型
51+ obj_model_path="models/obj_det_paddle(obj_det.onnx)",
52+ # 边角检测表格模型(从复杂环境得到表格多边形框)
53+ edge_model_path="models/edge_det_paddle(edge_det.onnx)",
54+ # 角点方向识别
55+ cls_model_path="models/cls_det_paddle(cls_det.onnx)",
56+ # 文档场景已经由版面识别模型提取设置为False
57+ use_obj_det=True,
58+ # 只有90,180,270大角度旋转且无透视时候设置为False
59+ use_edge_det=True,
60+ # 小角度(-90~90)旋转设置为False
61+ use_cls_det=True,
62+ )
63+ ```
64+
4565### 快速使用
4666``` python {linenos=table}
4767from rapid_table_det.inference import TableDetector
@@ -117,23 +137,7 @@ print(
117137# cv2.imwrite(f"{out_dir}/{file_name}-visualize.jpg", img)
118138
119139```
120- #### 参数说明
121- ```
122- table_det = TableDetector(\
123- # 目标检测表格模型
124- obj_model_path="models/obj_det_paddle(obj_det.onnx)", \
125- # 边角检测表格模型(从复杂环境得到表格多边形框)
126- edge_model_path="models/edge_det_paddle(edge_det.onnx)", \
127- # 角点方向识别
128- cls_model_path="models/cls_det_paddle(cls_det.onnx)", \
129- # 文档场景已经由版面识别模型提取可以不使用
130- use_obj_det=True, \
131- # 只有90,180,270大角度旋转且无透视时候可以不使用
132- use_edge_det=True, \
133- # 小角度(-90~90)旋转可以不使用
134- use_cls_det=True, \
135- )
136- ```
140+
137141
138142## FAQ (Frequently Asked Questions)
139143
You can’t perform that action at this time.
0 commit comments