YOLO V5出现RuntimeError: result type Float can‘t be cast to the desired output type long int解决方法

2023-05-16

在使用YOLO框架训练自己的数据集时候,开始跑train.py,出现如下报错:RuntimeError: result type Float can‘t be cast to the desired output type long int

Traceback (most recent call last):
  File "/home/sjh/project/yolov5-5.0/train.py", line 543, in <module>
    train(hyp, opt, device, tb_writer)
  File "/home/sjh/project/yolov5-5.0/train.py", line 304, in train
    loss, loss_items = compute_loss(pred, targets.to(device))  # loss scaled by batch_size
  File "/home/sjh/project/yolov5-5.0/utils/loss.py", line 117, in __call__
    tcls, tbox, indices, anchors = self.build_targets(p, targets)  # targets
  File "/home/sjh/project/yolov5-5.0/utils/loss.py", line 211, in build_targets
    indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))  # image, anchor, grid indices
RuntimeError: result type Float can't be cast to the desired output type long int

由于是初次使用,并不确定是哪里的问题,只知道是跟着其他人的教程来的,经过查询,得到问题原因:

官网的yolov5-master可以正常运行,但是yolov5-5.0/yolov5-6.1等版本会出问题;

这是因为yolov5-master版本和yolov5-5.0/yolov5-6.1等版本下的【utils】中的【loss.py】文件不同,大概是yolov5-5.0/yolov5-6.1等版本在更新版本的时候出了问题

因此解决方法也明了了:修改loss.py文件

打开【utils】文件夹下的【loss.py】文件,一共需要修改两处代码,以yolov5-5.0为例:

1,大概是177行左右,可通过搜索找到下面这句代码

anchors = self.anchors[i]

将这一行其替换为

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

2,在程序的最后,找到下面这行代码

indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))  # image, anchor, grid indices

将其替换为

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

保存程序,重新跑train.py,便可以正常运行了

 

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

YOLO V5出现RuntimeError: result type Float can‘t be cast to the desired output type long int解决方法 的相关文章

随机推荐

  • 视频转换处理软件

    视频转换处理软件是一款专业 易用 高效的视频处理工具 xff0c 软件内置了格式转换 视频剪切 视频转GIF 视频加水印 视频优化 旋转和翻转等六大功能 xff0c 并每个功能模块也都是互相独立的 xff0c 这也就是说 xff0c 用户在
  • 视频剪切软件

    视频剪切是一款功能强势 操作简单的视频剪切工具 xff0c 它不但可以帮助用户将视频中的精彩片段剪切出来 xff0c 轻松制作需要的短片视频 xff0c 而且还支持任意设置输出视频的视频属性和音频属性 xff0c 并支持MP4 rmvb a
  • 神奇电商宝贝下载软件

    神奇电商宝贝下载软件是一款专业 易用 高效的电商软件 xff0c 可帮助用户快速获取1688 天猫 淘宝 速卖通 企业叮咚等平台店铺中的主图视频 详情视频 主图图片 详情图片等所有信息 xff0c 从而轻松满足电商用户的所有需求 而且该软件
  • 布谷鸟算法

    布谷鸟算法是将布谷鸟育雏行为与Levy飞行算法相结合的一种算法 在布谷鸟算法中 xff0c 有两个算法或者说两个位置更新是关键 xff1a 第一个是布谷鸟寻找最优解时的算法 xff1a 一个是布谷鸟寻找鸟窝下蛋的寻找路径是采用早已就有的萊维
  • matlab实现虚拟力策略传感器分布

    第一次写文章 主要想记录一下编程过程中的坎 1 虚拟力策略 虚拟力法原理是 将机器人在环境中的运动视为一种机器人在虚拟的人工受力场的运动 障碍物对机器人产生斥力 目标点对机器人产生引力 引力和斥力的合力作为机器人的加速力 来控制机器人的运动
  • pandas:sample函数解释

    目录 1 函数定义 2 作用 xff1a 3 举个栗子 4 参数解释 1 函数定义 DataFrame sample self FrameOrSeries n 61 None frac 61 None replace 61 False we
  • [Pixhawk/PX4]开发环境搭建(Ubuntu 18.04)和问题总结

    主要记录了PX4环境在Ubuntu 18 04下的搭建过程和问题总结 由于gazebo在ros安装中一般会包括 xff0c 所以gazebo安装没有介绍 一 基础资源下载 这是我第一次接触ubuntu系统的安装 xff0c 所以可能有些步骤
  • 数独基础技巧

    数独基础思想 数独作为一个益智游戏 xff0c 想必大家都或多或少接触过 xff0c 基础思想就是在9 9的方格中 xff0c 每一行每一列每个九宫格都不重复地存在1 9这九个数字 xff0c 这也意味着对一个正确的数独题目都有一个固定的答
  • VM VirtualBox虚拟机路径迁移

    在学习Ubuntu的时候把虚拟机放到了机械硬盘 xff0c 所以系统运行很卡顿 因此尝试了迁移虚拟机到固态硬盘中 方法如下 xff1a 导出需要迁移的虚拟电脑 点击管理 xff0c 点击导出虚拟电脑 选择需要导出的虚拟电脑 xff0c 点击
  • 设计使用 | 四个免费的渐变配色网站

    在进行设计工作时 xff0c 有时候需要一些渐变的配色做底图更好看些 xff0c 本来用Photoshop也可以自己制作 xff0c 但是经常一做就很丑 xff0c 所以还是直接找其他网站上的吧 一 CoolHue 地址 xff1a Gra
  • Ubuntu+Windows双系统,开机默认启动项设置Windows优先

    电脑是Ubuntu18 04 43 Win10双系统 xff0c 开机一直都是默认先Ubuntu xff0c Windows在最下面 xff0c 有时候按下开机键一不留神就进Ubuntu了 xff0c 于是寻思着在启动项中设置为默认优先Wi
  • 解决Gazebo闪退,提示escalating to SIGKILL on server的问题,Melodic更新gazebo9到gazebo11

    0 xff0c 问题描述 今天拿Gazebo自定义一个简单地图 xff0c 建完墙都没什么事 xff0c 结果在墙上加门的时候 xff0c 程序卡死了 xff0c 重复几次都是这样的情况 xff0c 在终端中也出现了提示 escalatin
  • Windows 10 安装 Android 13版本的安卓子系统(带 Google Play 商店和 Magisk) 2210.40000.7.0

    根据微软的说明 xff0c WSA 仅支持 Windows11 xff0c Windows10 及以前的操作系统无缘WSA 不过我们来看一下原理哈 xff1a Android 这个系统 xff0c 它是基于Linux 系统来开发的 xff0
  • Windows 11 安装 Android 13版本的安卓子系统(带 Google Play 商店和 Magisk) 2210.40000.10.0

    微软最近刚刚发布了适用于 Android 的 Windows 子系统 xff0c 构建版本为 2211 xff0c 采用 Android 13 xff0c 目前这还是一个预览版本 xff0c 理论上只有Beta用户才能用 xff0c 但是用
  • 利用 ChatGPT 简化微信聊天内容

    以下文章来源于蓝点网 xff0c 作者山外的鸭子哥 xff0c 侵删 各种聊天软件的群太多 消息太多压根没时间看怎么办 xff1f 那试试使用 ChatGPT 帮你整理群聊记录提取重点和发个总结内容 xff1f ChatGPT目前已经在很多
  • 一步设置,立刻开启New Bing!

    前阵子微软的新必应不是特火嘛 xff0c 我也想试试来着 xff0c 结果除了要一堆设置之外 xff0c 还要等待候补 xff0c 所以也就不了了之了 xff0c 结果今天突然发现 xff0c 只需要进行简单的一步设置 xff0c 就可以立
  • pandas:reset_index及set_index的解释

    目录 1 pandas DataFrame set index 2 pandas DataFrame reset index 1 pandas DataFrame set index 函数原型 xff1a DataFrame set ind
  • New Bing 桌面客户端,支持 Linux/macOS/Win,开源免费,国人开发

    实话实说 xff0c 我并不是很喜欢Edge浏览器 xff0c 但是想用 New Bing 的话 xff0c 就得用它 xff0c 而且我经常使用Linux系统 xff0c 想上 New Bing 就更麻烦了 所以就找到了一个 New Bi
  • 解决“python-roslaunch : 依赖: python-roslib 但是它将不会被安装”问题

    在Ubuntu 18 04安装ROS melodic后 xff0c 测试ros环境 xff1a roscore 发现有提示报错 xff1a Command 39 roscore 39 not found but can be install
  • YOLO V5出现RuntimeError: result type Float can‘t be cast to the desired output type long int解决方法

    在使用YOLO框架训练自己的数据集时候 xff0c 开始跑train py xff0c 出现如下报错 xff1a RuntimeError result type Float can t be cast to the desired out