APP下载

基于ResNet50的列车常见垃圾自动分类算法

2021-07-28张丽艳赵艺璇牟钰鹏李林

大连交通大学学报 2021年4期
关键词:分类模型

张丽艳,赵艺璇,牟钰鹏,李林

(大连交通大学 电气信息工程学院,辽宁 大连 116028) *

党的十八大以来,党和国家对生态文明日益重视,将生态文明列入“五位一体”的总体布局,而垃圾分类则是生态文明建设的重要环节和关键领域.目前垃圾分类设施过于简单,为了提高分类效率,避免发生错分、误分的现象,智能垃圾分类产品成为当今研究重点[1].美国CleanRobotics公司研制了一款名为Transhbot的全自动分类垃圾桶.Trashbot操纵摄像头、传感器、金属探测器和马达,把垃圾放在精确的位置.法国Uzer公司研发了一款名为Eugene的智能垃圾箱,只需把丢弃的垃圾上的条形码对准扫描仪就可识别出垃圾是何种类.吴凡[2]等研究人员发明的智能垃圾桶只需扫描垃圾袋上的二维码,所属类别垃圾桶的盖子会自动打开,指引投放.但以上研究对于列车上常见垃圾的分类并不适用,为此本文基于ResNet50神经网络模型提出一种列车垃圾分类器,实现垃圾智能分类.

1 垃圾自动分类算法模型设计

深度学习是机器学习领域中最重要的研究方向之一,是机器模拟人脑对文字、图像、声音等数据进行一系列分析,从而解决模式识别中的困难问题[3].通常情况下,具有良好分类能力的模型需要经过大量数据集训练,消耗巨大的时间资源和计算资源,因此研究者提出迁移学习方法来解决此类问题[4-5].本文使用在ImageNet上进行过预训练的ResNet50模型对列车常见12种垃圾进行分类识别.

1.1 迁移学习

迁移学习(Transfer Learning)是一种机器……

登录APP查看全文

猜你喜欢

分类模型
一半模型
分类算一算
垃圾分类的困惑你有吗
重尾非线性自回归模型自加权M-估计的渐近分布
教你一招:数的分类
3D打印中的模型分割与打包
FLUKA几何模型到CAD几何模型转换方法初步研究
给塑料分分类吧
一个相似模型的应用