1.始于STN:Spatial Transformer Networks
文章提出了一个module, 可以是模型自主学习单应变换系数, 对输入图片或者feature进行单应变换, 校正目标, 提高下游任务. 模型自己学习系数, 所以不需要增加额外标注. 该部分pyrotch已经实现. 可以加入到任何网络中. 主要是两部分, 一部分回归系数, 一部分变换. 变换分两部分, 先affin_grid生成grid网格, 在使用grid_sampler, 对图像进行插值, 获取校正后的输出.
参考:https://blog.csdn.net/qq_39422642/article/details/78870629
github: