[发明专利]基于深度图像及多光谱图像融合的垃圾分类方法及装置

摘要

本发明属于垃圾分类技术领域，尤其涉及一种基于深度图像及多光谱图像融合的垃圾分类方法及装置，所述方法包括：获取待分类垃圾的垃圾图像数据集；垃圾图像数据集包括待分类垃圾深度图像和待分类垃圾多光谱图像；将待分类垃圾深度图像和待分类垃圾多光谱图像作图像配准处理；对配准后垃圾深度图像和配准后垃圾多光谱图像作分类标注处理；将标注后垃圾深度图像和标注后垃圾多光谱图像加载至基于Pytorch框架的YOLOv4神经网络中作模型训练，并生成最优神经网络模型；根据最优神经网络模型生成分选结果。本发明实现了对垃圾实现自动评分以及自动分类，输出一个最大概率的类别标签，进而实现对垃圾的高精准度和高效率分类，满足垃圾分类的生产需求。