pytorch 更改预训练模型网络结构的方法
作者：wayne980 发布时间：2022-04-02 04:41:19
标签：pytorch,预训练,模型,网络结构
一个继承nn.module的model它包含一个叫做children()的函数，这个函数可以用来提取出model每一层的网络结构，在此基础上进行修改即可，修改方法如下(去除后两层)：
resnet_layer = nn.Sequential(*list(model.children())[:-2])
那么，接下来就可以构建我们的网络了：
class Net(nn.Module):
def __init__(self , model):
super(Net, self).__init__()
#取掉model的后两层
self.resnet_layer = nn.Sequential(*list(model.children())[:-2])
self.transion_layer = nn.ConvTranspose2d(2048, 2048, kernel_size=14, stride=3)
self.pool_layer = nn.MaxPool2d(32)
self.Linear_layer = nn.Linear(2048, 8)
def forward(self, x):
x = self.resnet_layer(x)
x = self.transion_layer(x)
x = self.pool_layer(x)
x = x.view(x.size(0), -1)
x = self.Linear_layer(x)
return x
最后，构建一个对象，并加载resnet预训练的参数就可以啦～
resnet = models.resnet50(pretrained=True)
model = Net(resnet)
来源：https://blog.csdn.net/wayne980/article/details/84026939
0
投稿猜你喜欢
- 这是fenye.asp具体代码：以下是引用片段：<％Const maxPagesize=1000 '设置分
- DataSet是tensorflow 1.3版本推出的一个high-level的api，在1.3版本还只是处于测试阶段，1.4版本已经正式推
数学是优美的. 听上去有点奇怪? 当我第一次开始设计的时候，我确信如此。数学如此刻板乏味。你可能会惊讶的发现，最美观的设计，艺术作品，物体，- 本文实例讲述了Python实现链表反转的方法。分享给大家供大家参考，具体如下：Python实现链表反转链表反转（while迭代实现）：链表的
- 本文实例为大家分享了python实现简单颜色识别程序的具体代码，供大家参考，具体内容如下import numpy as npimport c
- 这系列文章将介绍一下Oracle的基础知识，将会使用docker下运行的oracle11g的精简版进行说明。这篇文章介绍一下oracle的版
- 关于电子邮件 大学之前，基本不用邮箱，所以基本感觉不到它的存在，也不知道有什么用；然而大学之后，随着认识的人越来越多，知识越来越广
- 有人把Go比作21世纪的C语言，第一是因为Go语言设计简单，第二，21世纪最重要的就是并行程序设计，而Go从语言层面就支持了并行。gorou
- 导语还记得那些年，我们玩过的Windows小游戏吗？说起Windows自带的游戏，相信许多80、90后的朋友都不陌生。在很早的那个游戏贫瘠的
- 行业首页改版的缘故，为了让我们设计师可以更好的了解需求、了解我们的用户，和部门的用研童鞋一起讨论决定使用电话来进行用户访谈，以此来了解用户。
- 讲这个方法之前，我们应该先了解下插入节点时浏览器会做什么。在浏览器中，我们一旦把节点添加到document.body（或者其他节点）中，页面
- 前言我们都知道 Node.js 是以单线程的模式运行的，但它使用的是事件驱动来处理并发，这样有助于我们在多核 cpu 的系统上创建多个子进程
- pycharm简介PyCharm是一种Python IDE，带有一整套可以帮助用户在使用Python语言开发时提高其效率的工具，比如调试、语
- 引言列表、字典：可变序列，可以执行增删改排序等字典：无序的一、字典的创建#使用{}创建scores = {'张三':100
- 不加(0)的用法:set rs=conn.execute(sql)'将这个结果赋给rs这时要读取这个记录集第一个字段的数据就用rs(
- CTE（Common Table Expressions）是从SQL Server 2005以后版本才有的。指定的临时命名结果集，这些结果集
- 在写Python的时候经常会遇到时间格式的问题，首先就是最近用到的时间戳（timestamp）和时间字符串之间的转换。所谓时间戳，就是从 1
- select a.f_username from ( SELECT /*+parallel(gu,4)*/distinct gu.f_use
- 如何把程序打包为whl首先需要一个库：setuptools如果是conda环境的话，这个包是自带的，不需要另外安装。首先把需要打包的py文件
- python实现四舍五入""" 四舍五入 :param
