- 专利标题: 一种图像分类模型的训练方法、装置及电子设备
-
申请号: CN202410400212.2申请日: 2024-04-03
-
公开(公告)号: CN117994611A公开(公告)日: 2024-05-07
- 发明人: 沈西 , 杨再初 , 李昱廷 , 林容泰 , 黄世华
- 申请人: 英特灵达信息技术(深圳)有限公司 , 北京英特灵达信息技术有限公司
- 申请人地址: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室;
- 专利权人: 英特灵达信息技术(深圳)有限公司,北京英特灵达信息技术有限公司
- 当前专利权人: 英特灵达信息技术(深圳)有限公司,北京英特灵达信息技术有限公司
- 当前专利权人地址: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室;
- 代理机构: 北京柏杉松知识产权代理事务所
- 代理商 马敬; 项京
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/764 ; G06N3/0464 ; G06N3/088
摘要:
本申请实施例提供了一种图像分类模型的训练方法、装置及电子设备,涉及计算机视觉技术领域,本申请实施例包括:针对每两张样本图像,对该两张样本图像按照指定比例混合,得到混合图像,并对该两张样本图像的训练标签按照指定比例混合,得到混合图像的混合标签。再将样本图像和混合图像分别输入图像分类网络,之后基于图像分类网络输出的样本图像所属的类别和训练标签,确定样本损失值,并基于图像分类网络输出的混合图像所属的类别和混合标签,确定混合损失值。再基于样本损失值和混合损失值,调整图像分类网络的网络参数,直至图像分类网络收敛时,将当前的图像分类网络作为图像分类模型。能够提高图像分类的准确度。
公开/授权文献
- CN117994611B 一种图像分类模型的训练方法、装置及电子设备 公开/授权日:2024-07-02