基于双分支网络的户外垃圾检测识别

2024-05-25 电子科技360 1.49M 0

  摘要:现有户外垃圾检测算法在特征提取方面没有充分考虑CNN(Convolutional Neural Network)和Transformer的优缺点,限制了网络的整体性能。文中提出了一种由CNN和Transformer组成的双分支融合网络检测算法。在编码阶段,基于CNN和Transformer的优点构建了双分支主干网络来对原始图像的特征信息进行提取。通过多尺度卷积模块和多尺度池化模块消除所提取特征信息在维度和语义上的差异,通过加强特征提取网络减少深层神经网络中细节信息的丢失。收集了6个类别的户外常见垃圾图片,建立了一个背景复杂的垃圾图片数据集来验证所提算法在户外垃圾检测识别任务上的性能。实验结果表明,相比最新目标检测算法,所提算法在该数据集上的平均准确率(mean Average Precision, mAP)提升约5%。为了验证所提算法的泛化性能,在华为垃圾分类挑战杯数据集上进行了泛化实验,实验结果表明所提算法相比最新目标检测算法mAP提升约2%。

  文章目录

  1 双分支融合网络

  1.1 网络总体结构

  1.2 CNN-Transformer融合网络

  1.3 特征交互模块

  1.4 加强特征提取网络

  2 数据集与评价指标

  2.1 数据集建立

  2.2 模型训练细节与评测指标

  3 实验分析

  3.1 消融实验

  3.2 对比实验

  3.3 泛化实验

  4 结束语



您还没有登录,请登录后查看详情



 

1/26专辑:论文下载

举报收藏 0打赏 0评论 0
相关文档
本类推荐
下载排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  版权声明  |  网站地图  |  排名推广  |  广告服务  |  积分换礼  |  网站留言  |  RSS订阅  |  违规举报  |  蜀ICP备2024057410号-1