1 Star 27 Fork 7

肆十二 / trash_classification_tf2.3

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

基于tensorflow2.3的垃圾分类系统

教程链接:【03】手把手教你构建垃圾分类系统-基于tensorflow2.3_dejahu的博客-CSDN博客

课程设计要做一个垃圾分类系统,需要识别可回收垃圾、厨余垃圾、有害垃圾和其他垃圾等四个大类,在网上找到了很多开源的数据集,但是质量参差不齐,而且有坏图的存在,所以我就将这些数据集还有自己爬取的数据一起清洗了一遍,全部保存为了jpg的格式,一共有245个小类和4个大类。模型训练使用的是tensorflow2.3,其中mobilenet的准确率有82%,并使用pyqt5构建了图形化界面。

如何获取

代码直接在本地址下载即可

需要模型和数据集的朋友请在csdn下载,链接如下:

垃圾分类数据集和tf代码-8w张图片245个类.zip-深度学习文档类资源-CSDN下载

代码结构

images 目录主要是放置一些图片,包括测试的图片和ui界面使用的图片
models 目录下放置训练好的两组模型,分别是cnn模型和mobilenet的模型
results 目录下放置的是训练的训练过程的一些可视化的图,两个txt文件是训练过程中的输出,两个图是两个模型训练过程中训练集和验证集准确率和loss变化曲线
utils 是主要是我测试的时候写的一些文件,对这个项目没有实际的用途
mainwindow.py 是界面文件,主要是利用pyqt5完成的界面,通过上传图片可以对图片种类进行预测
testmodel.py 是测试文件,主要是用于测试两组模型在验证集上的准确率,这个信息你从results的txt的输出中也能获取
train_cnn.py 是训练cnn模型的代码
train_mobilenet.py 是训练mobilenet模型的代码

效果

image-20210618133633509

如何运行

不熟悉环境配置的朋友可以看这篇博客,里面有详细的教程:

手把手教你用tensorflow2.3训练自己的分类数据集_dejahu的博客-CSDN博客

空文件

简介

基于tensorflow2.3的垃圾分类系统 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
1
https://gitee.com/song-laogou/trash_classification_tf2.3.git
git@gitee.com:song-laogou/trash_classification_tf2.3.git
song-laogou
trash_classification_tf2.3
trash_classification_tf2.3
master

搜索帮助