当前位置: 首页 > 专利查询>启元实验室专利>正文

一种小目标检测模型的训练方法和系统技术方案

技术编号:41246029 阅读:12 留言:0更新日期:2024-05-09 23:56
本发明专利技术公开了小目标检测模型的训练方法,包括:构建小目标识别深度神经网络结构。训练骨干网络层识别小目标检测样本图像中的目标图像特征。通过不确定学习方法训练检测头层,根据目标图像特征得到目标分类信息,以识别测试图片得到目标检测结果。本发明专利技术通过设定的深度神经网络架构结合不确定学习方法,得到了识别准确度、可靠性高的小目标检测模型,其训练智能体控制网络模型的层数设计紧凑、精简,所需模型的计算量、参数量小,整套算法耗时少,易部署在微纳平台上。解决了现有小目标识别在部署于微纳平台时,容易出现误识、漏识和算力限制等问题。同时,本发明专利技术提供了小目标检测模型的训练系统。

【技术实现步骤摘要】

本专利技术涉及目标识别和跟踪的。特别应用于无人机等拍摄设备的目标图像采集,本专利技术具体涉及一种小目标检测模型的训练方法和系统


技术介绍

1、针对小目标的目标检测算法,因检测图像所占像素数较少,携带有效信息少,因此检测精度低且对复杂背景下目标适应能力弱,容易发生漏警和虚警的结果。特别是在实施于微纳量级无人机时,因为受到机身重量的限制,可配备的处理器存储和计算资源受限。加之计算量、参数量和耗时的限制,其小目标识别功能难以实施于小型机身且无法保证识别精度。


技术实现思路

1、本专利技术的目的是提供一种小目标检测模型的训练方法,通过深度神经网络架构结合不确定学习方法,得到了识别准确度、可靠性高的小目标检测模型,解决了现有小目标识别时,容易出现误识、漏识和算力限制等问题。

2、为解决上述问题,本专利技术的第一方面,提供了一种小目标检测模型的训练方法,包括:

3、构建小目标识别深度神经网络结构。小目标识别深度神经网络结构包括骨干网络层和检测头层。

4、将多幅小目标检测样本图像输入小目本文档来自技高网...

【技术保护点】

1.一种小目标检测模型的训练方法,其特征在于,包括:

2.根据权利要求1所述的小目标检测模型的训练方法,其特征在于,还包括:

3.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,得到所述小目标检测样本图像包括,模糊处理所述小目标检测样本图像或减少像素值处理所述小目标检测样本图像。

4.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,平面增强处理所述小目标检测图像包括,通过旋转、剪裁、缩放、水平翻转和/或镜像方法增强所述小目标检测图像。

5.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,平面增强处理所述小目标检...

【技术特征摘要】

1.一种小目标检测模型的训练方法,其特征在于,包括:

2.根据权利要求1所述的小目标检测模型的训练方法,其特征在于,还包括:

3.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,得到所述小目标检测样本图像包括,模糊处理所述小目标检测样本图像或减少像素值处理所述小目标检测样本图像。

4.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,平面增强处理所述小目标检测图像包括,通过旋转、剪裁、缩放、水平翻转和/或镜像方法增强所述小目标检测图像。

5.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,平面增强处理所述小目标检测图像包括,通过mosaic马赛克数据增强方法增强所述小目标检测图像。

6.根据权利要求2所述的小目标检测模型的训练方法,其特征在于,得到所述小目标检测样本图像包括:

7.根据权利要求1所述的小目标检测模型的训练方法,其特征在于,所述骨干网络层包括特征提取模块和特征修正模块;所述特征提取模块提取所述小目标检测样本图像中的目标图像特征;所述特征修正模块,修正所述特征提取模块输出的所述小目标检测样本图像中的目标图像特征;

8.根据权利要求7所述的小目标检测模型的训练方法,其特征在于,所述检测头层包括检测模块;所述检测模块包括多个检测层和设置在所述检测层之间的池化层;所述检测层包括多个卷积层和归一化层;

9.根据权利要求1所述的小目标检测模型的训练方法,其特征在于,在构建小目标识别深度神经网络结构后包括:

10.根据权利要求9所述的小目标检测模型的训练方法,其特征在于,有监督训练所述小目标识别深度神经网络包括:

11.根据权利要求9所述的小目标检测模型的训练方法,其特征在于,有监督训练所述小目标识别深度神经网络包括:

12.根据权利要求10所述的小目标检测模型的训练方法,其特征在于,所述目标检测结果中包括目标检测框;

13.根据权利要求1所述的小目标检测模型的训练方法,其特征在于,还包括:

14.一种利用小...

【专利技术属性】
技术研发人员:尤政史慧宇戴宇翔纪兴龙浦一雯丁皓伦韩毓邢飞鲁文帅
申请(专利权)人:启元实验室
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1