非均衡小样本目标检测方法、装置和计算机设备

基本信息

申请号 CN202110721044.3 申请日 -
公开(公告)号 CN113255670A 公开(公告)日 2021-08-13
申请公布号 CN113255670A 申请公布日 2021-08-13
分类号 G06K9/32(2006.01)I;G06K9/46(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 郭建京;索红亮;周忠诚;段炼;张圣栋;黄九鸣 申请(专利权)人 湖南星汉数智科技有限公司
代理机构 - 代理人 -
地址 410000湖南省长沙市长沙高新开发区青山路699号湖南省军民融合科技创新产业园8栋B座19层
法律状态 -

摘要

摘要 本发明涉及计算机视觉技术领域,提供了一种非均衡小样本目标检测方法、装置和计算机设备,所述方法包括:构建敏感标志图像数据集,计算敏感标志图像数据集中各类型敏感标志图像的数据分布权重;根据数据分布权重确定敏感标志图像数据集中各类型敏感标志图像对应的均衡化数量;基于各均衡化数量,分别对敏感标志图像数据集中各类型敏感标志图像进行数据均衡化,得到均衡敏感数据集;基于均衡敏感数据集对预训练模型进行训练,得到初始目标检测模型;对初始目标检测模型进行迭代的小样本学习,得到最终的目标检测模型,利用目标检测模型对包含敏感标志的图像进行检测。采用本方法能够提高非均衡小样本目标检测的准确率。