ZIPLeNet, AlexNet, VGG-16, SENet.zip 871.21KB

2403_88021836

资源文件列表:

LeNet, AlexNet, VGG-16, SENet.zip 大约有8个文件
  1. AlexNet.py 1.5KB
  2. AlexNet_2.png 391.04KB
  3. lenet.png 320.05KB
  4. lenet.py 1.2KB
  5. SEBlock.png 40.57KB
  6. SEBlock.py 664B
  7. VGG-16.png 155.11KB
  8. vgg16.py 1.82KB

资源介绍:

LeNet, AlexNet, VGG-16, SENet.zip
import torch import torch.nn as nn class VGG16(nn.Module): def __init__(self): super(VGG16, self).__init__() self.features = nn.Sequential( nn.Conv2d(3, 64, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(64, 64, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2d(2, 2), nn.Conv2d(64, 128, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(128, 128, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2d(2, 2), nn.Conv2d(128, 256, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(256, 256, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(256, 256, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2d(2, 2), nn.Conv2d(256, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(512, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(512, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2d(2, 2), nn.Conv2d(512, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(512, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.Conv2d(512, 512, 3, 1, 1), nn.ReLU(inplace=True), nn.MaxPool2d(2, 2), ) self.fc = nn.Sequential( nn.Linear(7*7*512, 4096), nn.ReLU(inplace=True), nn.Linear(4096, 4096), nn.ReLU(inplace=True), nn.Linear(4096, 1000), ) def forward(self, x): x = self.features(x) x = x.reshape(x.shape[0], -1) x = self.fc(x) return x x = torch.randn(4, 3, 224, 224).cuda() model = VGG16().cuda() output = model(x) print(output.shape)
100+评论
captcha
    相关资源
    类型标题大小时间
    ZIPSSM网上旅游信息管理系统（附源码+数据库）0697531.13MB6月前
    ZIP魔兽自制工具包，替换主界面ui7.31MB6月前
    ZIPqt打包所需相关基础依赖13.21MB6月前
    ZIPinstantclient-basic-windows.x64-19.20.0.0.0dbru.zip76.76MB6月前
    ZIP计算机操作系统模拟点餐系统课程设计67.75MB6月前
    ZIPLinux学习笔记14-EPIT 定时器实验例程367.2KB6月前
    ZIPQT QPainter C++测试源代码3.46KB6月前
    ZIPSTM32F103C8T6小系统板 单片机 核心板 STM32开发板学习板2.87MB6月前