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"

代码这样写不会报错!

相关推荐
汤姆yu1 天前
基于YOLOv8模型的行人车辆多目标检测计数与跟踪系统
yolo
Hcoco_me1 天前
大模型面试题30:Padding 的 mask 操作
人工智能·rnn·深度学习·lstm·word2vec
2401_841495641 天前
【机器学习】深度信念网络(DBN)
人工智能·python·深度学习·神经网络·机器学习·无监督预训练·有监督微调
音沐mu.1 天前
【38】混凝土裂缝数据集(有v5/v8模型)/YOLO混凝土裂缝检测
yolo·目标检测·计算机视觉·数据集·混凝土裂缝数据集·混凝土裂缝检测
QtHalcon1 天前
YOLOv11 +QT环境安装+标注训练+部署
深度学习·qt·yolo
Hcoco_me1 天前
大模型面试题33:Transformer为什么用LayerNorm,而非BatchNorm?
人工智能·rnn·深度学习·自然语言处理·transformer·word2vec
谁刺我心1 天前
深度学习pytorch环境配置【保姆级】
人工智能·pytorch·深度学习
IT1425463551 天前
大模型核心技术解析
人工智能·深度学习·大模型·llm
qunaa01011 天前
YOLOV8与CGAFusion融合实现建筑工人头部安全装备检测
安全·yolo·目标跟踪
电科_银尘1 天前
【Python/Pytorch】-- 创建 tiny-cuda-nn 环境
开发语言·pytorch·python