公开(公告)号：CN116630604A

公开(公告)日：2023-08-22

申请号：CN202310404270.8

申请日：2023-04-14

Applicant: 无锡学院

Inventor： 孙可盈 ,  刘政达 ,  孙家豪 ,  韩锦煜 ,  李守娟 ,  李妮 ,  刘佳晶 ,  朱硕

IPC: G06V10/25 ,  G06V10/44 ,  G06V10/774 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464

Abstract: 本发明公开了一种垃圾图像分类方法及系统，所述方法包括以下步骤：步骤1、获取垃圾数据集，将垃圾数据集分为训练集和测试集；步骤2、改进YOLOv5模型：使用CBAM注意力机制作为特征提取模块、使用BiFPN结构作为特征融合结构、使用边界框损失函数LCIoU；步骤3、利用K‑means聚类方法得到训练集的先验锚框，利用训练集的先验锚框、边界框损失函数LCIoU对改进的YOLOv5模型进行训练；步骤4、利用训练好的YOLOv5模型对测试集进行检测，输出垃圾目标检测结果，得到正确的垃圾类别。通过本发明方法可以提高垃圾识别准确率。