Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import gradio as gr
|
|
| 4 |
|
| 5 |
# 从官方仓库加载模型
|
| 6 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True)
|
| 7 |
-
model.conf = 0.
|
| 8 |
model.iou = 0.2 # 设置交并比阈值
|
| 9 |
|
| 10 |
# 创建预测函数
|
|
|
|
| 4 |
|
| 5 |
# 从官方仓库加载模型
|
| 6 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True)
|
| 7 |
+
model.conf = 0.6 # 设置置信度阈值
|
| 8 |
model.iou = 0.2 # 设置交并比阈值
|
| 9 |
|
| 10 |
# 创建预测函数
|