2022最新版-李宏毅机器学习深度学习课程-P25 Spacial Transformer Layer

data augmentation/spacial transformer

CNN 并不能够处理影像放大缩小,或者是旋转的问题。所以在做影像辨识的时候,往往都要做 Data Augmentation,把你的训练数据截一小块出来放大缩小、把图片旋转,CNN 才会做到好的结果。

有一个架构叫 spacial Transformer Layer可以处理。

设计一个层,需要的地方=1,不要的地方=0

经过一个NN,FP发生平移

还可以放大或缩小,只需要改变参数

旋转一共需要6个参数

当6个参数全部设定的是整数时,结果比较好处理

但如果是小数时呢?通过最终答案的四舍五入得到结果。但此时梯度也不好计算。

可以这样假设:假设最终结果发生一些微小的变化,四舍五入后得到的结果是一样的,这就说明梯度几乎=0!!

使用插值方法代替四舍五入

最终结果与周围四个值都有关。此时输出值的变化时刻影响最终结果。

把ST看成橙色,它可以放在很多地方。

这个视频的含义是改变一些输入后(变大 变小 左右旋转),经过ST层,输出结果基本不变。

实例

这里用了single ST和multi ST,效果有些变化

这个例子用了2层和4层的ST,输入图片有两种大小规格。

2层的是只识别鸟嘴和鸟翼;4层还识别了其他地方。

相关推荐
我是李武涯22 分钟前
PyTorch Dataloader工作原理 之 default collate_fn操作
pytorch·python·深度学习
无风听海1 小时前
神经网络之计算图repeat节点
人工智能·深度学习·神经网络
吃饭睡觉发paper1 小时前
Learning Depth Estimation for Transparent and Mirror Surfaces
人工智能·机器学习·计算机视觉
MYX_3091 小时前
第三章 线型神经网络
深度学习·神经网络·学习·算法
小龙3 小时前
【基础理论】位置向量|位置编码学习笔记
大模型·transformer·基础理论·位置编码
患得患失9493 小时前
【NestJS】class-transformer什么用
transformer·nestjs
Aaplloo3 小时前
机器学习作业七
人工智能·机器学习
Cathy Bryant4 小时前
矩阵乘以向量?向量乘以向量?
笔记·神经网络·考研·机器学习·数学建模
应用市场4 小时前
GPS车辆实时定位与轨迹预测技术实现
深度学习
技术闲聊DD5 小时前
深度学习(5)-PyTorch 张量详细介绍
人工智能·pytorch·深度学习