pytorch学习笔记-----现有模型的使用和修改

最新推荐文章于 2024-07-12 07:47:18 发布

完◎笑

最新推荐文章于 2024-07-12 07:47:18 发布

阅读量391

点赞数

本文链接：https://blog.csdn.net/qq_43482596/article/details/121503739

版权

PyTorch VGG16 预训练模型模型修改线性层

关键词由CSDN通过智能技术生成

注意事项：

1.很多现有的，比较好的网络模性都在torchvision.models模块下

下面观看完整代码

import torchvision
from torch import nn
#现有模型的使用和修改
vgg16_flase = torchvision.models.vgg16(pretrained=False)#pretrained=false仅仅就是将网络模型在这里进行替换，参数都是随机初始化的
#vgg16_true = torchvision.models.vgg16(pretrained=True,progress=True)#progress=true 会显示下载进度条，pretrained=true时会从网络中下载好预训练权重参数
print(vgg16_flase)

#在vgg16的基础上增加一个线性层，从而对10类别的物体进行分类
vgg16_flase.classifier.add_module("add_linear",nn.Linear(1000,10))#增加了一个线性层
print(vgg16_flase)
#或者直接对最后一个全连接层进行修改
#vgg16_flase.classifier[6] = nn.Linear(4096,10)