【pytorch】简单的CNN神经网络

最新推荐文章于 2024-07-15 21:32:29 发布

haohulala

最新推荐文章于 2024-07-15 21:32:29 发布

阅读量1.3k

点赞数

分类专栏：计算机视觉 python

本文链接：https://blog.csdn.net/haohulala/article/details/105120842

版权

cnn神经网络主要由卷积层，池化层和全连接层构成，下面这段代码就是一个简单的cnn神经网络模型

#coding=utf-8
"""
一个简单的CNN网络模型
"""

#明天写

import torch
from torch import nn

class simpleCNN(nn.Module):
    def __init__(self):
        super(simpleCNN, self).__init__()
        #第一层
        layer1 = nn.Sequential()
        layer1.add_module("conv1", nn.Conv2d(3, 32, 3, 1, padding=1))
        layer1.add_module("relu1", nn.ReLU(True))
        layer1.add_module("pool1", nn.MaxPool2d(2, 2))
        self.layer1 = layer1
        #第二层
        layer2 = nn.Sequential()
        layer2.add_module("conv2", nn.Conv2d(32, 64, 3, 1, padding=1))
        layer2.add_module("relu2", nn.ReLU(True))
        layer2.add_module("pool2", nn.MaxPool2d(2, 2))
        self.layer2 = layer2
        #第三层
        layer3 = nn.Sequential()
        layer3.add_module("conv3", nn.Conv2d(64, 128, 3, 1, paddi