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])
|