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之入门(二)基本数据类型
May 25 Python
Python3中关于cookie的创建与保存
Oct 21 Python
在Python中将函数作为另一个函数的参数传入并调用的方法
Jan 22 Python
Python3进制之间的转换代码实例
Aug 24 Python
pygame实现俄罗斯方块游戏(基础篇1)
Oct 29 Python
python随机生成库faker库api实例详解
Nov 28 Python
Pytorch evaluation每次运行结果不同的解决
Jan 02 Python
python爬虫实现获取下一页代码
Mar 13 Python
Django model重写save方法及update踩坑详解
Jul 27 Python
Pycharm2020.1安装无法启动问题即设置中文插件的方法
Aug 07 Python
next在python中返回迭代器的实例方法
Dec 15 Python
Python使用OpenCV和K-Means聚类对毕业照进行图像分割
Jun 11 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/06 咖啡文化
基于PHP静态类的原罪详解
2013/05/06 PHP
php之Memcache学习笔记
2013/06/17 PHP
实用的简单PHP分页集合包括使用方法
2013/10/21 PHP
推荐几个开源的微信开发项目
2014/12/28 PHP
分享3个php获取日历的函数
2015/09/25 PHP
php使用curl详细解析及问题汇总
2016/08/11 PHP
Flexigrid在IE下不显示数据的有效处理方法
2014/09/04 Javascript
2014年最火的Node.JS后端框架推荐
2014/10/27 Javascript
jQuery中使用animate自定义动画的方法
2016/05/29 Javascript
JavaScript 深层克隆对象详解及实例
2016/11/03 Javascript
关于Vue.js一些问题和思考学习笔记(2)
2016/12/02 Javascript
基于nodejs 的多页面爬虫实例代码
2017/05/31 NodeJs
jQuery 实现双击编辑表格功能
2017/06/19 jQuery
微信小程序组件 marquee实例详解
2017/06/23 Javascript
mpvue跳转页面及注意事项
2018/08/03 Javascript
教你使用vue-cli快速构建的小说阅读器
2019/05/13 Javascript
Vue  webpack 项目自动打包压缩成zip文件的方法
2019/07/24 Javascript
React.js组件实现拖拽排序组件功能过程解析
2020/04/27 Javascript
jquery实现简易验证插件封装
2020/09/13 jQuery
Python socket网络编程TCP/IP服务器与客户端通信
2017/01/05 Python
Python Dataframe 指定多列去重、求差集的方法
2018/07/10 Python
安装好Pycharm后如何配置Python解释器简易教程
2019/06/28 Python
Python+pyftpdlib实现局域网文件互传
2020/08/24 Python
使用HTML5里的classList操作CSS类
2016/06/28 HTML / CSS
业务总经理岗位职责
2014/02/03 职场文书
经销商订货会主持词
2014/03/27 职场文书
演讲稿祖国在我心中
2014/05/04 职场文书
给妈妈洗脚活动方案
2014/08/16 职场文书
会计实训报告范文
2014/11/04 职场文书
工作检讨书怎么写
2015/01/23 职场文书
认真学习保证书
2015/02/26 职场文书
反邪教警示教育活动总结
2015/05/09 职场文书
2016年员工政治思想表现评语
2015/12/02 职场文书
蓝天保卫战收官在即 :15行业将开展环保分级评价
2019/07/19 职场文书
Python 如何实现文件自动去重
2021/06/02 Python