【经典网络结构实现】
MING.MING
学生一枚
展开
-
【经典网络结构实现】VGG-16
经典网络结构实现之VGGVGG网络的简单实现VGG网络的简单实现由于VGG网络在电脑上跑不起来,等以后有机会在GPU服务器上跑。import torch.nn as nnimport torchclass VGG(nn.Module): def __init__(self, num_classes): super().__init__() # 输出的类别个数 self.num_classes = num_classes #原创 2020-10-12 22:41:23 · 314 阅读 · 0 评论 -
【经典网络结构实现】AlexNet
经典网络结构实现之AlexNetPytorch实现手写数字识别AlexNet网络的简单实现Pytorch实现手写数字识别AlexNet网络的简单实现import torchimport torch.nn as nnimport torch.nn.functional as Fclass AlexNet(nn.Module): def __init__(self): super(AlexNet,self).__init__() self.conv1 = n原创 2020-10-12 20:46:34 · 391 阅读 · 0 评论 -
【经典网络结构实现】LeNet-5
经典网络结构实现之LeNet-5Pytorch实现手写数字识别LeNet-5网络的简单实现LeNet-5应用于手写数字识别Pytorch实现手写数字识别LeNet-5网络的简单实现# 导入相应的包import torch.nn as nn# 搭建LeNet-5class LeNet5(nn.Module): def __init__(self): super(LeNet5,self).__init__() # 搭建卷积池化卷积池化层原创 2020-10-10 11:23:55 · 442 阅读 · 0 评论