无人机航拍车辆检测与道路分割的多任务实现方法

专利主权项内容

1.一种无人机航拍车辆检测与道路分割的多任务实现方法，其特征在于包括如下步骤：S1：获取无人机航拍视频，对视频进行处理，获得图像数据，对图像进行标注，并对标注结果进行划分，构建多任务数据集；S2：对多任务数据集进行数据增强处理，并调整数据集中图像大小，使其符合多任务网络输入大小的要求；S3：构建多任务网络，通过多任务网络对经过处理后的图像进行特征提取；S4：多任务网络对图像进行处理后，车辆检测网络和道路分割网络输出相应的特征张量；S5：对多任务网络的编码器网络、车辆检测解码网络和道路分割解码网络分别进行预训练，得到预训练权重；S6：通过预训练权重，在多任务数据集上优化无人机航拍图像车辆检测与道路分割多任务网络联合损失函数，对多任务网络进行训练，使用NMS算法对车辆检测输出候选框进行筛选，得到最终训练结果，使用评估指标对结果进行评估。