首页 > 其他分享 >YOLOv5-6.0训练出错及解决方法(RuntimeError)

YOLOv5-6.0训练出错及解决方法(RuntimeError)

时间:2024-10-24 09:32:17浏览次数:6  
标签:YOLOv5 RuntimeError shape 出错 6.0 替换

知识经验分享——YOLOv5-6.0训练出错及解决方法(RuntimeError)

一、问题bug

不知道小伙伴们在训练YOLOv5-6.0或者其他版本有没有遇到以下问题:

RuntimeError: result type Float can't be cast to the desired output type long int

二、解决方法

步骤1:打开utils/loss.py文件

步骤2:找到 for i in range(self.nl) 函数(Ctrl+F),作以下替换:

 

替换代码:

anchors, shape = self.anchors[i], p[i].shape

替换之后:

步骤3:找到# Append部分(Ctrl+F),作以下替换:

 

替换代码:

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))

替换之后:

步骤4:顺利运行train.py文件

原文地址:知识经验分享——YOLOv5-6.0训练出错及解决方法(RuntimeError) - 知乎

标签:YOLOv5,RuntimeError,shape,出错,6.0,替换
From: https://www.cnblogs.com/wanggang123/p/18498875

相关文章

  • 基于YOLOV5车牌检测的数据集
    数据集数量为四百张,内标签为face、green、blue、yellow。即人脸、绿色车牌、蓝色车牌、黄色车牌各一百张数据。本资源免费下载,因数据量较少可以给大家玩一下,不能作为任务来做的。图片均为网上数据,自己用工具标注,想要使用自动辅助标注工具可以看https://blog.csdn.net/willnzd......
  • 服务器端训练yolov5使用tensorboard+端口转发 实时查看训练成果
    服务器端训练yolov5使用tensorboard+端口转发实时查看训练成果本文参照博客园的一位大佬(相当感谢!!!):本地浏览器查看云服务器训练模型的tensorboard界面-拾一贰叁-博客园服务器端操作运行train.py开始训练新开一个终端进入到yolov5目录输入tensorboard--logdirruns/exp{......
  • 【大疆Tello无人机图传+使用YOLOv5检测】
    B站视频demo:https://www.bilibili.com/video/BV1ZvytYZEqM/代码使用注意事项:运行前,已连接Tello无人机的WIFI,并确保公用网络的防火墙已关闭!self.model=替换为自己的模型路径!视频demo代码如下:importthreadingimporttimeimportcv2importtorchimportrobomasterf......
  • YOLOv5: 原理与源码
    视频链接:YOLOv5:原理与源码_哔哩哔哩_bilibili 《YOLOv5:原理与源码》课程致力于帮助学生学习YOLOv5目标检测算法的原理与源码。常心老师将手把手从0开始解读YOLOv5-v6.0工程目录结构,解读YOLOv5-v6.0的Backbone,Neck,Head网络结构原理与源码,解读训练全流程的原理与源码,解读推......
  • Ubuntu 16.04 编译安装Python 2.7.18
    安装python2.7.18(注)使用aptinstallpython安装的版本是2.7.10,该版本对部分项目存在兼容性问题,因此需要手动编译安装安装python编译环境sudoaptinstallpython-devpkg-configlibreadline-devlibc6-devlibncursesw5-devbuild-essentialgdbpkg-configlibbz2-devlibffi-......
  • 使用飞浆ai训练yolov5
    使用飞浆ai训练yolov5飞浆ai创建项目安装环境数据集训练在yolov5目录下创建一个data.yaml,可改名因为包安装不在python的路径下,需要在py文件中添加如下命令可以导入包的位置然后可以再终端中执行训练命令参数:训练结束预测数据参数最简单的检测命令创新、修改飞浆ai......
  • Navicat Premium 16.0.90 for Mac 破解版教程
    说明本人的系统是macOS15.0.1(Sequoia),安装的版本为NavicatPremium16.0.90,资源均为从网络处下载,安装破解版有风险,请自己权衡。下载资源从https://www.mhero.com/mac-soft/dev/navicat-premium.html下载免费的Navicat。解压的密码为mhero.com。也可以从此处通过网盘分......
  • YOLO系列:YOLOv5总结
    介绍2020年6月25日,Ultralytics发布了YOLOV5的第一个正式版本,其性能与YOLOV4不相伯仲,同样也是现今最先进的对象检测技术,并在推理速度上是目前最强,yolov5按大小分为四个模型yolov5s、yolov5m、yolov5l、yolov5x。操作流程图如下:环境配置Anaconda+Pycharm安装好所需版本的A......
  • 基于yolov5的手机屏幕缺陷检测系统,支持图像、视频和摄像实时检测【pytorch框架、pytho
     更多目标检测和图像分类识别项目可看我主页其他文章功能演示:yolov5,手机屏幕缺陷检测系统,支持图像、视频和摄像实时检测【pytorch框架、python源码】_哔哩哔哩_bilibili(一)简介基于YOLOv5的手机屏幕缺陷检测系统是在PyTorch框架之下得以实现的。这是一个完备的项目,涵盖......
  • yolov5,v8船舰识别检测(包含训练代码Pyqt源码超详细)
    前言:随着海上活动的增加,对船舰进行实时、准确的检测与识别成为了海上安全、海洋监测、海事管理等领域的迫切需求。传统的船舶识别方法主要依赖于船舶自动识别系统(AIS)和雷达等通信导航设备,但这些方法存在一定的局限性,例如目标误识别、目标丢失和易受环境噪声为海上船只的自动......