骨干网络的训练方法以及图像分类方法
摘要:
本公开提供了一种骨干网络的训练方法以及图像分类方法,涉及人工智能技术领域,具体涉及深度学习和计算机视觉技术。该方法包括:获取训练图像集;利用初始骨干网络提取训练图像集中各个训练图像的特征;分别计算训练图像的特征与该训练图像对应的类别代理向量之间的对比损失值;基于对比损失值反向更新初始骨干网络的参数,得到更新后的骨干网络。本公开提供的骨干网络的训练方法降低了过拟合风险,提升了骨干网络的训练效率和骨干网络的精度。
公开/授权文献
0/0