文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2023-09-07 05:49

关注

  小白刚开始学习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].shape            gain[2:6] = torch.tensor(shape)[[3, 2, 3, 2]]  # xyxy gain            # Match targets to anchors            t = targets * gain  # shape(3,n,7)            if nt:                # Matches                r = t[..., 4:6] / anchors[:, None]  # wh ratio                j = 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                # Offsets                gxy = t[:, 2:4]  # grid xy                gxi = gain[[2, 3]] - gxy  # inverse                j, k = ((gxy % 1 < g) & (gxy > 1)).T                l, m = ((gxi % 1 < g) & (gxi > 1)).T                j = 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            # Define            bc, gxy, gwh, a = t.chunk(4, 1)  # (image, class), grid xy, grid wh, anchors            a, (b, c) = a.long().view(-1), bc.long().T  # anchors, image, class            gij = (gxy - offsets).long()            gi, gj = gij.T  # grid indices            # Append            indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, grid            tbox.append(torch.cat((gxy - gij, gwh), 1))  # box            anch.append(anchors[a])  # anchors            tcls.append(c)  # class

来源地址:https://blog.csdn.net/weixin_54713879/article/details/125612388

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-服务器
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯