Invention Publication
- Patent Title: 一种目标检测模型的小样本训练方法及装置
-
Application No.: CN202311186407.3Application Date: 2023-09-14
-
Publication No.: CN117351300APublication Date: 2024-01-05
- Inventor: 高顺利 , 崔涛 , 王亮 , 姜海 , 朱妍 , 马人杰 , 王一君 , 李夏喜 , 邓井川
- Applicant: 北京市燃气集团有限责任公司
- Applicant Address: 北京市西城区西直门南小街22号
- Assignee: 北京市燃气集团有限责任公司
- Current Assignee: 北京市燃气集团有限责任公司
- Current Assignee Address: 北京市西城区西直门南小街22号
- Agency: 北京天方智力知识产权代理有限公司
- Agent 路远
- Main IPC: G06V10/774
- IPC: G06V10/774 ; G06V10/40 ; G06V10/764

Abstract:
本发明提供一种目标检测模型的小样本训练方法及装置。所述方法包括以下步骤:建立由数量较少的标注后的图像样本数据构成的训练数据集D;采用ImageNet数据集对目标检测模型进行预训练,得到预训练模型;将D输入到预训练模型进行特征提取,并对所提取的特征进行扩展;基于扩展后的特征对预训练模型进行微调,得到训练后的目标检测模型。本发明通过特征扩展,并基于扩展后的特征对预训练模型进行微调,可解决由于标签样本数据少而导致的目标检测模型出现过拟合的问题,提高目标检测精度。
Public/Granted literature
- CN117351300B 一种目标检测模型的小样本训练方法及装置 Public/Granted day:2024-05-14
Information query