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

yolov5训练自定义数据集的时候出现如下问题

RuntimeError: result type Float can’t be cast to the desired output type long int
问题原因:
官网的yolov5-master版本可以正常运行,但是yolov5-5.0/yolov5-6.1等版本就是不可以运行
问题解决:
修改【utils】中的【loss.py】里面的两处内容

  • 一处替换成:anchors, shape = self.anchors[i], p[i].shape

    在这里插入图片描述

  • 一处替换成:indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid
    在这里插入图片描述


版权声明:本文为weixin_49148527原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。