(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210156044.8 (22)申请日 2022.02.21 (71)申请人 中国工商银行股份有限公司 地址 100140 北京市西城区复兴门内大街 55号 (72)发明人 李兆佳　 (74)专利代理 机构 北京同立钧成知识产权代理 有限公司 1 1205 专利代理师 章愫　黄健 (51)Int.Cl. G06V 20/13(2022.01) G06T 7/11(2017.01) G06T 3/40(2006.01) G06T 3/00(2006.01) G06N 3/04(2006.01)G06K 9/62(2022.01) G06V 10/774(2022.01) G06V 10/82(2022.01) G06V 10/44(2022.01) (54)发明名称 小目标对象的检测方法和检测装置 (57)摘要 本申请提供一种小目标对象的检测方法和 检测装置， 可用于人工智 能领域， 有利于提高小 目标对象的检测精度。 该方法包括： 获取遥感卫 星拍摄的待检测图像； 将待检测图像输入至目标 检测模型， 获得待检测图像的目标检测结果。 其 中， 目标检测模型是通过增强数据集训练得到 的， 增强数据集中包括样本图像和样本图像的多 个增强图像， 样本图像包括小目标对象， 多个增 强图像是通过对样本图像的多个图像副本中每 个图像副本进行随机变换组合得到的， 随机变换 组合包括平移、 旋转、 加噪、 饱和度、 亮度或者对 比度中的至少一种。 权利要求书2页 说明书12页 附图4页 CN 114529834 A 2022.05.24 CN 114529834 A 1.一种小目标对象的检测方法， 其特 征在于， 包括： 获取遥感卫星拍摄的待检测图像； 将所述待检测图像输入至目标检测模型， 获得 所述待检测图像的目标检测结果； 其中， 所述目标检测模型是通过增强数据集训练得到的， 所述增强数据集中包括样本 图像和所述样本图像的多个增强 图像， 所述样本图像包括小目标对 象， 所述多个增强 图像 是通过对所述样本图像的多个图像副本中每个图像副本进 行随机变换组合得到的， 所述随 机变换组合包括平 移、 旋转、 加噪、 饱和度、 亮度或者对比度中的至少一种。 2.根据权利要求1所述的方法， 其特征在于， 在所述将所述待检测图像输入至目标检测 模型， 获得 所述待检测图像的目标检测结果之前， 所述方法还 包括： 从原始数据集中获取所述样本图像， 所述样本图像中的小目标对象的像素掩码面积小 于预设阈值； 对所述样本图像进行复制， 获得 所述多个图像副本； 对所述多个图像副本中每个图像副本进行所述随机变换组合， 获得所述多个增强图 像； 基于所述样本图像和所述样本图像的多个增强图像， 训练所述目标检测模型。 3.根据权利要求2所述的方法， 其特 征在于， 所述增强数据集还 包括多个 轮廓图像； 在所述基于所述样本图像和所述样本图像的多个增强图像， 训练所述目标检测模型之 前， 所述方法还 包括： 获取所述多个增强图像中每个增强图像的小目标对象的像素掩码信 息， 所述像素掩码 信息包括小目标对象的像素掩码值和像素掩码位置； 基于所述像素掩码信息， 对所述每个增强图像中的小目标对象进行裁剪和复制， 得到 所述每个增强图像中的小目标对象的多个 轮廓副本； 对所述多个轮廓副本进行仿射变换， 并将所述仿射变换之后的所述多个轮廓副本粘贴 至所述每 个增强图像中， 获得 所述多个 轮廓图像； 所述基于所述样本 图像和所述样本 图像的多个增强图像， 训练所述目标检测模型， 包 括： 基于所述样本 图像、 所述样本 图像的多个增强图像以及所述多个轮廓图像， 训练所述 目标检测模型。 4.根据权利要求3所述的方法， 其特征在于， 所述对所述多个轮廓副本进行仿射变换， 包括： 从预设剪切比例范围内选择剪切比例对所述多个轮廓副本中的全部或者部分轮廓副 本进行剪切变换； 或者， 从预设旋转角度范围内选择旋转角度对所述多个轮廓副本中的全部或者部分轮廓副 本进行旋转变换； 或者， 从预设缩放比例范围内选择缩放比例对所述多个轮廓副本中的全部或者部分轮廓副 本进行缩放变换。 5.根据权利要求2所述的方法， 其特征在于， 所述基于所述样本图像和所述样本图像的 多个增强图像， 训练所述目标检测模型， 包括： 基于所述样本 图像和所述样本 图像的多个增强图像， 对预训练模型进行迁移学习， 获权　利　要　求　书 1/2 页 2 CN 114529834 A 2得优化后的预训练模型， 所述预训练模型 是通过所述原始数据集训练得到的； 获取遥感卫星拍摄的图像数据集， 并利用所述图像数据集对所述优化后的预训练模型 进行微调， 获得 所述目标检测模型。 6.根据权利要求2所述的方法， 其特征在于， 在所述对所述多个图像副本 中每个图像副 本进行所述随机变换组合， 获得 所述多个增强图像之前， 所述方法还 包括： 确定所述多个图像副本中第一图像副本的随机变换组合； 所述对所述多个图像副本 中每个图像副本进行所述随机变换组合， 获得所述多个增强 图像， 包括： 在所述第一图像副本的随机变化组合包括预设随机变换的情况下， 对所述第 一图像副 本以及所述第一图像副本中的小目标对象的像素掩码信息进 行所述预设随机变换， 得到所 述第一图像副本的增强图像， 其中， 所述预设随机变换包括平 移和/或旋转。 7.根据权利要求6所述的方法， 其特征在于， 所述确定所述多个图像副本 中第一图像副 本的随机变换组合， 包括： 根据预设概率确定平移、 旋转、 加噪、 饱和度、 亮度或者对比度中的至少一种为所述第 一图像副本的随机变化组合。 8.一种小目标对象的检测装置， 其特 征在于， 包括： 获取模块， 用于获取遥感卫星拍摄的待检测图像； 处理模块， 用于将所述待检测图像输入至目标检测模型， 获得所述待检测图像的目标 检测结果； 其中， 所述目标检测模型是通过增强数据集训练得到的， 所述增强数据集中包括样本 图像和所述样本图像的多个增强 图像， 所述样本图像包括小目标对 象， 所述多个增强 图像 是通过对所述样本图像的多个图像副本中每个图像副本进 行随机变换组合得到的， 所述随 机变换组合包括平 移、 旋转、 加噪、 饱和度、 亮度或者对比度中的至少一种。 9.一种小目标对象的检测装置， 其特征在于， 包括： 处理器， 所述处理器和存储器耦合， 所述存储器用于存储计算机程序， 当所述处理器调用所述计算机程序时， 使得所述装置执 行权利要求1至7中任一项所述的方法。 10.一种计算机可读存储介质， 其特征在于， 所述计算机可读存储介质用于存储计算机 程序， 当所述计算机程序在计算机上运行时， 使得所述计算机执行如权利要求1至7中任一 项所述的方法。 11.一种计算机程序产品， 其特征在于， 包括计算机程序代码， 所述计算机程序代码被 运行时， 实现如权利要求1至7中任一项所述的方法。权　利　要　求　书 2/2 页 3 CN 114529834 A 3