记录下关于pytorch meshgrid 与stack,在yolov5 中的作用

文章目录

    • meshgrid
    • 作用
    • torch.stack
    • yolov5 中的make_grid
    • 6.1 结构图

meshgrid

用于扩展两个相同类型的同维度张量,举个列子

记录下关于pytorch meshgrid 与stack,在yolov5 中的作用_第1张图片

  • 创建了两个一维的张量,分别为5, 4
  • 将两个张量扩张到一起
  • 扩张后的维度为第一个张量的维度X 第二个张量的维度

作用

记录下关于pytorch meshgrid 与stack,在yolov5 中的作用_第2张图片
在yolov5中需要取到特征图网格中的坐标,再加上相对0~1的偏移量,需要用到坐标值的计算

刚好对应图中的坐标, x 为5个坐标, y为4个坐标,
因为yolov5中偏移不需要计算最后一个边界点,所以少了一个维度(自行根据需要调整维度)

比如要取(2, 1)的坐标
我们取xv的第二行,第2列

你可能感兴趣的:(yolo,专栏,一起学习进步,深度学习,yolov5,pytorch,python,深度学习)