Spaces:
Runtime error
Runtime error
yoloV9 is too slow on CPU, move default model to yoloV8n
Browse files
app.py
CHANGED
|
@@ -183,7 +183,7 @@ def detect_and_track(input_path: str, output_path: str, detect_class: int, model
|
|
| 183 |
if __name__ == "__main__":
|
| 184 |
|
| 185 |
# YoloV8、V9官方模型列表,从左往右由小到大,第一次使用会自动下载
|
| 186 |
-
model_list = ["
|
| 187 |
|
| 188 |
# 获取YoloV8模型可以检测的所有类别,默认调用model_list中第一个模型
|
| 189 |
detect_classes = get_detectable_classes(model_list[0])
|
|
|
|
| 183 |
if __name__ == "__main__":
|
| 184 |
|
| 185 |
# YoloV8、V9官方模型列表,从左往右由小到大,第一次使用会自动下载
|
| 186 |
+
model_list = ["yolov8n.pt", "yolov9c.pt", "yolov9e", "yolov8s.pt", "yolov8m.pt", "yolov8l.pt", "yolov8x.pt"]
|
| 187 |
|
| 188 |
# 获取YoloV8模型可以检测的所有类别,默认调用model_list中第一个模型
|
| 189 |
detect_classes = get_detectable_classes(model_list[0])
|