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 import自定义模块方法
Feb 12 Python
JSON Web Tokens的实现原理
Apr 02 Python
解决Scrapy安装错误:Microsoft Visual C++ 14.0 is required...
Oct 01 Python
Python生成数字图片代码分享
Oct 31 Python
numpy自动生成数组详解
Dec 15 Python
python删除某个字符
Mar 19 Python
python操作excel的方法(xlsxwriter包的使用)
Jun 11 Python
Python 对输入的数字进行排序的方法
Jun 23 Python
Python使用matplotlib绘制Logistic曲线操作示例
Nov 28 Python
Django 解决新建表删除后无法重新创建等问题
May 21 Python
用Python 爬取猫眼电影数据分析《无名之辈》
Jul 24 Python
Python面向对象多态实现原理及代码实例
Sep 16 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
用php来检测proxy
2006/10/09 PHP
一个捕获函数输出的函数
2007/02/14 PHP
php explode函数实例代码
2012/02/27 PHP
PHP批量生成静态HTML的简单原理和方法
2014/04/20 PHP
PHP实现过滤掉非汉字字符只保留中文字符
2015/06/04 PHP
Save a File Using a File Save Dialog Box
2007/06/18 Javascript
封装好的省市地区联动控件附下载
2007/08/13 Javascript
Javascript学习笔记5 类和对象
2010/01/11 Javascript
jQuery EasyUI API 中文文档 可调整尺寸
2011/09/29 Javascript
JS for循环中i++ 和 ++i的区别介绍
2016/07/20 Javascript
javascript实现右下角广告框效果
2017/02/01 Javascript
使用jQuery的load方法设计动态加载及解决被加载页面js失效问题
2017/03/01 Javascript
代码实例ajax实现点击加载更多数据图片
2018/10/12 Javascript
Node.js 多线程完全指南总结
2019/03/27 Javascript
开发中常用的25个JavaScript单行代码(小结)
2019/06/28 Javascript
vue video和vue-video-player实现视频铺满教程
2020/10/30 Javascript
Python实现拼接多张图片的方法
2014/12/01 Python
Python 递归函数详解及实例
2016/12/27 Python
Python内建模块struct实例详解
2018/02/02 Python
python3解析库BeautifulSoup4的安装配置与基本用法
2018/06/26 Python
基于python-opencv3的图像显示和保存操作
2019/06/27 Python
关于pytorch多GPU训练实例与性能对比分析
2019/08/19 Python
Python学习笔记之字符串和字符串方法实例详解
2019/08/22 Python
Python操作SQLite/MySQL/LMDB数据库的方法
2019/11/07 Python
Python加密模块的hashlib,hmac模块使用解析
2020/01/02 Python
通过 Python 和 OpenCV 实现目标数量监控
2020/01/05 Python
我有一个char * 型指针正巧指向一些int 型变量, 我想跳过它们。 为什么如下的代码((int *)p)++; 不行?
2013/05/09 面试题
如何写一个自定义标签
2012/12/28 面试题
劳资专员岗位职责
2013/12/27 职场文书
高三体育教学反思
2014/01/29 职场文书
实习指导老师评语
2014/04/26 职场文书
电子专业求职信
2014/06/19 职场文书
献爱心标语
2014/06/21 职场文书
安全生产工作汇报
2014/10/28 职场文书
公司2015年终工作总结
2015/05/26 职场文书
龙猫观后感
2015/06/09 职场文书