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脚本
Feb 10 Python
Python中.py文件打包成exe可执行文件详解
Mar 22 Python
centos 安装python3.6环境并配置虚拟环境的详细教程
Feb 22 Python
Python实现将多个空格换为一个空格.md的方法
Dec 20 Python
对python调用RPC接口的实例详解
Jan 03 Python
python使用PIL模块获取图片像素点的方法
Jan 08 Python
Python实现插入排序和选择排序的方法
May 12 Python
Python中import导入不同目录的模块方法详解
Feb 18 Python
Python生成器实现简单"生产者消费者"模型代码实例
Mar 27 Python
浅谈Python协程
Jun 17 Python
使用python把xmind转换成excel测试用例的实现代码
Oct 12 Python
Python数据结构之队列详解
Mar 21 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
一个很方便的 XML 类!!原创的噢
2006/10/09 PHP
用PHP读注册表
2006/10/09 PHP
php 伪造本地文件包含漏洞的代码
2011/11/03 PHP
基于PHP常用字符串的总结(待续)
2013/06/07 PHP
php-fpm配置详解
2014/02/12 PHP
PHP的pcntl多进程用法实例
2015/03/19 PHP
Symfony2框架学习笔记之HTTP Cache用法详解
2016/03/18 PHP
PHP文件系统管理(实例讲解)
2017/09/19 PHP
js实现鼠标拖动图片并兼容IE/FF火狐/谷歌等主流浏览器
2013/06/06 Javascript
JavaScript前端开发之实现二进制读写操作
2015/11/04 Javascript
javascript跑马灯抽奖实例讲解
2020/04/17 Javascript
工作中比较实用的JavaScript验证和数据处理的干货(经典)
2016/08/03 Javascript
RequireJS 依赖关系的实例(推荐)
2017/01/21 Javascript
node+express+ejs使用模版引擎做的一个示例demo
2017/09/18 Javascript
微信小程序实时聊天WebSocket
2018/07/05 Javascript
js拖动滑块和点击水波纹效果实例代码
2018/10/16 Javascript
vuex实现及简略解析(小结)
2019/03/01 Javascript
layui 上传文件_批量导入数据UI的方法
2019/09/23 Javascript
微信小程序学习总结(五)常见问题实例小结
2020/06/04 Javascript
vue项目使用$router.go(-1)返回时刷新原来的界面操作
2020/07/26 Javascript
[00:30]明星选手化身超级英雄!2018DOTA2亚洲邀请赛全明星赛来临!
2018/04/06 DOTA
Python中实现两个字典(dict)合并的方法
2014/09/23 Python
用Python编写一个每天都在系统下新建一个文件夹的脚本
2015/05/04 Python
python基础入门学习笔记(Python环境搭建)
2016/01/13 Python
Python实现求解一元二次方程的方法示例
2018/06/20 Python
浅析CSS3 中的 transition,transform,translate之间区别和作用
2020/03/26 HTML / CSS
阿玛瑞酒店中文官方网站:Amari.com
2018/02/13 全球购物
一道输出判断型Java面试题
2014/10/01 面试题
法警的竞聘演讲稿
2014/01/02 职场文书
建筑总经理岗位职责
2014/02/02 职场文书
幼儿园保教管理制度
2014/02/03 职场文书
会计电算化实训报告
2014/11/04 职场文书
SQL写法--行行比较
2021/08/23 SQL Server
使用pipenv管理python虚拟环境的全过程
2021/09/25 Python
vue-cli3.0修改打包后的文件名和文件地址,打包后本地运行报错解决
2022/04/06 Vue.js
Python序列化模块JSON与Pickle
2022/06/05 Python