图像分类方法及装置
摘要:
本公开涉及计算机技术领域,提供了一种图像分类方法及装置。该方法包括:获取待分类图像;将待分类图像输入到预置的图像分类模型,采用图像分类模型对待分类图像进行分类,得到分类结果,其中,图像分类模型采用训练图片、初始分类模型、第一损失函数和第二损失函数训练得到,第一损失函数根据第一任务数据构建,第一任务数据为对进行随机掩码处理后的分片执行对比学习任务的任务数据,第二损失函数根据第二任务数据构建,第二任务数据为对进行随机掩码处理后的分片输入初始分类模型执行掩码图像还原任务的任务数据。本公开的技术方案可以提高图像分类模型的图像分类的精度。
公开/授权文献
0/0