ref
com.microsoft - GridSample — Python Runtime for ONNX
onnx opset 16增加了这个定义
torch转onnx遇到的坑(一)_JoeyChen1219的博客-CSDN博客
不同align corner模式时的对齐方式
首先grid的输入x,y是归一化到+-1范围的。align_corners=0和1时,+-1边缘像素对齐到边缘像素的不同位置。
这里以输入图像有4个像素为例,展示了不同的对齐的方式。
竖向的箭头表示每个像素值对应的位置,横向的双向箭头表示每个像素占据的空间范围。
可见align_corners为0时,图像缩放系数scale为width/2,而align_corners为1时,图像缩放系数scale为(width-1)/2。缩放后0对应的中心点都是(width-1)/2。
假定像素的真实坐标为0到width-1,那么grid坐标到输入坐标的映射为out_pos*scale + (width-1)/2。