onnx GridSample算子

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。

onnx GridSample算子_第1张图片

你可能感兴趣的:(推理引擎,算子)