基于改进RT-DETR网络的无人机航拍小目标检测方法

本发明公开了一种基于改进RT‑DETR网络的无人机航拍小目标检测方法，其特征在于，包括如下步骤：步骤1、获取公开数据集并预处理；步骤2、构建小目标检测模型，以RT‑DETR网络为基础网络，所述RT‑DETR网络包括主干网络和高效混合编码器，将主干网络Resnet‑r18中的常规卷积替换成部分卷积重参数模块；将高效混合编码器中的AIFI模块替换为可变形注意力模块；在高效混合编码器的CCFM模块中引入p2检测头得到CCFM‑p2模块；步骤3、配置训练环境将预处理的数据集作为输入对小目标检测模型进行训练；步骤4、应用完成训练的小目标检测模型，将待检测的目标图像作为输入输出检测结果。该方法降低模型的参数量与计算量，并且能够更好的捕获图像中的结构信息和语义特征，提高了对小目标的敏感度，以获得更好的检测结果。