tensor([0.00, 0.00, 0.01, ..., 0.99, 1.00, 1.00])
torch.meshgrid()函数看不懂可以看这里
torch.meshgrid()函数解析_小娜美要努力努力的博客-优快云博客
torch.stack((-w, -h, w, h)).T.repeat(in_height * in_width, 1)
这时候的shape为[h*w*5, 4],第一维度也就是一张图片所有的锚框数,第二维度为两个顶点的xy坐标。
但我们所需的是半宽半高(因为我们使用半宽半高加上中心点才是顶点坐标)
————————————————
版权声明:本文为优快云博主「CuddleSabe」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.youkuaiyun.com/qq_15534667/article/details/127991387