一种基于改进Cswin Transformer模型的垃圾分类方法

本发明公开了一种垃圾的自动识别方法，所述方法为：通过摄像头和互联网搜索获取垃圾的图像，对图片进行数据增强预处理；在Cswin Transformer模型中引入Agent注意力模块，增强模型对特征的提取能力；采用LionBatch优化策略，提高模型训练速度和资源利用效率；采用标签平滑交叉熵损失函数对困难样本进行动态加权；将预处理后的图像数据输入到改进模型中进行训练与测试，获得最优模型；最后将待分类图像输入到最优模型中获取分类结果。本发明提出的一种基于改进Transformer的垃圾的自动识别方法，可确保对垃圾自动识别并获得正确类型，具有良好的应用前景。