【开源计划】图像配准中变形操作（Warp）的pytorch实现

前言

按照开源计划的预告，这次我来分享图像配准流程中的变形操作的代码实现。首先我们先来回顾一下配准的流程，我们以这篇Unsupervised End-to-end Learning for Deformable Medical Image Registration论文中的流程图为例，进行说明。该论文提出的配准框架是基于无监督学习的端到端的非刚性图像配准，配准网络（registration network）根据输入的浮动图像（moving image）与固定图像(fixed image)预测出变形场（deformation field），然后采样网格生成器（sampling grid generator）生成规则网格（regular grid），并与变形场结合得到采样网格（sampling grid），再经过采样器（sampler）对浮动图像进行重采样即得到变形后的图像（warped image）。其中，从规则网格生成到得到变形后的图像的过程一般称为变形（warp，可能翻译的不够准确与专业）。另外，这个过程也正是论文Spatial Transformer Networks的思想，只不过最初该论文的应用是在自然图像的检测与分类任务上的，后来才被引入到医学图像配准上。正是它的引入，使得这种基于无监督学习的图像配准变得可行。

该流程图的原始说明引用如下：

Fig. 2: Illustration of the unsupervised training strategy of our fully convolutional image-to-image registration network. The registratio