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"

代码这样写不会报错!

相关推荐
weixin_445238122 小时前
Pytorch|RNN-心脏病预测
人工智能·pytorch·rnn
Thomas_Cai3 小时前
Bert论文解析
人工智能·深度学习·nlp·bert·transformer
羡江0074 小时前
机器学习(概述)
人工智能·深度学习·机器学习
Cl_rown去掉l变成C4 小时前
第P10周:Pytorch实现车牌识别
人工智能·pytorch·python
西红柿土豆丶5 小时前
人脸考勤管理一体化系统(人脸识别系统,签到打卡)
python·深度学习·opencv·人脸识别·人脸识别系统·考勤管理系统·签到打卡
AIGC_ZY5 小时前
PyTorch 实现图像版多头注意力(Multi-Head Attention)和自注意力(Self-Attention)
人工智能·pytorch·python
万里鹏程转瞬至5 小时前
深度学习中模型量化那些事
人工智能·深度学习
Ronin-Lotus6 小时前
深度学习篇---模型训练(1)
人工智能·python·深度学习
硬水果糖7 小时前
神经网络之损失函数
人工智能·深度学习·神经网络
xuebodx09237 小时前
私有部署stable-diffusion-webui
图像处理·pytorch·ai作画·stable diffusion·视觉检测·transformer·dall·e 2