YOLOv5 AssertionError: “XXX” acceptable suffix is [‘.pt‘]

使用终端训练YOLOv5模型报错,原命令为:

"python train.py --img 640 --batch 1 --epochs 25 --data "C:\Users\GRT\PycharmProjects\yolov5-7.0\animal_training\dataset.yaml

" --weights "C:\Users\GRT\PycharmProjects\yolov5-7.0\MyFunction\yolov5s.pt" --device 0

问题点:

"C:\Users\GRT\PycharmProjects\yolov5-7.0\MyFunction\yolov5s.pt"这一行代码带有双引号
解决方法:

把双引号去掉就可以了

C:\Users\GRT\PycharmProjects\yolov5-7.0\MyFunction\yolov5s.pt

完整指令代码:

"python train.py --img 640 --batch 1 --epochs 25 --data "C:\Users\GRT\PycharmProjects\yolov5-7.0\animal_training\dataset.yaml

" --weights C:\Users\GRT\PycharmProjects\yolov5-7.0\MyFunction\yolov5s.pt --device 0

疑问

在没有添加"--device 0"的时候,使用cpu训练带双引号是没问题的,但加上了"--device 0" 这一代码之后使用cuda训练就报这个错误,在网上也有看到有的人说把单引号改成双引号,有的说把双引号改成单引号,我都试了,最后把所有引号都去掉就可以正常运行,所以遇到这个问题,见仁见智吧

后续

把"--device 0" 位置调整一下就不会出现心上错误

"python train.py --img 640 --batch 1 --epochs 25 --device 0 --data "C:\Users\GRT\PycharmProjects\yolov5-7.0\animal_training\dataset.yaml" --weights "C:\Users\GRT\PycharmProjects\yolov5-7.0\MyFunction\yolov5s.pt"

代码这样写不会报错!

相关推荐
GoCoding15 分钟前
Triton + RISC-V
pytorch·openai·编译器
GoCodingInMyWay1 小时前
Triton + RISC-V
pytorch·riscv·triton
duyinbi75172 小时前
ADown高效下采样改进YOLOv26目标检测性能提升
yolo·目标检测·目标跟踪
棱镜研途2 小时前
EI会议分享 | 2026年图像处理与模式识别国际会议(IC-IPPR 2026)【SPIE出版】
图像处理·人工智能·深度学习·目标检测·计算机·计算机视觉·视觉检测
rebekk2 小时前
pytorch custom op的简单介绍
人工智能·pytorch·python
von Neumann3 小时前
大模型从入门到应用——HuggingFace:Transformers-[AutoClass]
人工智能·深度学习·机器学习·ai·大模型·huggingface
AidLux3 小时前
手机上AidLux2.1.0 运行模型广场的yolov8模型
yolo·智能手机
羊小猪~~3 小时前
【论文精度】Transformer---大模型基石
人工智能·深度学习·考研·算法·机器学习·transformer
scott1985124 小时前
扩散模型之(十六)生成高分辨率图像
人工智能·深度学习