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