PyTorch实现AlexNet示例


Posted in Python onJanuary 14, 2020

PyTorch: https://github.com/shanglianlm0525/PyTorch-Networks

PyTorch实现AlexNet示例

import torch
import torch.nn as nn
import torchvision

class AlexNet(nn.Module):
  def __init__(self,num_classes=1000):
    super(AlexNet,self).__init__()
    self.feature_extraction = nn.Sequential(
      nn.Conv2d(in_channels=3,out_channels=96,kernel_size=11,stride=4,padding=2,bias=False),
      nn.ReLU(inplace=True),
      nn.MaxPool2d(kernel_size=3,stride=2,padding=0),
      nn.Conv2d(in_channels=96,out_channels=192,kernel_size=5,stride=1,padding=2,bias=False),
      nn.ReLU(inplace=True),
      nn.MaxPool2d(kernel_size=3,stride=2,padding=0),
      nn.Conv2d(in_channels=192,out_channels=384,kernel_size=3,stride=1,padding=1,bias=False),
      nn.ReLU(inplace=True),
      nn.Conv2d(in_channels=384,out_channels=256,kernel_size=3,stride=1,padding=1,bias=False),
      nn.ReLU(inplace=True),
      nn.Conv2d(in_channels=256,out_channels=256,kernel_size=3,stride=1,padding=1,bias=False),
      nn.ReLU(inplace=True),
      nn.MaxPool2d(kernel_size=3, stride=2, padding=0),
    )
    self.classifier = nn.Sequential(
      nn.Dropout(p=0.5),
      nn.Linear(in_features=256*6*6,out_features=4096),
      nn.ReLU(inplace=True),
      nn.Dropout(p=0.5),
      nn.Linear(in_features=4096, out_features=4096),
      nn.ReLU(inplace=True),
      nn.Linear(in_features=4096, out_features=num_classes),
    )
  def forward(self,x):
    x = self.feature_extraction(x)
    x = x.view(x.size(0),256*6*6)
    x = self.classifier(x)
    return x


if __name__ =='__main__':
  # model = torchvision.models.AlexNet()
  model = AlexNet()
  print(model)

  input = torch.randn(8,3,224,224)
  out = model(input)
  print(out.shape)

以上这篇PyTorch实现AlexNet示例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python操作xml文件示例
Apr 07 Python
Python实现的文本编辑器功能示例
Jun 30 Python
python操作小程序云数据库实现简单的增删改查功能
Jun 06 Python
set在python里的含义和用法
Jun 24 Python
Apache,wsgi,django 程序部署配置方法详解
Jul 01 Python
超简单的Python HTTP服务
Jul 22 Python
对Django url的几种使用方式详解
Aug 06 Python
scikit-learn线性回归,多元回归,多项式回归的实现
Aug 29 Python
python nmap实现端口扫描器教程
May 28 Python
Pytorch中accuracy和loss的计算知识点总结
Sep 10 Python
PyCharm2020.1.2社区版安装,配置及使用教程详解(Windows)
Aug 07 Python
Python基础之元类详解
Apr 29 Python
Pytorch 实现focal_loss 多类别和二分类示例
Jan 14 #Python
Python实现钉钉订阅消息功能
Jan 14 #Python
Python Tensor FLow简单使用方法实例详解
Jan 14 #Python
Python利用全连接神经网络求解MNIST问题详解
Jan 14 #Python
基于pytorch的lstm参数使用详解
Jan 14 #Python
Python利用逻辑回归模型解决MNIST手写数字识别问题详解
Jan 14 #Python
np.random.seed() 的使用详解
Jan 14 #Python
You might like
绿山咖啡和蓝山咖啡
2021/03/04 新手入门
PHP树形结构tree类用法示例
2019/02/01 PHP
htm调用JS代码
2007/03/15 Javascript
List the UTC Time on a Computer
2007/06/11 Javascript
新老版本juqery获取radio对象的方法
2010/03/01 Javascript
Javascript 入门基础学习
2010/03/10 Javascript
javascript中的事件代理初探
2014/03/08 Javascript
基于jquery的文字向上跑动类似跑马灯的效果
2014/09/22 Javascript
jQuery带箭头提示框tooltips插件集锦
2014/11/17 Javascript
PHP+jQuery实现随意拖动层并即时保存拖动位置
2015/04/30 Javascript
js日期插件dateHelp获取本月、三个月、今年的日期
2016/03/07 Javascript
JQuery 的跨域方法推荐_可跨任何网站
2016/05/18 Javascript
所见即所得的富文本编辑器bootstrap-wysiwyg使用方法详解
2016/05/27 Javascript
BootStrap Table 分页后重新搜索问题的解决办法
2016/08/08 Javascript
不得不知的ES6小技巧
2018/07/28 Javascript
详解Next.js页面渲染的优化方案
2019/01/27 Javascript
实例详解vue中的$root和$parent
2019/04/29 Javascript
js+html实现点名系统功能
2019/11/05 Javascript
js面向对象方式实现拖拽效果
2021/03/03 Javascript
Python自动化测试工具Splinter简介和使用实例
2014/05/13 Python
Linux下将Python的Django项目部署到Apache服务器
2015/12/24 Python
python日志记录模块实例及改进
2017/02/12 Python
CentOS 6.5中安装Python 3.6.2的方法步骤
2017/12/03 Python
python的numpy模块安装不成功简单解决方法总结
2017/12/23 Python
pandas object格式转float64格式的方法
2018/04/10 Python
Python命令行参数argv和argparse该如何使用
2021/02/08 Python
HTML5 File接口在web页面上使用文件下载
2017/02/27 HTML / CSS
英国一家专门出售品牌鞋子的网站:Allsole
2016/08/07 全球购物
英国在线定制百叶窗网站:Swift Direct Blinds
2020/02/25 全球购物
药学专业大学生自荐信
2013/09/28 职场文书
网络教育毕业生自我鉴定
2013/10/10 职场文书
大学生简单自荐信
2013/11/10 职场文书
销售人员求职的自我评价分享
2014/03/15 职场文书
2016年度先进班组事迹材料
2016/03/01 职场文书
详细聊聊MySQL中慢SQL优化的方向
2021/08/30 MySQL
css让页脚保持在底部位置的四种方案
2022/07/23 HTML / CSS