垃圾分类模型建模方法、垃圾分类方法、装置及介质

摘要

本发明涉及一种垃圾分类模型建模方法、垃圾分类方法、装置及介质。基于自编码器重构误差与置信度概念，首先在只包含ID(In Distribution)样本的垃圾分类数据集上训练模型的主干网络及其串联的全连接网络，然后固定主干网络及全连接网络的连接权重，在只包含ID样本的垃圾分类数据集上并行训练自编码器；最后固定主干网络及其串联的全连接网络、自编码器的连接权重，在包含ID和OOD(Out of Distribution)样本的垃圾分类数据集上并行训练自编码器串联的全连接网络层。该方法训练出来的模型既能够准确分类已知类别垃圾也可以精准辨别未知类别垃圾。有利于解决垃圾分类任务中数据中存在异常样本而导致的分类模型性能不好的问题。