手动实现各类模型结构之分类
文章平均质量分 57
骚火棍
这个作者很懒,什么都没留下…
展开
-
手动搭建ResNet模型(pytorch)
手动搭建ResNet模型(pytorch)一、ResNet模型结构?二、代码示例三、参考链接一、ResNet模型结构?ResNet18、34、50、101、152ResNet18、34对应左下的残差块;ResNet50、101、152对应右下的残差块;二、代码示例import torchvisionimport torchimport torch.nn as nn__all__ = ['ResNet50','ResNet101','ResNet152']def Conv1原创 2021-04-20 13:50:00 · 2347 阅读 · 5 评论 -
手动搭建Inception V1模型(pytorch)
手动搭建Inception V1模型(pytorch)一、Inception V1模型结构二、代码示例三、参考链接一、Inception V1模型结构Inception V1 moduleInception V1完整结构二、代码示例import torchvisionimport torchimport torch.nn as nn# iv1 = torchvision.models.googlenet(pretrained=False)## print (iv1).原创 2021-04-08 10:53:16 · 720 阅读 · 0 评论 -
手动搭建VGG16模型(pytorch)
文章目录一、VGG16模型结构二、代码示例三、参考链接一、VGG16模型结构二、代码示例import torch.nn as nnfrom torchvision import models#vgg16 = models.vgg16_bn(pretrained=False)class VGG16(nn.Module): def __init__(self): super(VGG16,self).__init__() self.features = n原创 2021-04-02 14:37:39 · 1044 阅读 · 0 评论