- 博客(2)
- 收藏
- 关注
RSS订阅原创 yolov5-OTA 注意力 OTA--Optimal Transport Assignment
1.1若出现RuntimeError: shape mismatch: value tensor of shape 【2, 1】 cannot be broadcast to indexing result of shape 【2】出现1.1的错误，也可以直接将报错的iou.detach()改成torch.squeeze(iou.detach())即可，通过debug找到问题点。
2023-10-24 21:56:12
335
原创 YOLOv5结构
nx 和 ny 分别表示特征图的宽度和高度，i 表示当前特征图的索引，torch_1_10 是一个用于检查 torch 版本的函数，用于选择不同的方式生成网格坐标，以保持兼容性。最后，anchor boxes 张量 self.anchors[i] 被乘以对应的步长 self.stride[i]，并且形状被调整为 (1, self.na, 1, 1, 2)，然后扩展为与网格张量相同的形状 (1, self.na, ny, nx, 2)，得到了 anchor boxes 在输入图像上的坐标。
2023-09-27 10:37:17
139
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人