← 返回列表
一种标签优化方法、装置、存储介质及电子设备
申请人信息
- 申请人:之江实验室
- 申请人地址:311121 浙江省杭州市余杭区中泰街道科创大道之江实验室
- 发明人: 之江实验室
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种标签优化方法、装置、存储介质及电子设备 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410194170.1 |
| 申请日 | 2024/2/21 |
| 公告号 | CN117746193A |
| 公开日 | 2024/3/22 |
| IPC主分类号 | G06V10/774 |
| 权利人 | 之江实验室 |
| 发明人 | 窦宝成; 马嘉; 兰昆艳; 任祖杰; 施航 |
| 地址 | 浙江省杭州市余杭区中泰街道科创大道之江实验室 |
摘要文本
本说明书公开了一种标签优化方法、装置、存储介质及电子设备。所述标签优化方法包括:获取样本图像的各基础标签,将样本图像和各基础标签输入到预先训练的细粒度遥感目标检测模型中,得到样本图像的预测标签以及预测标签对应的置信度,针对每个基础标签,根据该基础标签和该基础标签对应的匹配标签集中的预测标签之间一致程度,以及,该基础标签对应的匹配标签集中包含的各预测标签的置信度,确定该基础标签对应的质量权重,以根据质量权重从各基础标签中筛选出待优化基础标签,并根据匹配标签集对待优化基础标签进行优化,得到优化后标签,根据优化后标签以及其他基础标签,对细粒度遥感目标检测模型进行训练。
专利主权项内容
1.一种标签优化方法,其特征在于,所述方法用于训练细粒度遥感目标检测模型,所述方法包括:获取样本图像的各基础标签,所述基础标签包括:所述样本图像中包含的目标物的类别,所述样本图像中所述目标物对应的目标框的角点位置;将所述样本图像和各基础标签输入到预先训练的细粒度遥感目标检测模型中,得到所述样本图像的预测标签以及所述预测标签对应的置信度;根据每个预测标签中目标框的各角点位置和各基础标签中目标框的各角点位置之间的位置关系,从各预测标签中确定出每个基础标签对应的匹配标签集;针对每个基础标签,根据该基础标签和该基础标签对应的匹配标签集中的预测标签之间一致程度,以及,该基础标签对应的匹配标签集中包含的各预测标签的置信度,确定该基础标签对应的质量权重;根据每个基础标签对应的质量权重,从各基础标签中筛选出待优化基础标签,并根据所述待优化基础标签对应的匹配标签集中的各预测标签对所述待优化基础标签进行优化,得到优化后标签,根据所述优化后标签以及其他基础标签,对所述细粒度遥感目标检测模型进行训练。