一种基于遥感图像军用飞机目标检测方法及装置

    公开(公告)号:CN117593662A

    公开(公告)日:2024-02-23

    申请号:CN202311688845.X

    申请日:2023-12-11

    Abstract: 本发明公开了一种基于遥感图像军用飞机目标检测方法及装置,包括以下步骤:(1)获取遥感军用飞机的图像数据,对所有的图像数据标注飞机位置及其类别;(2)构建基于改进YOLOV5的目标检测模型,具体改进为:使用几何、颜色变换等数据增强策略扩充训练数据集的规模;在原模型的Conv结构上新增一条卷积分支,并在推理模型中将分支以及归一层和激活函数进行合并;在原模型中增加一个SO模块;(3)对目标检测模型进行迭代训练;(4)将待检测的图像或视频流数据输入训练好的目标检测模型中,得到军用飞机的预测结果。利用本发明,可以减少推理模型的计算量,在不影响检测精度的前提下提高检测速度。

Patent Agency Ranking