记录下关于pytorch meshgrid 与stack,在yolov5 中的作用
在yolov5中需要取到特征图网格中的坐标,再加上相对0~1的偏移量,需要用到坐标值的计算。因为yolov5中偏移不需要计算最后一个边界点,所以少了一个维度(自行根据需要调整维度)刚好得到4列5行的坐标值,维度为2(y,x), 矩阵乘法。刚好对应图中的坐标, x 为5个坐标, y为4个坐标,取d[3][1] 第一个维度是y坐标的,第二个是x。我们取xv的第二行,第2列, 为2。yv的第1行,第2列, 为1。比如要计算(1,3)的偏移量。比如要取(2, 1)的坐标。所以坐标为(2, 1)
原创
2023-03-28 11:51:12 ·
225 阅读 ·
0 评论