一种基于集成并行注意力模块的垃圾分拣方法

    公开(公告)号:CN118629020A

    公开(公告)日:2024-09-10

    申请号:CN202410623760.1

    申请日:2024-05-20

    Abstract: 本发明公开了一种基于集成并行注意力模块的垃圾分拣方法。该方法利用集成并行注意力模块构建可回收垃圾分拣网络,可回收垃圾分拣网络利用通道‑空间注意力提取垃圾特征图的关键信息,实现可回收垃圾图像的自动识别与分类;可回收垃圾分拣网络的具体构建方法为:基于层归一化搭建通道注意力模块,捕捉不同通道之间的相互依赖关系;基于实例归一化搭建空间注意力模块,捕捉特征图在空间维度上的变化;将通道注意力模块和空间注意力模块并行连接,即可得到集成并行注意力模块;将集成并行注意力模块插入卷积神经网络,得到所述可回收垃圾分拣网络;搭建TrashZJ数据集可回收垃圾图像数据集用于网络的训练和测试。该方法可有效提高垃圾处理的效率和准确性。

Patent Agency Ranking