融合双维度注意力和残差网络的道路垃圾检测方法

    公开(公告)号:CN117710935A

    公开(公告)日:2024-03-15

    申请号:CN202311616241.4

    申请日:2023-11-28

    Abstract: 本发明涉及一种融合双维度注意力和残差网络的道路垃圾检测方法,属于目标检测技术领域。该方法包括:采集道路垃圾图像制作模型构建数据集,其中模型构建数据集中包含至少一个道路垃圾目标的图像;基于YOLOv5s的网络结构,引入Res2Net网络,融合注意力机制并将损失函数替换为EIOU得到改进的道路垃圾检测模型;使用模型构建数据集对改进的道路垃圾检测模型进行训练;将训练好的道路垃圾检测模型部署到巡检机器人中,进行道路垃圾检测,并将检测结果输出至巡检机器人显示端。本发明可在降低模型参数量的同时以更细的粒度提取多尺度特征,能表现出更好的鲁棒性和准确性。

Patent Agency Ranking