返回搜索
档案
一种用于X光违禁品小样本增量检测的模型构建方法
发明专利有效
3浏览
10权利要求 · 4 独立
§ Ⅰ
卷宗概要
发明人
付彬; 肖纪伟; 高成志; 王瑞平; 陈熙霖
IPC 分类
G06V 10/774 (2022.01)G06V 10/25 (2022.01)G06V 10/44 (2022.01)G06V 10/764 (2022.01)G06V 20/60 (2022.01)
本发明提供一种用于X光违禁品小样本增量检测的模型构建方法,包括:基类检测模型训练:构建初始模型,包括特征提取器、候选框生成器、候选框重构器和包围框检测器,特征提取器包含提取X光图像的特征图的多个特征通道;获取基类训练集,基类包括预设的多类违禁品,每类有多张X光图像,利用该训练集预训练初始模型得到基类检测模型;增量训练:获取多个新类训练集,每个新类训练集包括一个或多个与基类训练集中的类别不同的违禁品,每类有多张X光图像且满足预设的小样本数量;采用多个新类训练集对基类检测模型进行多次增量训练,每次增量训练在上一次的基础上进行,且每次冻结检测模型中已完成训练的特征通道并扩展多个特征通道后进行训练。