基于yolov4的垃圾检测系统 |
| |
作者单位: | 东北石油大学 物理与电子工程学院,黑龙江 大庆163318;东北石油大学 机械工程学院,黑龙江 大庆163318 |
| |
摘 要: | 为解决垃圾分类困难问题,根据大量垃圾图片的特点,提出了基于yolov4的垃圾检测的方法,实现垃圾的自动识别分类和定位。该方法通过手机拍摄以及百度等收集不同种类的垃圾图像集,运用图像增强的方式进行图像预处理,再利用迁移学习模型和k-means++聚类算法进行垃圾网络分类模型的训练测试,最后将训练好的模型通过摄像头对采集的垃圾图片实时识别分类和定位。实验表明:该方法的m AP达到95.15%,能够快速有效地对垃圾分类和定位。
|
关 键 词: | 垃圾分类 定位 k-means++ 摄像头 |
本文献已被 CNKI 万方数据 等数据库收录! |
|