图像检测神经网络模型及其训练方法、图像检测方法

基本信息

申请号 CN202110530803.8 申请日 -
公开(公告)号 CN113159300A 公开(公告)日 2021-07-23
申请公布号 CN113159300A 申请公布日 2021-07-23
分类号 G06N3/04;G06N3/08;G06T3/40 分类 计算;推算;计数;
发明人 胡凡;方效林;吴文甲;杨明;罗军舟 申请(专利权)人 南京逸智网络空间技术创新研究院有限公司
代理机构 南京经纬专利商标代理有限公司 代理人 李鑫
地址 210000 江苏省南京市雨花台区大周路34号科创城B3栋第8层及第7层702、703室
法律状态 -

摘要

摘要 本申请实施例提供了一种图像检测神经网络模型及其训练方法、图像检测方法,包括获取训练数据,其中,所述训练数据包括原始图像样本,对所述原始图像样本进行缩小,得到缩小目标样本;构建包含至少包括特征提取网络、特征金字塔网络和预测网络的初始神经网络模型,所述特征金字塔网络包括可参数化的四层特征图与可参数化的特征增强层,所述四层特征图与特征增强层是对卷积四层特征图与卷积特征增强层的训练参数进行训练后得到的;使用所述训练数据训练所述初始神经网络模型的训练参数,并通过损失函数在训练过程中优化所述初始神经网络模型,得到目标神经网络模型,可以在检测速度几乎不受影响的情况下有效改善遥感图像中小尺寸物体的检测效果。