一种垃圾图像分类方法及系统.pdfVIP

  • 3
  • 0
  • 约1.29万字
  • 约 12页
  • 2023-08-23 发布于四川
  • 举报
本发明公开了一种垃圾图像分类方法及系统,所述方法包括以下步骤:步骤1、获取垃圾数据集,将垃圾数据集分为训练集和测试集;步骤2、改进YOLOv5模型:使用CBAM注意力机制作为特征提取模块、使用BiFPN结构作为特征融合结构、使用边界框损失函数LCIoU;步骤3、利用K‑means聚类方法得到训练集的先验锚框,利用训练集的先验锚框、边界框损失函数LCIoU对改进的YOLOv5模型进行训练;步骤4、利用训练好的YOLOv5模型对测试集进行检测,输出垃圾目标检测结果,得到正确的垃圾类别。通过本发明方法

(19)国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 116630604 A (43)申请公布日 2023.08.22 (21)申请号 202310404270.8 G06V 10/82 (2022.01)

文档评论(0)

1亿VIP精品文档

相关文档