基于yolov4的垃圾检测系统

作者：王明吉; 陈秋梦; 任福深^*; 胡庆; 刘博; 王一安

来源：工业仪表与自动化装置, 2021, (05): 20-72.

DOI：10.19950/j.cnki.cn61-1121/th.2021.05.005

• 垃圾分类
• 定位
• k-means++
• 摄像头

为解决垃圾分类困难问题,根据大量垃圾图片的特点,提出了基于yolov4的垃圾检测的方法,实现垃圾的自动识别分类和定位。该方法通过手机拍摄以及百度等收集不同种类的垃圾图像集,运用图像增强的方式进行图像预处理,再利用迁移学习模型和k-means++聚类算法进行垃圾网络分类模型的训练测试,最后将训练好的模型通过摄像头对采集的垃圾图片实时识别分类和定位。实验表明:该方法的m AP达到95.15%,能够快速有效地对垃圾分类和定位。

• 单位
  电子工程学院; 东北石油大学