卷积网络项目:实现识别鲜花四分类对比LeNet5、VGG16、ResNet18、ResNet34分类网络

卷积四分类项目

Gitee传送门

分类目标选取

鲜花

  • 杏花 apricot_blossom
  • 桃花 peach_blossom
  • 梨花 pear_blossom
  • 梅花 plum_blossom

模型选择

卷积

  • LeNet5
  • VGG16
  • ResNet18
  • ResNet34

以图搜图

获取相似度前10的搜图结果

数据清洗

鲜花四分类

删除非图片文件

image.png

删除重复图片

image.png
image.png
image.png
image.png
image.png

整理数据集

鲜花四分类

每种类别数据:训练500、测试50、预测10
总训练集:2500
总测试集:250
总预测集:40

训练模型

报错

ValueError: num_samples should be a positive integer value, but got num_samples=0

换了电脑后,数据集的存储位置不同,更换路径后解决

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

原因:错误内容就在类型不匹配,根据报错内容可以看出Input type为torch.FloatTensor(CPU数据类型),而weight type(即网络权重参数这些)为torch.cuda.FloatTensor(GPU数据类型)
方案:将输入类型转变为GPU类型
输入数据和网络都切换到cuda,但问题仍存在
检查网络,修改模型隐藏层初始化方式后,解决了问题

鲜花

v1:LeNet5:bn

输出4分类

image.png
image.png
image.png

v2:VGG16:bn

数据太差,提前中断了训练
image.png
image.png
image.png

v3:ResNet18:bn

输出4分类
f4_v3:32x32

image.png
image.png
image.png
准确率仍上不去,预估增大迭代次数,准确率能慢慢提升

f4_v3.3:224x224

image.png
image.png
过拟合前最佳:
image.png
测试数据出现过拟合现象,考虑减小数据大小

f4_v3.4:112x112

image.png
image.png
过拟合前最佳:
image.png
再次出现过拟合,提前中断了训练

f4_v3.5:56x56

image.png
image.png
过拟合前最佳:
image.png
再次出现过拟合,提前中断了训练
结论:图片缩放大小无法解决过拟合问题

f4_v3.6:32x32,减4个残差块

image.png
image.png
测试集过拟合前
image.png
最佳
image.png

f4_v3.6:32x32,减4个残差块,transforms减Norm

image.png
image.png
测试集过拟合前
image.png
最佳
image.png

v4:ResNet34:bn

输出4分类

image.png
image.png
image.png
预估:增加迭代次数,可能能缓慢提升准确率

以图搜图

报错

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

image.png
LeNet5模型能正常运行,ResNet18和ResNet34模型报错
image.png正常运行
image.png报错
image.png报错
原因:模型中含有nn.BatchNorm层,训练时需要batch_size大于1,来计算当前batch的running mean and std。自定义数据数量除以batch_size后刚好余1,就发生了上述报错
方案1:在pytorch的Dataloader中设置drop_last=True即可,这样会忽略最后一个batch
尝试在数据集增加drop_last=True,再次训练,尚未解决这个问题
方案2:在添加数据前增加model.eval()
代码原本就有这个语句,仍存在这个问题
方案3:修改训练模型数据预处理中Resize大小32–>224,问题解决

搜图结果

v1:LeNet5:bn

image.png
image.png
没有一个是正确分类

v3:ResNet18:bn

f4_v3:32x32

image.png
image.png
没有一个是正确分类,且相似度差距很大

f4_v3.3:224x224

过拟合前最佳:
image.png
image.png
预测最佳类别中top10图片和原图类别相同,但与top1图片与原图相似度不是0
原因:检索库图片根据特征处理、带参数的模型生成对应的特征文件,更换特征处理方式或参数后,生成的特征文件有所不同,所以计算相似度,哪怕是原图也不为0
解决方案:更换特征处理方式或参数后,重新初始化特征文件,再进行预测,解决了这个问题

v4:ResNet34:bn

f4_v4:32x32

image.png
image.png
没有一个是正确分类,且相似度差距很大

f4_v4.3:224x224

image.png
image.png
出现了一个正确分类,由于时间问题,v4.3版没有完成足够的训练,不确定迭代后的数据能否达到预期效果

特征处理

feat_v3.3.0:tensor

image.png

feat_v3.3.1:tensor+Resize56

image.png

feat_v3.3.2:tensor+Resize56+Norm

image.png

feat_v3.3.3:tensor+crop+Resize56

image.png

feat_v3.3.4:tensor+Resize+crop+Resize56

image.png

feat_v3.3.5:tensor+Resize+crop+Resize224

image.png

feat_v3.3.6:tensor+Resize+crop+Resize112

image.png

feat_v3.3.7:tensor+Resize+crop+Resize32

image.png

总结

feat_v3.3.4.txt版本的特征处理效果最好
特征处理方式:tensor+Resize600+crop400+Resize56

搜图效果

相似度前10的结果,top1是原图,6张正确类别花,3张错误类别花
原因:这四类花本身比较相似,不便于学习;也可能是数据量不够多,训练效果不够好;也可能迭代的次数不够多,模型没有训练到足够好的效果
20240227002214_rec_.gif

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

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

相关文章

【JavaWeb】前后端分离SpringBoot项目快速排错指南

1 发起业务请求 打开浏览器开发者工具,同时显示网络(Internet)和控制台(console) 接着,清空控制台和网络的内容,如下图 然后,点击你的业务按钮,发起请求。 首先看控制台…

【C#进阶】简单数据结构类

简单数据结构类 文章目录 1、Arraylist1、ArrayList的本质2、声明3、增删查改4、装箱拆箱思考 背包售卖 2、Stack1、Stack的本质2、声明3、增取查改4、遍历思考 计算一个数的二进制 3、Queue1、Queue的本质2、声明3、增取查改4、遍历思考 每隔一段时间打印一条消息 4、Hashtab…

Json类型转换异常

class com.alibaba.fastjson.JSONObject cannot be cast to class com.xxx.xxOrder (com.alibaba.fastjson.JSONObject and com.xxx.xxOrder are in unnamed module of loader org.springframework.boot.loader.LaunchedURLClassLoader 25b5c5e3 大概率是因为 JSON.parseObjec…

Yocto学习笔记(三)配置Busybox

1 Busybox介绍 BusyBox 是一个开源的、轻量级的 Unix 工具箱,它集成了许多标准 Unix 工具的精简版本。它被设计用来运行在资源受限的环境中,例如嵌入式系统或者启动盘。BusyBox 将多个标准 Unix 实用工具整合到一个可执行文件中,这些工具包括…

如何加密保护U盘?U盘加密方法盘点

U盘是目前最常用的移动存储设备,可以帮助我们存储大量数据。而为了保护数据安全,我们需要加密保护U盘。下面我们就来盘点一下U盘加密的方法。 BitLocker加密 BitLocker是Windows的一种磁盘保护工具,通过加密整个磁盘来保护数据,同…

基于Springboot的校园疫情防控信息管理系统(有报告)。Javaee项目,springboot项目。

演示视频: 基于Springboot的校园疫情防控信息管理系统(有报告)。Javaee项目,springboot项目。 项目介绍: 采用M(model)V(view)C(controller)三层…

react生命周期及用法

React生命周期是React组件在挂载、更新和卸载过程中的一系列方法的集合。通过生命周期方法,我们可以在不同阶段执行特定的操作,例如初始化组件、更新状态、渲染UI等。 React生命周期分为三个阶段:挂载阶段、更新阶段和卸载阶段。以下是React…

Vscode编辑器 js 输入log自动补全

最近换了新电脑,新下载了Vscode,记录一下设置项。 Vscode 版本 想要的效果 js文件中输入log(点击tab键),自动补全为 console.log() Vscode 文件》首选项》设置 搜索:snippets Emmet: Show Suggestions…

暗区突围TWITCH掉宝领测试资格后,steam激活显示是无效激活码

自《暗区突围》测试启动以来,吸引了大量玩家关注,特别是通过在Twitch平台上观看直播即可获得测试资格的活动,更是掀起了热潮。然而,部分玩家在成功获得激活码后,在Steam平台激活时遭遇了“无效激活码”的问题。本文将提…

nginx 配置域名SSL证书HTTPS服务

下载 上传根目录 /home/wwwroot/xx.com/ssl 从新执行 添加域名命令 选择添加SSL SSL Certificate file: 填写 完整目录 PEM文件地址 SSL Certificate Key file:填写 完整目录 key文件地址

python 对矩阵与矩阵之间对应位置的元素,做softmax操作,代码实战

1.对矩阵中对应位置的元素,做softmax 对于一个向量,softmax函数会对其中每一个元素进行指数运算,然后除以所有元素指数和的结果。当将其应用到多个矩阵的相应位置上时,我们实际上是在对每个位置的一组数(从各个矩阵的同…

45.WEB渗透测试-信息收集-域名、指纹收集(7)

免责声明:内容仅供学习参考,请合法利用知识,禁止进行违法犯罪活动! 内容参考于: 易锦网校会员专享课 上一个内容:计算机王-CSDN博客 WEB指纹:Web指纹也叫web应用指纹。由于所使用的工具、技术…

PHP 自提时间

前端: 后台设置: 代码: public function getBusinessHour(){// 需求单门店$data (new StoreModel())->limit(1)->select()->toArray();$days explode(,, $data[0][shop_hours]);$businessHours $days[1];// 使用 explode 分割字符串,获取开始和结束时…

ListView[new]

activity_main.xml <?xml version"1.0" encoding"utf-8"?> <LinearLayout xmlns:android"http://schemas.android.com/apk/res/android"android:layout_height"match_parent"android:layout_width"match_parent"…

服务器租用中包含哪些?

服务器在目前的互联网社会中起着十分重要的作用&#xff0c;服务器是网络业务部署的重要载体&#xff0c;越来越多的企业和用户选择租用服务器业务来减少整体的成本&#xff0c;那么服务器租用中主要都包含哪些呢&#xff1f;接下来小编就来介绍一些其中的内容吧&#xff01; 首…

nginx 发布静态资源

一. nginx 发布静态资源 在nginx中nginx.conf配置文件中添加内容如下&#xff1a; server {listen 90;server_name localhost;# 配置静态资源文件&#xff0c;就可以访问了location / {root /home/fooie-shop;index index.html;}# 配置音频和图片资源location /imoo…

企业内部文化社区究竟有哪些好处?

首先&#xff0c;我们来了解下&#xff0c;企业内耗是什么? 在企业文化管理中&#xff0c;内耗是一个常见的问题&#xff0c;它会影响企业的团队协作、执行效率和绩效表现。在2023《哈佛商业评论》中国年会上&#xff0c;北大汇丰商学院管理实践教授陈玮分享了他对组织管理的…

冯喜运:5.14今日黄金原油涨跌走势分析操作建议

【黄金消息面分析】&#xff1a;本周黄金市场将迎来关键的美国通胀数据&#xff0c;包括周二的生产者价格指数(PPI)和周三的消费者物价指数(CPI)。这些数据对美联储的政策路径至关重要&#xff0c;可能会影响市场对利率调整的预期。目前&#xff0c;现货黄金价格小幅上涨&#…

使用java.io库序列化Java对象

在我们使用诸如Redis这类缓存系统时&#xff0c;我们往往会存在如下需求&#xff1a;将Java对象保存到Redis缓存中&#xff0c;然后在其他机器上还原回来。 Json方案 我们可以引入Json库等方式&#xff0c;将Java对象序列化为Json字符串来实现这个目的&#xff0c;但是这样的…

《四》系统模块整体功能关联与实现

在上一篇里&#xff0c;我们完成了动作的创建&#xff0c;那么这一次&#xff0c;我们把它加载到界面上&#xff0c;把需要是实现的动作都加上。 MyWord::MyWord(QWidget *parent): QMainWindow(parent) {mdiAreanew QMdiArea;mdiArea->setHorizontalScrollBarPolicy(Qt::S…