竞赛选题 深度学习卷积神经网络垃圾分类系统 - 深度学习 神经网络 图像识别 垃圾分类 算法 小程序

文章目录

  • 0 简介
  • 1 背景意义
  • 2 数据集
  • 3 数据探索
  • 4 数据增广(数据集补充)
  • 5 垃圾图像分类
    • 5.1 迁移学习
      • 5.1.1 什么是迁移学习?
      • 5.1.2 为什么要迁移学习?
    • 5.2 模型选择
    • 5.3 训练环境
      • 5.3.1 硬件配置
      • 5.3.2 软件配置
    • 5.4 训练过程
    • 5.5 模型分类效果(PC端)
  • 6 构建垃圾分类小程序
    • 6.1 小程序功能
    • 6.2 分类测试
    • 6.3 垃圾分类小提示
    • 6.4 答题模块
  • 7 关键代码
  • 8 最后

0 简介

🔥 优质竞赛项目系列,今天要分享的是

深度学习卷积神经网络垃圾分类系统

该项目较为新颖,适合作为竞赛课题方向,学长非常推荐!

🧿 更多资料, 项目分享:

https://gitee.com/dancheng-senior/postgraduate

1 背景意义

近年来,随着我国经济的快速发展,国家各项建设都蒸蒸日上,成绩显著。但与此同时,也让资源与环境受到了严重破坏。这种现象与垃圾分类投放时的不合理直接相关,而人们对于环境污染问题反映强烈却束手无策,这两者间的矛盾日益尖锐。人们日常生活中的垃圾主要包括有害垃圾、厨余垃圾、可回收垃圾以及其他垃圾这四类,对不同类别的垃圾应采取不同分类方法,如果投放不当,可能会导致各种环境污染问题。合理地进行垃圾分类是有效进行垃圾处理、减少环境污染与资源再利用中的关键举措,也是目前最合适最有效的科学管理方式,利用现有的生产水平将日常垃圾按类别处理、利用有效物质和能量、填埋无用垃圾等。这样既能够提高垃圾资源处理效率,又能缓解环境污染问题。

而对垃圾的分类首先是在图像识别的基础上的,因此本文想通过使用近几年来发展迅速的深度学习方法设计一个垃圾分类系统,从而实现对日常生活中常见垃圾进行智能识别分类,提高人们垃圾分类投放意识,同时避免人们错误投放而产生的环境污染。

2 数据集

数据集采用了中国发布的垃圾分类标准,该标准将人们日常生活中常见的垃圾分为了四大类。其中,将废弃的玻璃、织物、家具以及电器电子产品等适合回收同时可循环利用的废弃物归为可回收垃圾。将剩菜剩饭、果皮果壳、花卉绿植以及其他餐厨垃圾等容易腐烂的废弃物归为厨余垃圾。将废电池、废药品、废灯管等对人们身体健康和自然环境有害而且应当门处理的废弃物归为有害垃圾。除以上三类垃圾之外的废弃物都归为其他垃圾。

该数据集是图片数据,分为训练集85%(Train)和测试集15%(Test)。其中O代表Organic(有机垃圾),R代表Recycle(可回收)

在这里插入图片描述

在这里插入图片描述

3 数据探索

我们先简单的大致看看数据的情况

所得的垃圾图片数据集中有40个二级类别,图片数量合计
14802张。由图3-1可以看出,各个垃圾类别的图像数据量不均衡,其中图片数据量较少的类别有:类别0(一次性快餐盒)、类别3(牙签)、类别20(快递纸袋);数据量较多的类别是:类别11(菜叶根)、类别21(插头电线)、类别25(毛绒玩具)。

在这里插入图片描述

4 数据增广(数据集补充)

数据增广就是对基础数据集进行扩充,避免因为数据集太少导致在模型训练过程可能出现的过拟合现象,以此来提高模型泛化能力,达到更好的效果。根据扩充数据集的来源可分为两类:内部数据增广是对基础数据集进行水平翻转、垂直翻转、高斯噪声以及高斯模糊等变换操作,来产生新的特征;而外部数据增广是引入新的高质量外部数据来扩充数据集,包括数据爬取与数据筛选两个步骤。

数据爬取是通过网络爬虫技术来实现的,爬虫的流程是,首先向远程服务器端发送请求,获取目标网页的HTML文件;然后跟踪这个链接文件,获取文件数据。各种搜索引擎就是通过爬虫技术来实现网页数据更新,爬取的效率直接决定了搜索的效果。

在这里插入图片描述

根据流程图可以看到,爬虫的流程与用户浏览网页的过程相似,首先输入目标URL地址,向服务器发送请求,接着服务器端会返回包含大量链接的HTML文件,然后提取这些链接将其组成URL列表,通过串行或并行方式从服务器端中下载数据。

由于基础数据集中类别数量不均衡,所以本设计使用网络爬虫方式从百度图库对数量较少的类别进行数据扩充,首先输入想要爬取的图片名称关键字,然后输入想要爬取图片的数量以及存放的文件夹之后,进行图片爬取。

在这里插入图片描述

在这里插入图片描述

5 垃圾图像分类

5.1 迁移学习

5.1.1 什么是迁移学习?

迁移学习是指在一个数据集上,重新利用之前已经训练过的卷积神经网络,并将其迁移到另外的数据集上。

5.1.2 为什么要迁移学习?

卷积神经网络前面的层提取的是图像的纹理、色彩等特征,而越靠近网络后端,提取的特征就会越高级、抽象。所以常用的微调方法是,保持网络中其他参数不变,只修改预训练网络的最后几层,最后几层的参数在新数据集上重新训练得到。其他层的参数保持不变,作为特征提取器,之后再使用较小的学习率训练整个网络。因为从零开始训练整个卷积网络是非常困难的,而且要花费大量的时间以及计算资源,所以采取迁移学习的方式是一种有效策略。

通常在非常大的数据集上对ConvNet进行预训练,然后将ConvNet用作初始化或者是固定特征提取器,以下是两个主要的迁移学习方法:

1.微调卷积网络。使用预训练的网络来初始化网络而不使用随机初始化,比较常用的方法是使用在ImageNet数据集上训练好的模型参数进行初始化,然后训练自己的数据集。

2.将卷积网络作为固定特征提取器。冻结除了全连接层外的所有其他层的权重,将最后的那个全连接层替换为具有随机权重的层,然后只对该层进行训练。

要使用深度学习方法来解决垃圾图像识别分类问题,就需要大量的垃圾图片数据集,因为当数据集太小时,一旦加深模型结构,就很可能出现过拟合的情况,训练出的模型泛化能力不足,识别准确率不高。而基于迁移学习的方法,预训练模型已经具备了提取图像基本特征基的能力,这就能在一定程度上减缓过拟合发生的可能性,将预模型迁移到垃圾图像数据集上进行微调训练,提高识别准确率。

5.2 模型选择

采用迁移学习的方式导入预训练模型,冻结特征提取层,进行微调训练,选取了SeNet154、Se_ResNet50、Se_ResNext101、ResNext101_32x16d_WSL四种模型进行对比实验,选取结果较好的模型进行调优。其中,ResNext101_32x16d_WS预训练模型是由FaceBook在2019年开源的

SeNet154结构

在这里插入图片描述

学长采用的模型结构:

在这里插入图片描述

采用ResNext101_32x16d_WSL网络作为基本的网络结构进行迁移学习,将CBAM注意力机制模块添加在首层卷积层,来增强图像特征表征能力,关注图像的重要特征抑制不必要的特征,固定除全连接层之外的其他层的权重。为降低过拟合,在模型全连接层添加了Dropout层,损失函数采用交叉熵损失函数(CrossEntropyLoss),优化函数对比了SGD和Adam,Adam在起始收敛速度快,但最终SGD精度高,所以采用了SGD。

5.3 训练环境

5.3.1 硬件配置

在这里插入图片描述

5.3.2 软件配置

在这里插入图片描述

5.4 训练过程

构建好模型结构后,设置数据集加载路径,在搭建好的环境中进行模型训练,将训练过程中每轮迭代的Train Loss、Valid Loss、Train
Acc、Valid Acc等数据保存到log日志文件中,然后使用matplotlib库绘制在训练集和测试集上的Accuracy跟Loss的变化曲线。

目前模型训练集准确度83.8%,测试集准确度67.5%,仍有待提高。。

在这里插入图片描述

5.5 模型分类效果(PC端)

在这里插入图片描述

6 构建垃圾分类小程序

学长设计的垃圾分类系统的核心功能是从本地相册上传照片或拍照上传照片进行识别分类,除此之外,还引入了语音识别功能、文字搜索功能、垃圾分类答题功能等满足用户的不同需求。系统的模块设计如下图所示。

·在这里插入图片描述

其中识别模块是用户选择识别功能,包含拍照/相册识别,语音识别、文字搜索等功能,根据所选城市的不同展示相应的垃圾类别;指南功能模块是根据所选城市的不同介绍各种垃圾的种类以及投放要求;答题模块实现垃圾种类的选择答题功能。

6.1 小程序功能

识别模块的功能包括文字搜索、语音识别、拍照识别等,该模块界面设计如图所示:

在这里插入图片描述

首先选择用户所在城市,然后选择使用的搜索方式,当通过三种搜索方式搜索不到相应垃圾类别时,可以通过反馈功能将未识别的垃圾名称向后台反馈信息,以便进一步完善系统。系统核心功能为拍照识别功能,拍照识别功能即调用在前面已经部署在华为云Model
Arts平台上的垃圾分类识别模型,对用户从手机端提交的垃圾图片进行在线识别分类并返回识别结果,调用过程中用到了小程序的云函数功能。

6.2 分类测试

在这里插入图片描述

在这里插入图片描述

6.3 垃圾分类小提示

指南模块实现的功能是根据用户所选择的城市,将云数据库中的数据展示给用户,介绍目前不同城市发布的垃圾分类规则及投放的要求,如下图所示:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

6.4 答题模块

答题模块也是根据用户所选城市的不同,测评用户对其所在城市垃圾分类规则了解的程度,以此来科普垃圾分类知识以及增强人们垃圾分类的意识,该界面如下图所示,在答完题后显示分数以及正确答案。

在这里插入图片描述

在这里插入图片描述

答题答案表


├─ 其他垃圾_PE塑料袋
├─ 其他垃圾_U型回形针
├─ 其他垃圾_一次性杯子
├─ 其他垃圾_一次性棉签
├─ 其他垃圾_串串竹签
├─ 其他垃圾_便利贴
├─ 其他垃圾_创可贴
├─ 其他垃圾_卫生纸
├─ 其他垃圾_厨房手套
├─ 其他垃圾_厨房抹布
├─ 其他垃圾_口罩
├─ 其他垃圾_唱片
├─ 其他垃圾_图钉
├─ 其他垃圾_大龙虾头
├─ 其他垃圾_奶茶杯
├─ 其他垃圾_干燥剂
├─ 其他垃圾_彩票
├─ 其他垃圾_打泡网
├─ 其他垃圾_打火机
├─ 其他垃圾_搓澡巾
├─ 其他垃圾_果壳
├─ 其他垃圾_毛巾
├─ 其他垃圾_涂改带
├─ 其他垃圾_湿纸巾
├─ 其他垃圾_烟蒂
├─ 其他垃圾_牙刷
├─ 其他垃圾_电影票
├─ 其他垃圾_电蚊香
├─ 其他垃圾_百洁布
├─ 其他垃圾_眼镜
├─ 其他垃圾_眼镜布
├─ 其他垃圾_空调滤芯
├─ 其他垃圾_笔
├─ 其他垃圾_胶带
├─ 其他垃圾_胶水废包装
├─ 其他垃圾_苍蝇拍
├─ 其他垃圾_茶壶碎片
├─ 其他垃圾_草帽
├─ 其他垃圾_菜板
├─ 其他垃圾_车票
├─ 其他垃圾_酒精棉
├─ 其他垃圾_防霉防蛀片
├─ 其他垃圾_除湿袋
├─ 其他垃圾_餐巾纸
├─ 其他垃圾_餐盒
├─ 其他垃圾_验孕棒
├─ 其他垃圾_鸡毛掸
├─ 厨余垃圾_八宝粥
├─ 厨余垃圾_冰激凌
├─ 厨余垃圾_冰糖葫芦
├─ 厨余垃圾_咖啡
├─ 厨余垃圾_圣女果
├─ 厨余垃圾_地瓜
├─ 厨余垃圾_坚果
├─ 厨余垃圾_壳
├─ 厨余垃圾_巧克力
├─ 厨余垃圾_果冻
├─ 厨余垃圾_果皮
├─ 厨余垃圾_核桃
├─ 厨余垃圾_梨
├─ 厨余垃圾_橙子
├─ 厨余垃圾_残渣剩饭
├─ 厨余垃圾_水果
├─ 厨余垃圾_泡菜
├─ 厨余垃圾_火腿
├─ 厨余垃圾_火龙果
├─ 厨余垃圾_烤鸡
├─ 厨余垃圾_瓜子
├─ 厨余垃圾_甘蔗
├─ 厨余垃圾_番茄
├─ 厨余垃圾_秸秆杯
├─ 厨余垃圾_秸秆碗
├─ 厨余垃圾_粉条
├─ 厨余垃圾_肉类
├─ 厨余垃圾_肠
├─ 厨余垃圾_苹果
├─ 厨余垃圾_茶叶
├─ 厨余垃圾_草莓
├─ 厨余垃圾_菠萝
├─ 厨余垃圾_菠萝蜜
├─ 厨余垃圾_萝卜
├─ 厨余垃圾_蒜
├─ 厨余垃圾_蔬菜
├─ 厨余垃圾_薯条
├─ 厨余垃圾_薯片
├─ 厨余垃圾_蘑菇
├─ 厨余垃圾_蛋
├─ 厨余垃圾_蛋挞
├─ 厨余垃圾_蛋糕
├─ 厨余垃圾_豆
├─ 厨余垃圾_豆腐
├─ 厨余垃圾_辣椒
├─ 厨余垃圾_面包
├─ 厨余垃圾_饼干
├─ 厨余垃圾_鸡翅
├─ 可回收物_不锈钢制品
├─ 可回收物_乒乓球拍
├─ 可回收物_书
├─ 可回收物_体重秤
├─ 可回收物_保温杯
├─ 可回收物_保鲜膜内芯
├─ 可回收物_信封
├─ 可回收物_充电头
├─ 可回收物_充电宝
├─ 可回收物_充电牙刷
├─ 可回收物_充电线
├─ 可回收物_凳子
├─ 可回收物_刀
├─ 可回收物_包
├─ 可回收物_单车
├─ 可回收物_卡
├─ 可回收物_台灯
├─ 可回收物_吊牌
├─ 可回收物_吹风机
├─ 可回收物_呼啦圈
├─ 可回收物_地球仪
├─ 可回收物_地铁票
├─ 可回收物_垫子
├─ 可回收物_塑料制品
├─ 可回收物_太阳能热水器
├─ 可回收物_奶粉桶
├─ 可回收物_尺子
├─ 可回收物_尼龙绳
├─ 可回收物_布制品
├─ 可回收物_帽子
├─ 可回收物_手机
├─ 可回收物_手电筒
├─ 可回收物_手表
├─ 可回收物_手链
├─ 可回收物_打包绳
├─ 可回收物_打印机
├─ 可回收物_打气筒
├─ 可回收物_扫地机器人
├─ 可回收物_护肤品空瓶
├─ 可回收物_拉杆箱
├─ 可回收物_拖鞋
├─ 可回收物_插线板
├─ 可回收物_搓衣板
├─ 可回收物_收音机
├─ 可回收物_放大镜
├─ 可回收物_日历
├─ 可回收物_暖宝宝
├─ 可回收物_望远镜
├─ 可回收物_木制切菜板
├─ 可回收物_木桶
├─ 可回收物_木棍
├─ 可回收物_木质梳子
├─ 可回收物_木质锅铲
├─ 可回收物_木雕
├─ 可回收物_枕头
├─ 可回收物_果冻杯
├─ 可回收物_桌子
├─ 可回收物_棋子
├─ 可回收物_模具
├─ 可回收物_毯子
├─ 可回收物_水壶
├─ 可回收物_水杯
├─ 可回收物_沙发
├─ 可回收物_泡沫板
├─ 可回收物_灭火器
├─ 可回收物_灯罩
├─ 可回收物_烟灰缸
├─ 可回收物_热水瓶
├─ 可回收物_燃气灶
├─ 可回收物_燃气瓶
├─ 可回收物_玩具
├─ 可回收物_玻璃制品
├─ 可回收物_玻璃器皿
├─ 可回收物_玻璃壶
├─ 可回收物_玻璃球
├─ 可回收物_瑜伽球
├─ 可回收物_电动剃须刀
├─ 可回收物_电动卷发棒
├─ 可回收物_电子秤
├─ 可回收物_电熨斗
├─ 可回收物_电磁炉
├─ 可回收物_电脑屏幕
├─ 可回收物_电视机
├─ 可回收物_电话
├─ 可回收物_电路板
├─ 可回收物_电风扇
├─ 可回收物_电饭煲
├─ 可回收物_登机牌
├─ 可回收物_盒子
├─ 可回收物_盖子
├─ 可回收物_盘子
├─ 可回收物_碗
├─ 可回收物_磁铁
├─ 可回收物_空气净化器
├─ 可回收物_空气加湿器
├─ 可回收物_笼子
├─ 可回收物_箱子
├─ 可回收物_纸制品
├─ 可回收物_纸牌
├─ 可回收物_罐子
├─ 可回收物_网卡
├─ 可回收物_耳套
├─ 可回收物_耳机
├─ 可回收物_衣架
├─ 可回收物_袋子
├─ 可回收物_袜子
├─ 可回收物_裙子
├─ 可回收物_裤子
├─ 可回收物_计算器
├─ 可回收物_订书机
├─ 可回收物_话筒
├─ 可回收物_豆浆机
├─ 可回收物_路由器
├─ 可回收物_轮胎
├─ 可回收物_过滤网
├─ 可回收物_遥控器
├─ 可回收物_量杯
├─ 可回收物_金属制品
├─ 可回收物_钉子
├─ 可回收物_钥匙
├─ 可回收物_铁丝球
├─ 可回收物_铅球
├─ 可回收物_铝制用品
├─ 可回收物_锅
├─ 可回收物_锅盖
├─ 可回收物_键盘
├─ 可回收物_镊子
├─ 可回收物_闹铃
├─ 可回收物_雨伞
├─ 可回收物_鞋
├─ 可回收物_音响
├─ 可回收物_餐具
├─ 可回收物_餐垫
├─ 可回收物_饰品
├─ 可回收物_鱼缸
├─ 可回收物_鼠标
├─ 有害垃圾_指甲油
├─ 有害垃圾_杀虫剂
├─ 有害垃圾_温度计
├─ 有害垃圾_灯
├─ 有害垃圾_电池
├─ 有害垃圾_电池板
├─ 有害垃圾_纽扣电池
├─ 有害垃圾_胶水
├─ 有害垃圾_药品包装
├─ 有害垃圾_药片
├─ 有害垃圾_药瓶
├─ 有害垃圾_药膏
├─ 有害垃圾_蓄电池
└─ 有害垃圾_血压计

7 关键代码

import tensorflow as tfimport linecacheimport cv2import numpy as npimport osfrom select_object import pretreatment_imagetrain_images_path = 'D:/WorkSpace/Python/trash_classify_dataset/dataset/'train_labels_path = 'D:/WorkSpace/Python/trash_classify_dataset/train_label.txt'test_images_path = 'D:/WorkSpace/Python/trash_classify_dataset/dataset/'test_labels_path = 'D:/WorkSpace/Python/trash_classify_dataset/test_label.txt'classify_num = 50train_images_num = 29081test_images_num = 3232def load_train_dataset(index):  # 从1开始if index > train_images_num:if index % train_images_num == 0:index = train_images_numelse:index %= train_images_numline_str = linecache.getline(train_labels_path, index)image_name, image_label = line_str.split(' ')image = cv2.imread(train_images_path + image_name)# cv2.imshow('pic',image)image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)image = cv2.resize(image, (224, 224))return image, image_labeldef combine_train_dataset(count, size):train_images_load = np.zeros(shape=(size, 224, 224, 3))train_labels_load = np.zeros(shape=(size, classify_num))for i in range(size):train_images_load[i], train_labels_index = load_train_dataset(count + i + 1)train_labels_load[i][int(train_labels_index) - 1] = 1.0count += sizereturn train_images_load, train_labels_load, countdef load_test_dataset(index):  # 从1开始if index > test_images_num:if index % test_images_num == 0:index = test_images_numelse:index %= test_images_numline_str = linecache.getline(test_labels_path, index)image_name, image_label = line_str.split(' ')image = cv2.imread(test_images_path + image_name)image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)image = cv2.resize(image, (224, 224))return image, image_labeldef combine_test_dataset(count, size):test_images_load = np.zeros(shape=(size, 224, 224, 3))test_labels_load = np.zeros(shape=(size, classify_num))for i in range(size):test_images_load[i], test_labels_index = load_test_dataset(count + i + 1)test_labels_load[i][int(test_labels_index) - 1] = 1.0count += sizereturn test_images_load, test_labels_load, count# # 通过L2正则化防止过拟合# def weight_variable_with_loss(shape, stddev, lam):#     weight = tf.Variable(tf.truncated_normal(shape, stddev=stddev))#     if lam is not None:#         weight_loss = tf.multiply(tf.nn.l2_loss(weight), lam, name='weight_loss')#         tf.add_to_collection('losses', weight_loss)#     return weightdef weight_variable(shape, n, use_l2, lam):weight = tf.Variable(tf.truncated_normal(shape, stddev=1 / n))# L2正则化if use_l2 is True:weight_loss = tf.multiply(tf.nn.l2_loss(weight), lam, name='weight_loss')tf.add_to_collection('losses', weight_loss)return weightdef bias_variable(shape):bias = tf.Variable(tf.constant(0.1, shape=shape))return biasdef conv2d(x, w):return tf.nn.conv2d(x, w, strides=[1, 1, 1, 1], padding='SAME')def max_pool_2x2(x):return tf.nn.max_pool(x, ksize=[1, 2, 2, 1],strides=[1, 2, 2, 1], padding='SAME')# 输入层with tf.name_scope('input_layer'):x_input = tf.placeholder(tf.float32, [None, 224, 224, 3])y_input = tf.placeholder(tf.float32, [None, classify_num])keep_prob = tf.placeholder(tf.float32)is_training = tf.placeholder(tf.bool)is_use_l2 = tf.placeholder(tf.bool)lam = tf.placeholder(tf.float32)learning_rate = tf.placeholder(tf.float32)# 数据集平均RGB值mean = tf.constant([159.780, 139.802, 119.047], dtype=tf.float32, shape=[1, 1, 1, 3])x_input = x_input - mean# 第一个卷积层 size:224# 卷积核1[3, 3, 3, 64]# 卷积核2[3, 3, 64, 64]with tf.name_scope('conv1_layer'):w_conv1 = weight_variable([3, 3, 3, 64], 64, use_l2=False, lam=0)b_conv1 = bias_variable([64])conv_kernel1 = conv2d(x_input, w_conv1)bn1 = tf.layers.batch_normalization(conv_kernel1, training=is_training)conv1 = tf.nn.relu(tf.nn.bias_add(bn1, b_conv1))w_conv2 = weight_variable([3, 3, 64, 64], 64, use_l2=False, lam=0)b_conv2 = bias_variable([64])conv_kernel2 = conv2d(conv1, w_conv2)bn2 = tf.layers.batch_normalization(conv_kernel2, training=is_training)conv2 = tf.nn.relu(tf.nn.bias_add(bn2, b_conv2))pool1 = max_pool_2x2(conv2)  # 224*224 -> 112*112result1 = pool1# 第二个卷积层 size:112# 卷积核3[3, 3, 64, 128]# 卷积核4[3, 3, 128, 128]with tf.name_scope('conv2_layer'):w_conv3 = weight_variable([3, 3, 64, 128], 128, use_l2=False, lam=0)b_conv3 = bias_variable([128])conv_kernel3 = conv2d(result1, w_conv3)bn3 = tf.layers.batch_normalization(conv_kernel3, training=is_training)conv3 = tf.nn.relu(tf.nn.bias_add(bn3, b_conv3))w_conv4 = weight_variable([3, 3, 128, 128], 128, use_l2=False, lam=0)b_conv4 = bias_variable([128])conv_kernel4 = conv2d(conv3, w_conv4)bn4 = tf.layers.batch_normalization(conv_kernel4, training=is_training)conv4 = tf.nn.relu(tf.nn.bias_add(bn4, b_conv4))pool2 = max_pool_2x2(conv4)  # 112*112 -> 56*56result2 = pool2# 第三个卷积层 size:56# 卷积核5[3, 3, 128, 256]# 卷积核6[3, 3, 256, 256]# 卷积核7[3, 3, 256, 256]with tf.name_scope('conv3_layer'):w_conv5 = weight_variable([3, 3, 128, 256], 256, use_l2=False, lam=0)b_conv5 = bias_variable([256])conv_kernel5 = conv2d(result2, w_conv5)bn5 = tf.layers.batch_normalization(conv_kernel5, training=is_training)conv5 = tf.nn.relu(tf.nn.bias_add(bn5, b_conv5))w_conv6 = weight_variable([3, 3, 256, 256], 256, use_l2=False, lam=0)b_conv6 = bias_variable([256])conv_kernel6 = conv2d(conv5, w_conv6)bn6 = tf.layers.batch_normalization(conv_kernel6, training=is_training)conv6 = tf.nn.relu(tf.nn.bias_add(bn6, b_conv6))w_conv7 = weight_variable([3, 3, 256, 256], 256, use_l2=False, lam=0)b_conv7 = bias_variable([256])conv_kernel7 = conv2d(conv6, w_conv7)bn7 = tf.layers.batch_normalization(conv_kernel7, training=is_training)conv7 = tf.nn.relu(tf.nn.bias_add(bn7, b_conv7))pool3 = max_pool_2x2(conv7)  # 56*56 -> 28*28result3 = pool3# 第四个卷积层 size:28# 卷积核8[3, 3, 256, 512]# 卷积核9[3, 3, 512, 512]# 卷积核10[3, 3, 512, 512]with tf.name_scope('conv4_layer'):w_conv8 = weight_variable([3, 3, 256, 512], 512, use_l2=False, lam=0)b_conv8 = bias_variable([512])conv_kernel8 = conv2d(result3, w_conv8)bn8 = tf.layers.batch_normalization(conv_kernel8, training=is_training)conv8 = tf.nn.relu(tf.nn.bias_add(bn8, b_conv8))w_conv9 = weight_variable([3, 3, 512, 512], 512, use_l2=False, lam=0)b_conv9 = bias_variable([512])conv_kernel9 = conv2d(conv8, w_conv9)bn9 = tf.layers.batch_normalization(conv_kernel9, training=is_training)conv9 = tf.nn.relu(tf.nn.bias_add(bn9, b_conv9))w_conv10 = weight_variable([3, 3, 512, 512], 512, use_l2=False, lam=0)b_conv10 = bias_variable([512])conv_kernel10 = conv2d(conv9, w_conv10)bn10 = tf.layers.batch_normalization(conv_kernel10, training=is_training)conv10 = tf.nn.relu(tf.nn.bias_add(bn10, b_conv10))pool4 = max_pool_2x2(conv10)  # 28*28 -> 14*14result4 = pool4# 第五个卷积层 size:14# 卷积核11[3, 3, 512, 512]# 卷积核12[3, 3, 512, 512]# 卷积核13[3, 3, 512, 512]with tf.name_scope('conv5_layer'):w_conv11 = weight_variable([3, 3, 512, 512], 512, use_l2=False, lam=0)b_conv11 = bias_variable([512])conv_kernel11 = conv2d(result4, w_conv11)bn11 = tf.layers.batch_normalization(conv_kernel11, training=is_training)conv11 = tf.nn.relu(tf.nn.bias_add(bn11, b_conv11))w_conv12 = weight_variable([3, 3, 512, 512], 512, use_l2=False, lam=0)b_conv12 = bias_variable([512])conv_kernel12 = conv2d(conv11, w_conv12)bn12 = tf.layers.batch_normalization(conv_kernel12, training=is_training)conv12 = tf.nn.relu(tf.nn.bias_add(bn12, b_conv12))w_conv13 = weight_variable([3, 3, 512, 512], 512, use_l2=False, lam=0)b_conv13 = bias_variable([512])conv_kernel13 = conv2d(conv12, w_conv13)bn13 = tf.layers.batch_normalization(conv_kernel13, training=is_training)conv13 = tf.nn.relu(tf.nn.bias_add(bn13, b_conv13))pool5 = max_pool_2x2(conv13)  # 14*14 -> 7*7result5 = pool5# 第一个全连接层 size:7# 隐藏层节点数 4096with tf.name_scope('fc1_layer'):w_fc14 = weight_variable([7 * 7 * 512, 4096], 4096, use_l2=is_use_l2, lam=lam)b_fc14 = bias_variable([4096])result5_flat = tf.reshape(result5, [-1, 7 * 7 * 512])fc14 = tf.nn.relu(tf.nn.bias_add(tf.matmul(result5_flat, w_fc14), b_fc14))# result6 = fc14result6 = tf.nn.dropout(fc14, keep_prob)# 第二个全连接层# 隐藏层节点数 4096with tf.name_scope('fc2_layer'):w_fc15 = weight_variable([4096, 4096], 4096, use_l2=is_use_l2, lam=lam)b_fc15 = bias_variable([4096])fc15 = tf.nn.relu(tf.nn.bias_add(tf.matmul(result6, w_fc15), b_fc15))# result7 = fc15result7 = tf.nn.dropout(fc15, keep_prob)# 输出层with tf.name_scope('output_layer'):w_fc16 = weight_variable([4096, classify_num], classify_num, use_l2=is_use_l2, lam=lam)b_fc16 = bias_variable([classify_num])fc16 = tf.matmul(result7, w_fc16) + b_fc16logits = tf.nn.softmax(fc16)# 损失函数with tf.name_scope('loss'):cross_entropy = tf.nn.softmax_cross_entropy_with_logits_v2(logits=fc16, labels=y_input)cross_entropy_mean = tf.reduce_mean(cross_entropy)tf.add_to_collection('losses', cross_entropy_mean)loss = tf.add_n(tf.get_collection('losses'))tf.summary.scalar('loss', loss)# 训练函数with tf.name_scope('train'):update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)with tf.control_dependencies(update_ops):  # 保证train_op在update_ops执行之后再执行。train_step = tf.train.AdamOptimizer(learning_rate).minimize(loss)# 计算准确率with tf.name_scope('accuracy'):correct_prediction = tf.equal(tf.argmax(logits, 1), tf.argmax(y_input, 1))accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))tf.summary.scalar('accuracy', accuracy)# 会话初始化# sess = tf.InteractiveSession()# tf.global_variables_initializer().run()saver = tf.train.Saver()save_dir = "classify_modles"checkpoint_name = "train.ckpt"merged = tf.summary.merge_all()  # 将图形、训练过程等数据合并在一起# writer_train = tf.summary.FileWriter('logs/train', sess.graph)  # 将训练日志写入到logs文件夹下# writer_test = tf.summary.FileWriter('logs/test', sess.graph)  # 将训练日志写入到logs文件夹下# 变量初始化training_steps = 25000display_step = 10batch_size = 20train_images_count = 0test_images_count = 0train_avg_accuracy = 0test_avg_accuracy = 0# # 训练# print("Training start...")## # # 模型恢复# # sess = tf.InteractiveSession()# # saver.restore(sess, os.path.join(save_dir, checkpoint_name))# # print("Model restore success!")## for step in range(training_steps):#     train_images, train_labels, train_images_count = combine_train_dataset(train_images_count, batch_size)#     test_images, test_labels, test_images_count = combine_test_dataset(test_images_count, batch_size)##     # 训练#     if step < 10000:#         train_step.run(#             feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 0.8, is_training: True, is_use_l2: True,#                        learning_rate: 0.0001, lam: 0.004})#     elif step < 20000:#         train_step.run(#             feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 0.8, is_training: True, is_use_l2: True,#                        learning_rate: 0.0001, lam: 0.001})#     else:#         train_step.run(#             feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 0.8, is_training: True, is_use_l2: True,#                        learning_rate: 0.00001, lam: 0.001})##     # 每训练10步,输出显示训练过程#     if step % display_step == 0:#         train_accuracy = accuracy.eval(#             feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 1.0, is_training: False,#                        is_use_l2: False})#         train_loss = sess.run(loss, feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 1.0,#                                                is_training: False, is_use_l2: False})#         train_result = sess.run(tf.argmax(logits, 1),#                                 feed_dict={x_input: train_images, keep_prob: 1.0, is_training: False, is_use_l2: False})#         train_label = sess.run(tf.argmax(y_input, 1), feed_dict={y_input: train_labels})##         test_accuracy = accuracy.eval(#             feed_dict={x_input: test_images, y_input: test_labels, keep_prob: 1.0, is_training: False,#                        is_use_l2: False})#         test_result = sess.run(tf.argmax(logits, 1),#                                feed_dict={x_input: test_images, keep_prob: 1.0, is_training: False, is_use_l2: False})#         test_label = sess.run(tf.argmax(y_input, 1), feed_dict={y_input: test_labels})##         print("Training dataset:")#         print(train_result)#         print(train_label)#         print("Testing dataset:")#         print(test_result)#         print(test_label)##         print("step {}\n training accuracy {}\n loss {}\n testing accuracy {}\n".format(step, train_accuracy, train_loss, test_accuracy))#         train_avg_accuracy += train_accuracy#         test_avg_accuracy += test_accuracy#         result_train = sess.run(merged, feed_dict={x_input: train_images, y_input: train_labels, keep_prob: 1.0,#                                                    is_training: False, is_use_l2: False})  # 计算需要写入的日志数据#         writer_train.add_summary(result_train, step)  # 将日志数据写入文件##         result_test = sess.run(merged, feed_dict={x_input: test_images, y_input: test_labels, keep_prob: 1.0,#                                                is_training: False, is_use_l2: False})  # 计算需要写入的日志数据#         writer_test.add_summary(result_test, step)  # 将日志数据写入文件##     # 每训练100步,显示输出训练平均准确度,保存模型#     if step % (display_step * 10) == 0 and step != 0:#         print("train_avg_accuracy {}".format(train_avg_accuracy / 10))#         train_avg_accuracy = 0#         print("test_avg_accuracy {}".format(test_avg_accuracy / 10))#         test_avg_accuracy = 0##         saver.save(sess, os.path.join(save_dir, checkpoint_name))#         print("Model save success!\n")## print("Training finish...")## # 模型保存# saver.save(sess, os.path.join(save_dir, checkpoint_name))# print("\nModel save success!")## # print("\nTesting start...")# # avg_accuracy = 0# # for i in range(int(test_images_num / 30) + 1):# #     test_images, test_labels, test_images_count = combine_test_dataset(test_images_count, 30)# #     test_accuracy = accuracy.eval(# #         feed_dict={x_input: test_images, y_input: test_labels, keep_prob: 1.0, is_training: False, is_use_l2: False})# #     test_result = sess.run(tf.argmax(logits, 1),# #                            feed_dict={x_input: test_images, keep_prob: 1.0, is_training: False, is_use_l2: False})# #     test_label = sess.run(tf.argmax(y_input, 1), feed_dict={y_input: test_labels})# #     print(test_result)# #     print(test_label)# #     print("test accuracy {}".format(test_accuracy))# #     avg_accuracy += test_accuracy# ## # print("\ntest_avg_accuracy {}".format(avg_accuracy / (int(test_images_num / 30) + 1)))## sess.close()# 识别# 模型恢复sess = tf.InteractiveSession()saver.restore(sess, os.path.join(save_dir, checkpoint_name))print("Model restore success!")def predict_img(img_path):img = cv2.imread(img_path)image = np.reshape(img, [1, 224, 224, 3])classify_result = sess.run(tf.argmax(logits, 1),feed_dict={x_input: image, keep_prob: 1.0, is_training: False, is_use_l2: False})probability = sess.run(logits, feed_dict={x_input: image, keep_prob: 1.0, is_training: False,is_use_l2: False}).flatten().tolist()[classify_result[0]]return classify_result[0], probabilitydef trash_classify(img_path, img_name, upload_path):img_name = img_name.rsplit('.', 1)[0]# print(img_name)pretrian_img_path, selected_img_path = pretreatment_image(img_path, img_name, upload_path)predict_result, predict_probability = predict_img(pretrian_img_path)return predict_result, predict_probability

8 最后

🧿 更多资料, 项目分享:

https://gitee.com/dancheng-senior/postgraduate

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

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

相关文章

Cookie技术

Cookie中文名称为小型文本文件&#xff0c;指某些网站为了辨别用户身份、进行会话跟踪而储存在用户本地终端上的数据。 Cookie是由服务器端生成&#xff0c;发送给User-Agent&#xff08;—般是浏览器&#xff09;&#xff0c;浏览器会将Cookie的key/value保存到某个目录下的文…

Harmony 个人中心(页面交互、跳转、导航、容器组件)

个人中心 前言正文一、创建工程二、登录① 更换启动页面② 拓展修饰符③ 页面跳转④ 等待进度条 三、导航栏四、首页① 轮播图② 网格列表 五、我的① 带参数跳转 六、源码 前言 今天是1024&#xff0c;祝各位程序员们&#xff0c;钱多事少离家近&#xff0c;不秃也强bug黄。在…

本机spark 通idea连接Oracle的坑

1. 报错&#xff1a;Exception in thread "main" java.lang.NoSuchMethodError: scala.Product.$init$(Lscala/Product;)V 查询网上资料&#xff0c;是idea引入的scala运行环境版本与idea默认的scala版本不一样 也就是写的项目中的pom的spark版本与idea默认的版本不…

记一次 .Net+SqlSugar 查询超时的问题排查过程

环境和版本&#xff1a;.Net 6 SqlSuger 5.1.4.* &#xff0c;数据库是mysql 5.7 &#xff0c;数据量在2000多条左右 业务是一个非常简单的查询&#xff0c;代码如下&#xff1a; var list _dbClient.Queryable<tb_name>().ToList(); tb_name 下配置了一对多的关系…

【C#】LIMS实验室信息管理系统源码

一、系统概述 LIMS(Laboratory Information Management System)即实验室信息管理系统,是通过对样品检验流程、分析数据及报告、实验室资源和客户信息等要素的综合管理,按照标准化实验室管理规范,建立符合实验室业务流程的质量体系,实现实验室信息化管理。是实验室提高分析水平…

【开源】基于SpringBoot的车险自助理赔系统的设计和实现

目录 一、摘要1.1 项目介绍1.2 项目录屏 二、功能模块2.1 数据中心模块2.2 角色管理模块2.3 车辆档案模块2.4 车辆理赔模块2.5 理赔照片模块 三、系统设计3.1 用例设计3.2 数据库设计3.2.1 角色表3.2.2 车辆表3.2.3 理赔表3.2.4 理赔照片表 四、系统展示五、核心代码5.1 查询车…

当生成式AI遇到业务流程管理,大语言模型正在变革BPM

生成式AI对各领域有很大影响&#xff0c;一个方面在于它改变了很多固有业务的工作流。 工作流&#xff08;Workflow&#xff09;是业务流程的一种实现方式&#xff0c;一个业务流程往往包含多个工作流范式以及相关的数据、组织和系统。 因此&#xff0c;提及工作流必然离不开业…

有线网卡通过无线网卡使其它设备上网

我现在的网络是无线路由器连接公网&#xff0c;电脑上的无线网卡连接路由器使电脑上网&#xff0c;这是完全正常的连接方式。 我现在又有了一台嵌入式设备&#xff0c;它只有有线网口&#xff0c;所以就只能用有线的方式连网&#xff0c;但是我的无线路由器不在电脑旁边&#x…

行业追踪,2023-10-27

自动复盘 2023-10-27 凡所有相&#xff0c;皆是虚妄。若见诸相非相&#xff0c;即见如来。 k 线图是最好的老师&#xff0c;每天持续发布板块的rps排名&#xff0c;追踪板块&#xff0c;板块来开仓&#xff0c;板块去清仓&#xff0c;丢弃自以为是的想法&#xff0c;板块去留让…

软考系列(系统架构师)- 2013年系统架构师软考案例分析考点

试题一 软件架构&#xff08;根据描述填表、ESB 定义和功能&#xff09; 【问题1】&#xff08;10分&#xff09; 服务建模是对Ramp Coordination信息系统进行集成的首要工作&#xff0c;公司的架构师首先对Ramp Coordination信息系统进行服务建模&#xff0c;识别出系统中的两…

化工园区数字孪生可视化管控平台,赋予园区安全环保智慧发展

化工行业作为国民经济的支柱和工业发展的引擎&#xff0c;对安全生产、环保节能、应急管控有着很高的要求。目前国内外化工园区面临安全和环保两大压力。为有效解决这两大难题&#xff0c;巨蟹数科综合运用物联网、数字孪生等新一代信息技术&#xff0c;建设了数字孪生园区智慧…

测试用例的设计方法(全):边界值分析方法

一.方法简介 1.定义&#xff1a;边界值分析法就是对输入或输出的边界值进行测试的一种黑盒测试方法。通常边界值分析法是作为对等价类划分法的补充&#xff0c;这种情况下&#xff0c;其测试用例来自等价类的边界。 2.与等价划分的区别 1)边界值分析不是从某等价类中随便挑…

大模型之十九-对话机器人

大语言模型的最早应用是Chatbot&#xff0c;其实我最早接触语义理解在2014年&#xff0c;2014年做智能音箱的时候&#xff0c;那时也是国内第一批做智能音箱的&#xff0c;在现在看起来当时的智能音箱比较傻&#xff0c;很多问题无法回答&#xff0c;长下文效果也不好&#xff…

Peter算法小课堂—归并排序

位运算 << 这个符号相当于将一个数二进制往左移动几位&#xff0c;如(100110)2<<1(001100)2。相当于乘以2的k次方 >> 这个符号相当于将一个数二进制往右移动几位&#xff0c;如(100110)2<<1(0100110)2。相当于除以2的k次方 归并排序 先看一个视频…

vue3 code format bug

vue code format bug vue客户端代码格式化缺陷&#xff0c;为了方便阅读和维护&#xff0c;对代码格式化发现这个缺陷 vue.global.min.3.2.26.js var Vuefunction(r){"use strict";function e(e,t){const nObject.create(null);var re.split(",");for(le…

docker 中给命令起别名

docker 的有些命令特别复杂&#xff0c;我们可以给它设置别名简化输入&#xff0c;就不用每次都输入那么多了&#xff01;&#xff01;&#xff01; 1. 进入 .bashrc 中修改配置&#xff08; .bashrc 是root下的隐藏文件&#xff09; cd /rootvim .bashrc2. 在 .bashrc 中加入…

使用Hystrix实现请求合并,降低服务器并发压力

1.引入Hystrix <dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-starter-netflix-hystrix</artifactId></dependency> 2.在启动类上开启Hystrix功能 EnableHystrix 3.请求合并实现代码 import com…

【k8s】kubeadm安装k8s集群

一、环境部署 master192.168.88.10docker、kubeadm、kubelet、kubectl、flannelnode01192.168.88.20docker、kubeadm、kubelet、kubectl、flannelnode02192.168.88.30docker、kubeadm、kubelet、kubectl、flannelhub.lp.com192.168.88.40 docker、docker-compose harbor-offli…

【深度学习】使用Pytorch实现的用于时间序列预测的各种深度学习模型类

深度学习模型类 简介按滑动时间窗口切割数据集模型类CNNGRULSTMMLPRNNTCNTransformerSeq2Seq 简介 本文所定义模型类的输入数据的形状shape统一为 [batch_size, time_step&#xff0c;n_features]&#xff0c;batch_size为批次大小&#xff0c;time_step为时间步长&#xff0c…

【抓包分析】通过ChatGPT解密还原某软件登录算法实现绕过手机验证码登录

文章目录 &#x1f34b;前言实现效果成品广告抓包分析一、定位加密文件二、编辑JS启用本地替换 利用Chatgpt进行代码转换获取计划任务id模拟数据请求最后 &#x1f34b;前言 由于C站版权太多&#xff0c;所有的爬虫相关均为记录&#xff0c;不做深入&#xff01; 今天发现gith…