通道顺序切换自监督提升长尾分布视觉识别能力的方法

摘要

本发明涉及计算机视觉领域，具体是通道顺序切换自监督提升长尾分布视觉识别能力的方法，包括两个阶段，自监督训练和有监督训练。准备数据集；将数据集中的图片进行预处理，包括对图片进行重采样处理和对图片进行随机通道顺序的变换；将预处理后的图片用于通道切换自监督训练，将通道顺序名称作为真实的标签计算损失函数，不断迭代网络直至收敛，保存模型；将数据集中的图片进行预处理，包括对图片进行重采样处理和对图片进行数据增强；初始化有监督训练网络，将保存的模型作为有监督训练过程的预训练模型，将预处理后的图片输入模型中进行训练；不断迭代训练网络，计算损失函数，直至模型达到预期的识别分类效果；解决了长尾分布带来的问题。