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"

代码这样写不会报错!

相关推荐
九年义务漏网鲨鱼10 小时前
【多模态大模型面经】 BERT 专题面经
人工智能·深度学习·bert
爱打球的白师傅10 小时前
python机器学习工程化demo(包含训练模型,预测数据,模型列表,模型详情,删除模型)支持线性回归、逻辑回归、决策树、SVC、随机森林等模型
人工智能·python·深度学习·机器学习·flask·逻辑回归·线性回归
MediaTea11 小时前
Python 第三方库:TensorFlow(深度学习框架)
开发语言·人工智能·python·深度学习·tensorflow
极客代码11 小时前
第七篇:深度学习SLAM——端到端的革命--从深度特征到神经辐射场的建图新范式
人工智能·python·深度学习·计算机视觉·slam·回环检测·地图构建
有Li11 小时前
面向超声半监督分割的类别特异性无标记数据风险最小化|文献速递-文献分享
人工智能·深度学习·计算机视觉
WGS.12 小时前
CMake Error at fc_base/gflags-src/CMakeLists.txt:73
深度学习
南汐汐月14 小时前
重生归来,我要成功 Python 高手--day35 深度学习 Pytorch
pytorch·python·深度学习
齐齐大魔王14 小时前
深度学习系列(二)
人工智能·深度学习
xier_ran14 小时前
深度学习:学习率衰减(Learning Rate Decay)
人工智能·深度学习·机器学习
Francek Chen14 小时前
【CANN】开启AI开发新纪元,释放极致计算效率
人工智能·深度学习·cann·ai开发