feat: 模型更新
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from core.chessboard_detector import ChessboardDetector
|
|
| 6 |
detector = ChessboardDetector(
|
| 7 |
det_model_path="onnx/det/v2.onnx",
|
| 8 |
pose_model_path="onnx/pose/4_v2.onnx",
|
| 9 |
-
full_classifier_model_path="onnx/layout_recognition/
|
| 10 |
)
|
| 11 |
|
| 12 |
# 数据集路径
|
|
@@ -64,6 +64,9 @@ with gr.Blocks(css="""
|
|
| 64 |
步骤:
|
| 65 |
1. 流程分成两步，第一步检测边缘
|
| 66 |
2. 对整个棋盘画面进行棋子分类预测
|
|
|
|
|
|
|
|
|
|
| 67 |
"""
|
| 68 |
)
|
| 69 |
with gr.Row():
|
|
|
|
| 6 |
detector = ChessboardDetector(
|
| 7 |
det_model_path="onnx/det/v2.onnx",
|
| 8 |
pose_model_path="onnx/pose/4_v2.onnx",
|
| 9 |
+
full_classifier_model_path="onnx/layout_recognition/nano_v1.onnx"
|
| 10 |
)
|
| 11 |
|
| 12 |
# 数据集路径
|
|
|
|
| 64 |
步骤:
|
| 65 |
1. 流程分成两步，第一步检测边缘
|
| 66 |
2. 对整个棋盘画面进行棋子分类预测
|
| 67 |
+
|
| 68 |
+
### log
|
| 69 |
+
2025-01-24 模型优化 200M -> 30M
|
| 70 |
"""
|
| 71 |
)
|
| 72 |
with gr.Row():
|