使用Python轻松完成垃圾分类(基于图像识别)


Posted in Python onJuly 09, 2019

0 环境

Python版本:3.6.8

系统版本:macOS Mojave

Python Jupyter Notebook

1 引言

七月了,大家最近一定被一项新的政策给折磨的焦头烂额,那就是垃圾分类。《上海市生活垃圾管理条例》已经正式实施了,相信还是有很多的小伙伴和我一样,还没有完全搞清楚哪些应该扔在哪个类别里。感觉每天都在学习一遍垃圾分类,真令人头大。

听说一杯没有喝完的珍珠奶茶应该这么扔

  • 首先,没喝完的奶茶水要倒在水池里
  • 珍珠,水果肉等残渣放进湿垃圾
  • 把杯子要丢入干垃圾
  • 接下来是盖子,如果是带盖子带热饮(比如大部分的热饮),塑料盖是可以归到可回收垃圾的嗷

看到这里,是不是大家突然都不想喝奶茶了呢,哈哈。不过不要紧,垃圾分类虽然要执行,但是奶茶也可以照喝。

那么,这里我们想讨论一下,人工智能和数据科学的方法能不能帮助我们进行更好的垃圾分类?这样我们不用为了不知道要扔哪个垃圾箱而烦恼。

2 思路

这问题的解决思路或许不止一条。这里只是抛砖引玉一下,提供一些浅显的见解。

第一种方案,可以把垃圾的信息制成表格化数据,然后用传统的机器学习方法。

第二种方案,把所有的垃圾分类信息做成知识图谱,每一次的查询就好像是在翻字典一样查阅信息。

第三种方案,可以借助现在的深度学习方法,来对垃圾进行识别和分类。每次我们给一张垃圾的图片,让模型识别出这是属于哪一种类别的:干垃圾,湿垃圾,有害垃圾还是可回收垃圾。

使用Python轻松完成垃圾分类(基于图像识别)

3 图像分类

图像分类是深度学习的一个经典应用。它的输入是一张图片, 然后经过一些处理,进入一个深度学习的模型,该模型会返回这个图片里垃圾的类别。这里我们考虑四个类别:干垃圾,湿垃圾,有害垃圾还是可回收垃圾。

报纸 :可回收垃圾

使用Python轻松完成垃圾分类(基于图像识别)

电池 :有害垃圾

使用Python轻松完成垃圾分类(基于图像识别)

一次性餐盒 :干垃圾

使用Python轻松完成垃圾分类(基于图像识别)

我们对图片里的物品进行分类,这是图像处理和识别的领域。人工智能里提出了使用卷积神经网络(Convolutional Neural Network, CNN)来解决这一类问题。

我会用keras包和Tensorflow后端来建立模型。 由于训练集的样本暂时比较缺乏,所以这里只能先给一套思路和代码。训练模型的工作之前还得进行一波数据收集。

我们就先来看看代码大致长什么样吧

先导入一些必要的包。

使用Python轻松完成垃圾分类(基于图像识别)

再做一下准备工作。

使用Python轻松完成垃圾分类(基于图像识别)

在上面,我们初始化了一些变量,batch size是128; num_classes = 4,因为需要分类的数量是4,有干垃圾,湿垃圾,有害垃圾和可回收垃圾这四个种类。epochs 是我们要训练的次数。接下来,img_rows, img_cols = 28, 28 我们给了图片的纬度大小。

在 .reshape(60000,28,28,1)中 , 60000 是图片的数量(可变), 28是图片的大小(可调),并且1是channel的意思,channel = 1 是指黑白照片。 .reshape(10000,28,28,1)也是同理,只是图片数量是10000。

到了最后两行,我们是把我们目标变量的值转化成一个二分类, 是用一个向量(矩阵)来表示。比如 [1,0,0,0] 是指干垃圾,[0,1,0,0]是指湿垃圾等等。

接下来是建模的部分。

使用Python轻松完成垃圾分类(基于图像识别)

我们加了卷积层和池化层进入模型。激活函数是 relu,relu函数几乎被广泛地使用在了卷积神经网络和深度学习。我们在层与层之间也加了dropout来减少过拟合。Dense layer是用来做类别预测的。

建完模型后,我们要进行模型的验证,保证准确性在线。

使用Python轻松完成垃圾分类(基于图像识别)

到这里,我们的建模预测已经大概完成了。一个好的模型,要不断地去优化它,提高精确度等指标要求,直到达到可以接受的程度。

这优化的过程,我们在这里就先不深入讨论了,以后继续。

4 总结

值得一提的是,尽管方法上是有实现的可能,但是实际操作中肯定要更复杂的多,尤其是对精度有着很高的要求。

而且当一个图片里面包含着好几种垃圾种类,这也会让我们的分类模型开发变得很复杂,增加了难度。

比如,我们想要对一杯奶茶进行垃圾分类,照片里面是包含了多个垃圾的种类,这就比较头大了,因为这并不是属于单一的类别。

前路的困难肯定是有的,不过就当这里的分享是个抛砖引玉的起点吧。

毕竟李白也说了,“长风破浪会有时,直挂云帆济沧海”。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
全面解析Python的While循环语句的使用方法
Oct 13 Python
python中pandas.DataFrame排除特定行方法示例
Mar 12 Python
Zookeeper接口kazoo实例解析
Jan 22 Python
浅谈pandas中Dataframe的查询方法([], loc, iloc, at, iat, ix)
Apr 10 Python
python matlibplot绘制多条曲线图
Feb 19 Python
解决Python3 被PHP程序调用执行返回乱码的问题
Feb 16 Python
python实现字符串加密 生成唯一固定长度字符串
Mar 22 Python
详解Python中正则匹配TAB及空格的小技巧
Jul 26 Python
python图像处理模块Pillow的学习详解
Oct 09 Python
浅谈tensorflow模型保存为pb的各种姿势
May 25 Python
社区版pycharm创建django项目的方法(pycharm的newproject左侧没有项目选项)
Sep 23 Python
教你怎么用Python操作MySql数据库
May 31 Python
Python分析彩票记录并预测中奖号码过程详解
Jul 09 #Python
python求最大值,不使用内置函数的实现方法
Jul 09 #Python
pandas的连接函数concat()函数的具体使用方法
Jul 09 #Python
python爬虫的一个常见简单js反爬详解
Jul 09 #Python
详解PANDAS 数据合并与重塑(join/merge篇)
Jul 09 #Python
python 返回一个列表中第二大的数方法
Jul 09 #Python
Mac在python3环境下安装virtualwrapper遇到的问题及解决方法
Jul 09 #Python
You might like
成本8450万,票房仅2亿,口碑两极分化,又一部DC电影扑街了
2020/04/09 欧美动漫
PHP使用CURL实现对带有验证码的网站进行模拟登录的方法
2014/07/23 PHP
Yii2创建表单(ActiveForm)方法详解
2016/07/23 PHP
PHP实现的XXTEA加密解密算法示例
2018/08/28 PHP
thinkphp5.1框架实现格式化mysql时间戳为日期的方式小结
2019/10/10 PHP
Laravel服务容器绑定的几种方法总结
2020/06/14 PHP
javascript打印html内容功能的方法示例
2013/11/28 Javascript
jquery选择器之基本过滤选择器详解
2014/01/27 Javascript
js利用prototype调用Array的slice方法示例
2014/06/09 Javascript
jquery实现倒计时代码分享
2014/06/13 Javascript
javascript实现的平方米、亩、公顷单位换算小程序
2014/08/11 Javascript
JS实现适合于后台使用的动画折叠菜单效果
2015/09/21 Javascript
谈谈我对JavaScript原型和闭包系列理解(随手笔记9)
2015/12/24 Javascript
jquery遍历函数siblings()用法实例
2015/12/24 Javascript
Javascript实现的SHA-256加密算法完整实例
2016/02/02 Javascript
谈一谈javascript中继承的多种方式
2016/02/19 Javascript
微信小程序实现YDUI的ScrollTab组件
2018/02/02 Javascript
AngularJS实时获取并显示密码的方法
2018/02/06 Javascript
使用Angular CLI快速创建Angular项目的一些基本概念和写法小结
2018/04/22 Javascript
详解通过源码解析Node.js中cluster模块的主要功能实现
2018/05/16 Javascript
vue 对axios get pust put delete封装的实例代码
2020/01/05 Javascript
vue-cli设置publicPath小记
2020/04/14 Javascript
vue-video-player 断点续播的实现
2021/02/01 Vue.js
python改变日志(logging)存放位置的示例
2014/03/27 Python
Python实现简单的HttpServer服务器示例
2017/09/25 Python
Windows下Anaconda的安装和简单使用方法
2018/01/04 Python
python requests更换代理适用于IP频率限制的方法
2019/08/21 Python
sealed修饰符是干什么的
2012/10/23 面试题
高一自我鉴定
2013/12/17 职场文书
老公给老婆的道歉信
2014/01/10 职场文书
党员2014两会学习心得体会
2014/03/17 职场文书
教师新年寄语
2014/04/03 职场文书
2014学校领导四风对照检查材料思想汇报
2014/09/23 职场文书
购房个人委托书范本
2014/10/11 职场文书
军人离婚协议书样本
2014/10/21 职场文书
赵氏孤儿观后感
2015/06/09 职场文书