KdaiP commited on
Commit
a1b6617
1 Parent(s): 0b00739

yoloV9 is too slow on CPU, move default model to yoloV8n

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = ["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])
 
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])