两种方法:
方法一:手动下载yolo模型
方法二:自动下载yolo模型
方法一
0. 核心思路
- 下载最新的YOLO模型 (.pt) 文件,放到自己指定的路径
- 把原有代码中的yolov8改成yolov10,重新运行即可
1. 下载最新模型
下载地址:ultralytics 的 Github repository,页面往下滑,找到下图所示的地方
把下载的.pt文件放到你需要的位置,如:D:\Software\Python\pyCode\yolo_models\yolov10n.pt
在VS Code中打开pyCode文件夹(自己指定的),找到刚刚下载的yolov10n.pt
,在名称上右键,复制相对链接,得到yolo_models\yolov10n.pt
2. 替换原有代码中的v8到v10
将原有运行成功的代码中的
# load a pretrained YOLOv8n model
model = YOLO('../yolo_models/yolov8n.pt')
替换成
# load a pretrained YOLOv8n model
model = YOLO('../yolo_models/yolov10n.pt')
其中YOLO后面括号内是从网上下载的yolov10n的相对路径。
3. 出现AttributeError: Can't get attribute
等类似错误
出现错误:AttributeError: Can’t get attribute ‘SCDown’ on <module ‘ultralytics.nn.modules.block’ from
解决办法:在终端运行pip install --upgrade ultralytics
然后重新测试更新的代码,成功👍
方法二:
先在终端运行pip install --upgrade ultralytics
,然后把原有代码中的yolov8改成yolov10,重新运行即可自动下载yolov10的模型(个人经验:成功下载yolov10的模型需要关闭科学上网工具),Happy learning 😎
其他:yolov10 相关资料
ultralytics YouTube video: link