一种图像识别模型训练方法、装置、设备及介质

基本信息

申请号 CN202111666239.9 申请日 -
公开(公告)号 CN114359547A 公开(公告)日 2022-04-15
申请公布号 CN114359547A 申请公布日 2022-04-15
分类号 G06V10/25(2022.01)I;G06V10/774(2022.01)I;G06V10/74(2022.01)I;G06V10/82(2022.01)I;G06N3/04(2006.01)I 分类 计算;推算;计数;
发明人 刘冰;百晓;陈科;薛姬荣;李军;周雷;王晨;张亮;江金寿;田建辉;王晓悦;叶金华;安辰 申请(专利权)人 中国兵器科学研究院
代理机构 北京同达信恒知识产权代理有限公司 代理人 王丽莉
地址 100089北京市海淀区车道沟十号院中国兵器科学研究院
法律状态 -

摘要

摘要 本申请实施例提供了一种图像识别模型训练方法、装置、设备及介质,由于本申请中,原始图像识别模型输出了每个样本图像对中参考图像样本对应的第一预估图像及第二预估图像,并分别根据第一预估图像中对应像素点的像素值确定了第一子损失值,根据第二预估图像中对应像素点的像素值,确定第二损失值,进行了像素级对比学习,还输出了每个样本图像对中参考图像样本对应的第一特征向量及每个样本图像对中目标图像样本对应的第二特征向量,并分别根据第一特征向量以及第二特征向量,确定第三子损失值,进行了图像级对比学习,由于进行了像素级对比和图像级对比学习,提高了图像识别模型的精度,及图像识别模型提取出的图像中物体特征的显著性或区分性。