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"

代码这样写不会报错!

相关推荐
一个王同学3 小时前
从零到一 | CV转多模态大模型 | week09 | Minillava Refactor结合手搓和llava源码深入理解多模态大模型原理
人工智能·深度学习·机器学习·计算机视觉·改行学it
赢乐5 小时前
大模型学习笔记:检索增强生成(RAG)架构
人工智能·python·深度学习·机器学习·智能体·幻觉·检索增强生成(rag)
多年小白11 小时前
紫光国微(002049) 分析
大数据·科技·深度学习·ai
动物园猫11 小时前
金属外表多种生锈检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·分类
cyyt12 小时前
深度学习周报(5.11~5.17)
人工智能·深度学习
搬砖的小码农_Sky12 小时前
如何在RX 7900 XTX显卡上运行ROCm?
深度学习·机器学习·gpu算力
RSTJ_162512 小时前
PYTHON+AI LLM DAY FOURTY-EIGHT
开发语言·人工智能·python·深度学习
weixin_5500831513 小时前
亚博智能K230跑YOLOv8实时目标检测模型识别,从模型训练→转换ONNX→kmodel→板子运行步骤识别代码和踩坑记录最终使用wifi转换到
yolo
Cosolar13 小时前
AI Agent 记忆机制全景对比:OpenClaw vs QwenPaw vs Hermes vs HiClaw
人工智能·深度学习·语言模型·chatgpt·面试
爱吃肉的鹏14 小时前
[特殊字符] 基于全YOLO系列(含YOLO26)的行人重识别项目——只需一条命令!可做嫌疑人检测、特定人员检测、走失儿童检测!
yolo