搭建CNN卷积神经网络(用pytorch搭建)

手撕卷积神经网络—CNN
卷积:提取特征

池化:压缩特征

heigh X weigh X depth

长度 宽度。深度(也就是特征图个数)

例如输入32x32x3 hxwxc

卷积就是取某个小区域进行特征计算用此处的 特征值x权重特征 进而得到最终数据输出

filter相当于助手可以有一个 但是也可以有多个 可以定义f1 f2 f3…当然得到特征图越丰富越是我们想要的结果但是多个卷积核的容积必须是相同 否则则无法相加。

移动步长 可以大一点也可以小一点用1就可以

卷积核尺寸 可以大一点 也可以小一点 大的话就比较粗糙 小的话提取特征就相对比较精细。3×3最好

边界尺寸 就是可以命名为0

搭建CNN卷积神经网络(用pytorch搭建)

计算方法

搭建CNN卷积神经网络(用pytorch搭建)

池化 (也是特征压缩)

也可以称为下采样 就是选取特征最大的部分就是最大池化Maxpooling和卷积一样就是筛选而已没有矩阵的计算。

基本正常的卷积神经网络都是两次卷积一次池化 进而进行压缩和筛选

全连接层

其实在最后还需要有一个拉长的过程 将立体的图形拉长为特征向量。

搭建CNN卷积神经网络(用pytorch搭建)

只有带计算的才可以称为层 如图就是七层的卷积神经网络。

搭建CNN卷积神经网络(用pytorch搭建)

经典卷积神经网络

1.Alexnet

搭建CNN卷积神经网络(用pytorch搭建)

2.VGG

搭建CNN卷积神经网络(用pytorch搭建)

3.resnet

搭建CNN卷积神经网络(用pytorch搭建)

resnet核心思想就是加一个同等映射如果学习无法使loss下降 那么我们有一个保底的卷积 就直接把上面的拿过来。那么也就称为其为残差网络。

无论将来做什么 做残差网络是一个首选项

因为resnet以出色的表现完全可以证明深度学习卷积的层数是完全可以继续往下堆叠的 尽管他的提升比较少 。

resnet不属于分类 属于一个特征提取的问题。属于一个万能网络

实例

import torch
import torch.nn as nn
from torchvision import datasets
from torchvision import transforms

inputsize = 228
num_classes = 2
num_epoche = 3
batch_size = 64

train_dateset = datasets.MNIST(root='./data',
                               train=True,
                               transform=transforms.ToTensor(),
                               download=True
                               )

test_dateset = datasets.MNIST(root='./data',
                              train=False,
                              transform=transforms.ToTensor(),
                              )

train_loader = torch.utils.data.DataLoader(dataset=train_dateset,
                                           batch_size = batch_size,
                                           shuffle=True
                                           )
test_loader = torch.utils.data.DataLoader(dataset=test_dateset,
                                           batch_size = batch_size,
                                           shuffle=True
                                           )

class CNN(nn.Module):
    def __init__(self):
        super(CNN,self).__init__()

        self.conv1= nn.Sequential(
            nn.Conv2d(
                in_channels=1,
                out_channels=16,
                kernel_size=5,
                stride=1,
                padding=2,
            ),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=2),
        )

        self.conv2 = nn.Sequential(

            nn.Conv2d(16,32,5,1,2),
            nn.ReLU(),
            nn.MaxPool2d(2),
        )

        self.out=nn.Linear(32*7*7,10)

    def forward(self,x):
        x= self.conv1(x)
        x= self.conv2(x)
        x= x.view(x.size(0),-1)
        output = self.out(x)

        return output

def accuracy(predictions,labels):
    pred = torch.max(predictions.data,1)[1]
    rights=pred.eq(labels.data.view_as(pred)).sum()
    return rights,len(labels)

net=CNN()

critertion=nn.CrossEntropyLoss()

optimizer =torch.optim.Adam(net.parameters(),lr=0.001)

for epoch in range(num_epoche):

    train_rights=[]
    for batch_idx,(data,target) in enumerate(train_loader):
        net.train()
        output=net(data)
        loss=critertion(output,target)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        right =accuracy(output,target)
        train_rights.append(right)
        if batch_idx %100==0:
            net.eval()
            val_rights=[]

            for (data,target) in test_loader:
                output=net(data)
                right=accuracy(output,target)
                val_rights.append(right)

            train_r=(sum([tup[0]for tup in train_rights]),sum([tup[1] for tup in train_rights]))
            val_r=(sum([tup[0]for tup in val_rights]),sum([tup[1] for tup in val_rights]))
            print('当前epoch:{}[{}/{}({:.0f}%)]\t损失:{:.6f}\t训练集准确率:{:.2f}\t测试集正确率:{:.2f}%'
                  .format(epoch,batch_idx*batch_size,len(train_loader.dataset),100.*batch_idx/len(train_loader),
                          loss.data,
                          100.*train_r[0].numpy()/train_r[1],
                          100.*val_r[0].numpy()/val_r[1]))

训练结果:

搭建CNN卷积神经网络(用pytorch搭建)
结果相对来说还可以

Original: https://blog.csdn.net/weixin_46900108/article/details/122609067
Author: 学成七步
Title: 搭建CNN卷积神经网络(用pytorch搭建)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/705375/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球