改进卷积神经网络算法在水面漂浮垃圾检测中的应用  

Application of improved convolutional neural network algorithm in detection of floating garbage in water

作　　者：汤伟[1,2] 高涵 TANG Wei;GAO Han(College of Electrical and Information Engineering,Shaanxi University of Science and Technology,Xi’an 710021,China;Institute of industrial automation,Shaanxi University of Science and Technology,Xi’an 710021,China)

机构地区：[1]陕西科技大学电气与信息工程学院,西安710021 [2]陕西科技大学工业自动化研究所,西安710021

出　　处：《中国科技论文》2019年第11期1210-1216,共7页China Sciencepaper

基　　金：陕西省重点科技创新团队计划项目(2014KCT-15);陕西省科技统筹创新工程计划项目(2012KTCQ01-19)

摘　　要：针对传统目标检测算法在检测水面漂浮垃圾时易受外界复杂环境影响而难以实现的问题,提出了一种基于改进卷积神经网络的水面漂浮垃圾检测算法。运用数据增强技术改善训练过程中因样本不足而导致的过拟合问题,并利用迁移学习的方法训练出水面漂浮垃圾目标检测模型。结果表明,与传统的ViBe背景建模算法相比,所提算法能对水面漂浮垃圾进行分类,并标记出垃圾位置,对垃圾的检测准确率高达93%,能完全克服水波、波光等外界干扰。In order to solve the problem that the traditional target detection algorithm is easy to be affected by the complex environment,an improved convolutional neural network algorithm was proposed.Data enhancement technology was used in the algorithm to overcome the over-fitting problem caused by insufficient samples in the training process.At the same time,the migration learning method was used to train the surface floating garbage target detection model.The results show that the algorithm can accurately detect the types of floating garbage on the surface and mark the location of garbage,compared with the traditional ViBe algorithm.The detection accuracy of garbage is up to 93%,and it completely overcomes external disturbances,such as water waves and sunlight.

关 键 词：卷积神经网络 数据增强 迁移学习 漂浮垃圾检测 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]