首页 > 其他 > 详细

PyTorch 1.0 中文官方教程:空间变换器网络教程

时间:2019-02-03 15:42:34      阅读:200      评论:0      收藏:0      [点我收藏+]

译者:冯宝宝

作者: Ghassen HAMROUNI

技术分享图片

在本教程中,您将学习如何使用称为空间变换器网络的视觉注意机制来扩充您的网络。你可以在 DeepMind paper阅读有关空间变换器网络的更多内容。

空间变换器网络是对任何空间变换的差异化关注的概括。空间变换器网络(简称STN)允许神经网络学习如何在输入图像上执行空间变换,以增强模型的几何不变性。例如,它可以裁剪感兴趣的区域,缩放并校正图像的方向。它可能是一种有用的机制,因为CNN对于旋转和缩放以及更一般的仿射变换并不是不变的。
关于STN的最棒的事情之一是能够简单地将其插入任何现有的CNN,只需很少的修改。

# License: BSD
# 作者: Ghassen Hamrouni

from __future__ import print_function  
import torch  
import torch.nn as nn  
import torch.nn.functional as F  
import torch.optim as optim  
import torchvision  
from torchvision import datasets, transforms  
import matplotlib.pyplot as plt  
import numpy as np  

plt.ion()   # 交互模式

加载数据

在这篇文章中,我们尝试了经典的MNIST数据集。使用标准卷积网络增强空间变换器网络。

阅读全文/改进本文

PyTorch 1.0 中文官方教程:空间变换器网络教程

原文:https://www.cnblogs.com/wizardforcel/p/10350423.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!