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()

>>阅读原文<<