Spatial Transformer Networks(STN)

本文深入探讨了卷积神经网络(CNN)的平移不变性原理,解释了为何CNN能够在特征位置变化时仍能捕捉到它们。同时,介绍了空间变换网络(STN)的局部化网部分,该网络通过一系列层计算回归出参数,实现对图像的空间变换。STN的Grid Generator组件在其中起到关键作用,调整图像坐标以适应不同的变换需求。
摘要由CSDN通过智能技术生成

1. 首先要知道CNN具有平移不变性

        这是为什么呢?可以回忆一下CNN都做了什么,第一就是卷积,第二就是maxpool。

        当图像的某一个特征从左上角移动到右下角了,那我们CNN会不会受到影响呢?

        因为我们卷积是整个图像都扫一遍,所以无论特征在哪里,我们的卷积核都会扫一遍,所以特征都会被采样到。

        接着是maxpool,maxpool的作用就是将感受野里面的最大值给拿出来,其实也是卷积的作用,只不过是规则不同了,所以只要特征在上面卷积保留下来了,在maxpool这里也会被输出的。

        所以不管特征在哪里,CNN都是把它给挖出来!

2. STN的结构

Localisation net 
        把原图作为输入,过连续若干层计算(如卷积、FC等),回归出参数\theta,θ的维度取决于网络选择的具体变换类型,如选择仿射变换则\theta\subseteq R^{2*3}。如选择投影变换则 \theta\subseteq R^{3*3} 。θ的值决定了网络选择的空间变换的”幅度大小”

Grid generator:

放一下,后面更新

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
空间变换网络(Spatial Transformer NetworksSTN)是一种神经网络结构,用于改善卷积神经网络(CNN)的空间不变性。STN可以对经过平移、旋转、缩放和裁剪等操作的图像进行变换,使得网络在变换后的图像上得到与原始图像相同的检测结果,从而提高分类的准确性。STN由三个主要部分组成:局部化网络(Localisation Network)、参数化采样网格(Parameterised Sampling Grid)和可微分图像采样(Differentiable Image Sampling)。 局部化网络是STN的关键组件,它负责从输入图像中学习如何进行变换。局部化网络通常由卷积和全连接层组成,用于估计变换参数。参数化采样网格是一个由坐标映射函数生成的二维网格,它用于定义变换后每个像素在原始图像中的位置。可微分图像采样则是通过应用参数化采样网格来执行图像的变换,并在变换后的图像上进行采样。 使用STN的主要优点是它能够在不改变网络结构的情况下增加空间不变性。这使得网络能够处理更广泛的变换,包括平移、旋转、缩放和裁剪等。通过引入STN层,CNN可以学习到更鲁棒的特征表示,从而提高分类准确性。 关于STN的代码实现,您可以在GitHub上找到一个示例实现。这个实现使用TensorFlow框架,提供了STN网络的完整代码和示例。您可以通过查看该代码来了解如何在您的项目中使用STN。 综上所述,spatial transformer networks(空间变换网络)是一种神经网络结构,用于增加CNN的空间不变性。它包括局部化网络、参数化采样网格和可微分图像采样三个部分。通过引入STN层,CNN可以学习到更鲁棒的特征表示,从而提高分类准确性。在GitHub上有一个使用TensorFlow实现的STN示例代码供参考。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值