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版权协议,转载请附上原文出处链接和本声明。