当前位置: 首页 > news >正文

RuntimeError: result type Float can‘t be cast to the desired output type __int64报错解决方法

  小白刚开始学习YOLOv5,跟随老哥的步骤走了一遍目标检测--手把手教你搭建自己的YOLOv5目标检测平台

  最后训练最后一步出现RuntimeError: result type Float can‘t be cast to the desired output type __int64报错

解决方法:找到5.0版报错的loss.py中最后那段for函数,将其整体替换为yolov5-master版中loss.py最后一段for函数即可正常运行

        for i in range(self.nl):anchors, shape = self.anchors[i], p[i].shapegain[2:6] = torch.tensor(shape)[[3, 2, 3, 2]]  # xyxy gain# Match targets to anchorst = targets * gain  # shape(3,n,7)if nt:# Matchesr = t[..., 4:6] / anchors[:, None]  # wh ratioj = torch.max(r, 1 / r).max(2)[0] < self.hyp['anchor_t']  # compare# j = wh_iou(anchors, t[:, 4:6]) > model.hyp['iou_t']  # iou(3,n)=wh_iou(anchors(3,2), gwh(n,2))t = t[j]  # filter# Offsetsgxy = t[:, 2:4]  # grid xygxi = gain[[2, 3]] - gxy  # inversej, k = ((gxy % 1 < g) & (gxy > 1)).Tl, m = ((gxi % 1 < g) & (gxi > 1)).Tj = torch.stack((torch.ones_like(j), j, k, l, m))t = t.repeat((5, 1, 1))[j]offsets = (torch.zeros_like(gxy)[None] + off[:, None])[j]else:t = targets[0]offsets = 0# Definebc, gxy, gwh, a = t.chunk(4, 1)  # (image, class), grid xy, grid wh, anchorsa, (b, c) = a.long().view(-1), bc.long().T  # anchors, image, classgij = (gxy - offsets).long()gi, gj = gij.T  # grid indices# Appendindices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, gridtbox.append(torch.cat((gxy - gij, gwh), 1))  # boxanch.append(anchors[a])  # anchorstcls.append(c)  # class

http://www.lryc.cn/news/133596.html

相关文章:

  • 解析Python爬虫常见异常及处理方法
  • 详解Spring的循环依赖问题、三级缓存解决方案源码分析
  • oracle分析函数学习
  • 代码随想录训练营day17|110.平衡二叉树 257. 二叉树的所有路径 404.左叶子之和 v...
  • C# Thread用法
  • 新榜 | CityWalk本地生活商业价值洞察报告
  • LVS负载均衡集群-NAT模式部署
  • C++学习笔记总结练习:effective 学习日志
  • Vue教程(五):样式绑定——class和style
  • 开放网关架构演进
  • torch一些操作
  • ICCV23 | Ada3D:利用动态推理挖掘3D感知任务中数据冗余性
  • 软件工程模型-架构师之路(四)
  • ubuntu20.04共享文件夹—— /mnt/hgfs里没有共享文件夹
  • Redis中的有序集合及其底层跳表
  • js 小程序限流函数 return闭包函数执行不了
  • 【数据结构】堆的初始化——如何初始化一个大根堆?
  • 【韩顺平 零基础30天学会Java】程序流程控制(2days)
  • 从入门到精通Python隧道代理的使用与优化
  • 19万字智慧城市总体规划与设计方案WORD
  • [赛博昆仑] 腾讯QQ_PC端,逻辑漏洞导致RCE漏洞
  • python Requests
  • 【深入解析:数据结构栈的魅力与应用】
  • 安卓机显示屏的硬件结构
  • 基于swing的超市管理系统java仓库库存进销存jsp源代码mysql
  • 常用系统命令
  • 【Spring专题】Spring之Bean生命周期源码解析——阶段四(Bean销毁)(拓展,了解就好)
  • 配置Docker,漏洞复现
  • 微信小程序 游戏水平评估系统的设计与实现_pzbe0
  • moba登录不进去提示修改问题问题解决方式