基于Swin Transformer的知识蒸馏模型在垃圾分类上的应用

  • 打印
  • 收藏
收藏成功


打开文本图片集

摘要:在绿色发展的主题下,深度学习高效地进行垃圾分类方法亟待研究。在垃圾图像分类问题中,大尺寸图像包含着更丰富的信息。然而主流的基于CNN的深度学习难以从大尺寸图片中提取长序列特征。针对此问题,文章基于垃圾图像数据设计了基于Swin Transformer模型的算法框架。该算法框架不仅解决了长序列特征提取问题,还通过知识蒸馏方法解决了大模型体积过大难以部署的问题。(剩余9413字)

目录
monitor