卷积神经网络--猫狗系列之构建模型【ResNet50】

news2024/9/20 0:56:49

在上一期:卷积神经网络--猫狗系列之下载、导入数据集,如果测试成功就说明对数据的预处理工作已经完成,接下来就是构建模型阶段了:

据说建立一个神经网络模型比较简单,只要了解了各层的含义、不同层之间参数的传递等等,那么一个完整的网络模型就可以被容易地构建出来。(这对于我这种初学的同学来说嘎嘎困难哈哈)

不扯了,加载一个预训练模型---ResNet50:

借助torchvision库,能够很容易的获得一组已经训练好的模型,这些模型大多数接收一个称为pretrained的参数,当这个参数为True时,它会下载为ImageNet分类问题调整好的权重。

network1=models.resnet50(weights=models.ResNet50_Weights.DEFAULT)

紧接着,我们需要冻结所有层,所有的权重不会随着训练而更新:

for param in network1.parameters():    param.requires_grad=False

然后!由于这个预训练模型不是专门针对这样猫猫狗狗的二分类问题,所以,我们需要将最后一层的输出特征从1000改为2。(默认是1000类)

首先,我们需要知道最后一层的名字,通过network1查看一下,告诉大家怎么操作:

将network1输在pycharm中,把光标放在这一行,然后快捷键【shift+alt+E】(也可以鼠标选中然后右键专门运行这一行,这其实就是pycharm的交互模式)

得到这样的界面:

【可以知道最后一层是一个全连接层,名为fc。其他的东西以后有时间再分析,就是一些卷积、池化以及激活操作】

所以,我们就要将最后一层替换为输出特征为2的全连接层:

import torch.nn as nnnetwork1.fc=nn.Linear(2048,2)

此时,该层为新的层,所以它的requires_grad=True,这样整个网络仅有这一层可以更新权重。

打印更新后的网络:

该猫猫狗狗的模型构建代码总结:

from torchvision import modelsimport torch.nn as nnimport torch.optim as optim#网络搭建network1=models.resnet50(weights=models.ResNet50_Weights.DEFAULT)for param in network1.parameters():    param.requires_grad=Falsenetwork1.fc=nn.Linear(2048,2)

利用已经训练好的模型主要目的是它能够提取出非常好的特征,最后一层接收前面层提取的特征,然后误差反向传播,仅更新这一层的权重,不断迭代。

【此猫狗系列会继续更】

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/709340.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

leetcode 1232. 缀点成线

题目描述解题思路执行结果 leetcode 1232. 缀点成线 题目描述 缀点成线 给定一个数组 coordinates ,其中 coordinates[i] [x, y] , [x, y] 表示横坐标为 x、纵坐标为 y 的点。请你来判断,这些点是否在该坐标系中属于同一条直线上。 示例 1&a…

进程的调度常用算法

目录 先来先服务(FCFS)调度算法 短作业优先(SJF)的调度算法 基于时间片的轮转调度(RR)算法 先来先服务(FCFS)调度算法 系统将按照作业到达的先后次序来进行作业调度,或…

RT-Thread 5.0.1 qemu-virt64-aarch64 解决网络问题

参考文章 qemu 源码编译 qemu-system-aarch64 的方法 RT-Thread 5.0.1 qemu-virt64-aarch64 解决编译问题 前言 最近需要使用 RT-Thread qemu-virt64-aarch64,验证 aarch64 平台,也就是 ARM64 平台的一些网络功能,需要开启 qemu-virt64-aa…

UE4/5用贴图和GeneratedDynamicMeshActor曲面细分与贴图位移制作模型

目录 制作逻辑: ​编辑 曲面细分函数: 添加贴图逻辑: 代码: 制作逻辑: 在之前的文章中,我们使用了网格细分,而这一次我们将使用曲面细分函数,使用方法和之前是一样的&#xff1a…

高精度定位|RTK定位模块常见应用领域_厘米级室外定位解决方案

在室外场景,北斗、GPS等GNSS定位技术在持续的演变,精度越来越高,应用面也越来越广。随着新基建热潮的到来,借助5G新基建,无人驾驶、自动驾驶等技术正在逐步完善,对于定位的需求已经不仅仅只是粗略的轨迹&am…

uniapp-ios打包步骤

前置条件:已申请Apple ID并注册Apple Developer Program 一、登录苹果开发者官网 登录:https://developer.apple.com/ 点击账户(Account)进行登录,登录成功出现如下页面 二、创建证书、标识符、描述文件等 点击…

vue3基本指令使用

<script setup lang"ts"> import {ref} from vue //响应式数据 const num: number 1 const arr1: number[] [1, 2, 3, 4, 5] const str: string "我是一段文字" const htmlstr: string <section style"color:red">我是一个secti…

微服务-基于Docker安装Sentinel

目录 1、拉取Sentinel镜像 2、构建Sentinel容器 3、访问Sentinel 1、拉取Sentinel镜像 代码&#xff1a; docker pull bladex/sentinel-dashboard:1.8.0 实例&#xff1a; rootlocalhost howlong]# docker pull bladex/sentinel-dashboard:1.8.0 1.8.0: Pulling from blade…

集合及Collection集合

1&#xff1a;集合特点&#xff1a; 集合的大小不确定&#xff0c;启动后可以动态变化&#xff0c;类型也可以选择不固定。集合更像气球集合非常适合做元素的增删操作注意&#xff1a;集合只能存储引用类型数据&#xff0c;如果要存储基本类型数据可以选用包装类。 2&#xf…

2.FreeRTOS系统任务调度简介及任务状态

目录 一、基础知识 1、FreeRTOS 任务状态 (1)运行态 (2)就绪态 (3) 阻塞态 (4) 挂起态 二、任务调度简介 1.抢占式调度 2.时间片调度 3.协程式调度 一、基础知识 1、FreeRTOS 任务状态 FreeRTOS 中任务存在四种任务状态&#xff0c;分别为运行态、就绪态、阻塞态和挂…

Word模板替换,并转PDF格式输出,Linux服务器乱码问题解决

Poi-tl参考文档地址&#xff1a;http://deepoove.com/poi-tl/1.8.x/#hack-loop-table word模板替换&#xff0c;转pdf 1. 依赖引入&#xff1a;2. word模板配置&#xff1a;3. 示例demo:4 . 效果图5. 本地测试没问题&#xff0c;上Linux服务乱码&#xff0c;出现小方框 1. 依赖…

Python———运行环境搭建

不管用什么工具开发 Python 程序&#xff0c;都必须安装 Python 的运行环境。 目前最常用的是Windows 、 Linux 平台。这里 我们以Windows10为主讲解。 其实编程和平台关系不大。大家也可以使用Linux、Mac。 Windows 平台下 Python 环境搭建 第一步&#xff1a;进入 python 官…

python绘制热力图,数据来源pandas.dataframe

通过pandas.dataframe绘制热力图&#xff0c;并标出颜色最深即z轴数据最大的点。 import matplotlib.pyplot as plt import pandas as pd import numpy as npclass Heatmap:def __init__(self, data, marker*, marker_colorwhite, marker_size10):self.data dataself.size l…

排序的概念,插入排序,希尔排序,选择排序

文章目录 排序的概念排序的概念&#xff1a;排序运用常见的排序与时间复杂度 插入排序希尔排序希尔排序的概念希尔排序总结 选择排序概念 排序的概念 排序的概念&#xff1a; 排序&#xff1a;所谓排序&#xff0c;就是使一串记录&#xff0c;按照其中的某个或某些关键字的大小…

【复习《剑指Offer》13-17题】【每天40分钟,我们一起用50天刷完 (剑指Offer)】第十三天 7/50

专注 效率 记忆 预习 笔记 复习 做题 欢迎观看我的博客&#xff0c;如有问题交流&#xff0c;欢迎评论区留言&#xff0c;一定尽快回复&#xff01;&#xff08;大家可以去看我的专栏&#xff0c;是所有文章的目录&#xff09;   文章字体风格&#xff1a; 红色文字表示&#…

什么是精益创业?精益创业的核心方法有哪些?

《经理人参阅&#xff1a;创业管理学》是一本非常有价值的创业类书籍&#xff0c;为创业者提供了许多实用的方法和经验&#xff0c;特别是关于精益创业方面的内容。精益创业是一种以快速验证假设为核心的创业方法&#xff0c;它可以帮助创业者在最短的时间内找到最有效的商业模…

Kafka跨集群备份解决方案MirrorMaker

文章目录 1 什么是 MirrorMaker&#xff1f;2 运行 MirrorMakerMirrorMaker 配置实例第 1 步&#xff1a;启动两套 Kafka 集群第 2 步&#xff1a;启动 MirrorMaker 工具第 3 步&#xff1a;验证消息是否拷贝成功 3 其他跨集群镜像方案4 小结 一般情况下&#xff0c;我们会使用…

Win2008下使用IIS+URL重写+Server Farms在同一台服务器实现混合 多域名网站与多个负载平衡集群网站 共存,配置及域名绑定笔记

因公司要在现有云服务器增加负载平衡集群网站&#xff0c;但要保证现有在用网站不动&#xff0c;操作系统为Win2008。 以前没有配置过IIS的负载平衡&#xff0c;只能百度一下&#xff0c;实现也简单&#xff1a;只要安装URL重写Server Farms就能实现负载平衡集群网站。于是根据…

IMX6ULL系统移植篇-uboot的烧写与运行

一. uboot 配置与编译 本文来学习 uboot的烧写&#xff0c;承接之前 uboot的配置与编译篇&#xff0c;博文如下&#xff1a; IMX6ULL系统移植篇-uboot的配置与编译_凌肖战的博客-CSDN博客 二. uboot的烧写 uboot 编程成功后&#xff0c;会在 uboot的根目录下生成 u-boot.b…

CentOS Linux的最佳替代方案(三)_银河麒麟Kylin Linux Advanced Server V10 SP2 for X86安装教程

文章目录 CentOS Linux的最佳替代方案&#xff08;三&#xff09;_银河麒麟Kylin Linux Advanced Server V10 SP2 for X86安装教程1 银河麒麟Kylin Linux Advanced Server V10简介1.1 同源构建1.2 自主CPU平台深入优化1.3 虚拟化及云原生支持1.4 高可用性1.5 可管理性1.6 内生本…