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"

代码这样写不会报错!

相关推荐
Faker66363aaa7 分钟前
航空基地设施目标检测 - YOLOv26实现战斗机机库非作战飞机旋翼飞机自动识别定位
人工智能·yolo·目标检测
Lun3866buzha8 分钟前
Bundaberg Rum 700mL酒瓶检测实战:基于YOLOv26的高精度识别方案
人工智能·yolo·目标跟踪
落雨盛夏2 小时前
深度学习|李哥考研4图片分类比较详细说明
人工智能·深度学习·分类
就这个丶调调8 小时前
VLLM部署全部参数详解及其作用说明
深度学习·模型部署·vllm·参数配置
森屿~~8 小时前
AI 手势识别系统:踩坑与实现全记录 (PyTorch + MediaPipe)
人工智能·pytorch·python
轴测君9 小时前
SE Block(Squeeze and Excitation Block)
深度学习·机器学习·计算机视觉
Piar1231sdafa11 小时前
基于YOLOv26的条形码与二维码识别检测实战
yolo
lixzest11 小时前
PyTorch基础知识简述
人工智能·pytorch·python
飞Link11 小时前
深度学习里程碑:ResNet(残差网络)从理论到实战全解析
人工智能·python·深度学习
轴测君11 小时前
MobileNet V1
人工智能·pytorch·笔记