Invention Publication
- Patent Title: 一种图像分类模型的训练方法、装置及电子设备
-
Application No.: CN202410400212.2Application Date: 2024-04-03
-
Publication No.: CN117994611APublication Date: 2024-05-07
- Inventor: 沈西 , 杨再初 , 李昱廷 , 林容泰 , 黄世华
- Applicant: 英特灵达信息技术(深圳)有限公司 , 北京英特灵达信息技术有限公司
- Applicant Address: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室;
- Assignee: 英特灵达信息技术(深圳)有限公司,北京英特灵达信息技术有限公司
- Current Assignee: 英特灵达信息技术(深圳)有限公司,北京英特灵达信息技术有限公司
- Current Assignee Address: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室;
- Agency: 北京柏杉松知识产权代理事务所
- Agent 马敬; 项京
- Main IPC: G06V10/774
- IPC: G06V10/774 ; G06V10/764 ; G06N3/0464 ; G06N3/088

Abstract:
本申请实施例提供了一种图像分类模型的训练方法、装置及电子设备,涉及计算机视觉技术领域,本申请实施例包括:针对每两张样本图像,对该两张样本图像按照指定比例混合,得到混合图像,并对该两张样本图像的训练标签按照指定比例混合,得到混合图像的混合标签。再将样本图像和混合图像分别输入图像分类网络,之后基于图像分类网络输出的样本图像所属的类别和训练标签,确定样本损失值,并基于图像分类网络输出的混合图像所属的类别和混合标签,确定混合损失值。再基于样本损失值和混合损失值,调整图像分类网络的网络参数,直至图像分类网络收敛时,将当前的图像分类网络作为图像分类模型。能够提高图像分类的准确度。
Public/Granted literature
- CN117994611B 一种图像分类模型的训练方法、装置及电子设备 Public/Granted day:2024-07-02
Information query