pytorch的STN（空间变换网络）

STN可以直接插入现有的CNN中。 Localisation Network：一CNN生成变换参数。 Grid Genator：生成与输出图像的每个像素相对应的坐标网络。 Sampler：输出结果 #定义一个stn网络：本地网络（卷积池化relu卷积池化relu)Linear relu Linear得到3 * 2的仿射矩阵 #grid = F.affine_grid(theta, x.size()

>>阅读原文<<