yolov5-6.0

最新推荐文章于 2024-10-03 09:34:56 发布

qq_40275822

最新推荐文章于 2024-10-03 09:34:56 发布

阅读量184

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_40275822/article/details/129397900

版权

报错的原因是：
loss计算过程中，出现Float的数值精度，需要进行精度的强转换。

解决方法：
修改【utils】中的【loss.py】里面的两处内容

第一处代码修改：

anchors, shape = self.anchors[i], p[i].shape
1
第二处代码修改：

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid
————————————————
版权声明：本文为CSDN博主「千与编程」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/baidu_39629638/article/details/128025583