根据教程运行yolov5-5.0的train.py出现以下问题

jitiw

于 2024-05-20 15:17:15 发布

阅读量518

点赞数 8

文章标签： YOLO 深度学习机器学习

本文链接：https://blog.csdn.net/jitiw/article/details/139066159

版权

File "D:\yolov5-5.0\yolov5-5.0\train.py", line 543, in <module> train(hyp, opt, device, tb_writer) File "D:\yolov5-5.0\yolov5-5.0\train.py", line 304, in train loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "D:\yolov5-5.0\yolov5-5.0\utils\loss.py", line 126, in __call__ ps = pi[b, a, gj, gi] # prediction subset corresponding to targets

~~^^^^^^^^^^^^^^

IndexError: tensors used as indices must be long, int, byte or bool tensors

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

jitiw

关注关注

8
点赞
踩
0

收藏

觉得还不错? 一键收藏
1
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

YOLOv5-LITE train训练报错总结

weixin_57011823的博客

07-27

1067

YOLOv5 lite 训练报错问题总结

记录YOLOv5运行train.py出现的问题

jehseh的博客

09-26

743

记录YOLOv5运行train.py出现的问题，以及解决办法

1 条评论您还未登录，请先登录后发表或查看评论

YOLOv5的常见报错总结

weixin_45303602的博客

09-12

1954

从上面点进~\torch\nn\modules\upsampling.py将forword中的return函数里的最后一个参数删除。如果你是下载别人的数据集进行训练的话，在其他路径配置无误的情况下报这种错说明train.py读取数据集的时候读取之前的缓存文件了。YOLOv6更新了common.py中的SPPF类而v5版本中没有，这个原因我也很迷惑，为啥跟v6有关系呢？执行train.py出现的报错，根据提示肯定是在models的common.py中出现的问题。torch的版本过高导致不兼容。

报错unrecognized arguments

qq_41670507的博客

10-17

4035

args=parser.parse_args([无空格])即可。ipynb报错如图，只需将以下注释代码修改为。

解决报错:train.py: error: unrecognized arguments: --local-rank=1 ERROR:torch.distributed.elastic.multipr

weixin_43960370的博客

04-20

1万+

由上图可以看出是–local_rank 与 --local-rank不一致导致的，追究原因，竟然是torch2.0版本launch.py里面写的全是–local-rank，而本yolov7源码用的是–local_rank。更换torch版本，我把torch版本换成1.13就好了。

【求助】error: unrecognized arguments:问题如何解决

weixin_46543875的博客

07-14

7561

【求助】error: unrecognized arguments:问题如何解决

解决python脚本错误train.py:error: unrecognized arguments

热门推荐

qq_39195733的博客

06-25

1万+

一个简单的debug 过程，希望对大家有所帮助，少踩坑

yolov5篇---官方代码docker部署训练

心惠天意

10-28

8734

yolov5训练自己的数据集（docker） yolov5官方代码：选择 tag v5.0 https://github.com/ultralytics/yolov5 训练方式可参照另一篇博客 https://blog.csdn.net/m0_46825740/article/details/119956658?spm=1001.2014.3001.5501 数据放置路径 1. 生成docker镜像 docker build -t yolov5:5.0 . 镜像成功生成 2. 生成容器 nvidia

YOLOv5目标检测之知识蒸馏实战

05-15

PyTorch版的YOLOv5是一个当前非常流行的目标检测器，本课程对YOLOv5进行知识蒸馏实战，来提升其性能。知识蒸馏（Knowledge Distillation）是模型压缩的一种常用的方法。它利用性能更好的大模型的监督信息，来训练一个轻量化小模型，使小模型达到更好的性能和精度。最早是由Hinton首次提出并应用在分类任务上，这个大模型称之为教师模型，小模型称之为学生模型。来自教师模型输出的监督信息称之为Knowledge(知识)，而学生网络学习迁移来自教师网络的监督信息的过程称之为Distillation(蒸馏)。本课程在YOLOv5 v6.1版本代码的基础上增加知识蒸馏方法，并实际演示针对自己的数据集训练和进行知识蒸馏过程，并讲解原代码针对知识蒸馏的修改部分。本课程分为原理篇、实战篇、代码讲解篇。原理篇包括：知识蒸馏的基础知识、目标检测中知识蒸馏的方法。实战篇包括：PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、训练学生网络（Yolov5s）、训练教师网络（训练YOLOv5m）、知识蒸馏训练。代码讲解篇包括：知识蒸馏具体修改代码的讲解。

yolov5-5.0.zip

08-12

此压缩包`yolov5-5.0.zip`内可能包含以下关键组件： 1. **模型架构**：YOLOv5的设计采用了先前版本的优点并进行了改进，包括更快速的检测速度和更高的精度。它采用了Focal Loss来处理类别不平衡问题，并利用Anchor ...

基于yolov5-5.0的手势识别实现

02-26

在本文中，我们将深入探讨如何基于YOLOv5-5.0实现手势识别技术。YOLO（You Only Look Once）是一种流行的实时目标检测算法，它以其高效和准确性而备受推崇。YOLOv5是该系列的最新版本，针对速度、准确性和易用性进行...

基于yolov5-5.0训练和推理目标检测的一套模板和算法

02-03

**基于yolov5-5.0的训练与推理目标检测技术详解** 在计算机视觉领域，目标检测是一项关键任务，其目标是识别图像中的不同对象并确定它们的位置。YOLO（You Only Look Once）是一种高效的目标检测算法，自2016年首次...

运行yolov5中的train.py

v587coder的博客

06-26

445

如果pycharm中报no model name numpy，暂时不用处理，是因为python的路径设置的不对导致的，可以点击file>>Settings>>project:xxx>>python interpreter>>选择python.exe所在的路径。用pycharm打开yolov5文件中的train.py文件夹，右键，点击运行，即可；

UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0x80 in position 218: illegal multibyte sequence

Vertira的博客

10-21

2041

UnicodeDecodeError: 'gbk' codec can't decode byte 0x80 in position 218: illegal multibyte sequence

Yolv5报错怎么解决

miichaelson的博客

08-15

1861

训练yolv5时出现的问题，有哪位大佬可以看看吗

yolov5单机多卡训练报错

qq_40396104的博客

07-28

898

Traceback (most recent call last): File “train.py”, line 638, in main(opt) File “train.py”, line 532, in main train(opt.hyp, opt, device, callbacks) File “train.py”, line 113, in train data_dict = data_dict or check_dataset(data) # check if None File “/ho

yolov5-5.0压缩包下载链接