博主,您好,最近正在看您写的YOLOv5涨点系列博文,在实现《高效涨点-Optimal Transport Assignment:OTA最优传输方法》这篇文章中的方法的时候,发现了一些问题,可以帮忙看一看嘛。
报错信息如下:
File "F:\PythonCode\2024\DL-project\yolov5\utils\loss.py", line 325, in __call__
tobj[b, a, gj, gi] = (1.0 - self.gr) + self.gr * iou.detach().clamp(0).type(tobj.dtype) # iou ratio
RuntimeError: shape mismatch: value tensor of shape [45, 1] cannot be broadcast to indexing result of shape [45]
仔细检查发现问题出在了这里:
希望博主抽时间可以帮忙看一下!感谢!