申请/专利权人:河北翔拓航空科技有限公司;石家庄铁道大学
申请日:2023-12-07
公开(公告)日:2024-03-22
公开(公告)号:CN117746264A
主分类号:G06V20/17
分类号:G06V20/17;G06V20/10;G06V20/54;G06V10/26;G06V10/96;G06V10/776;G06V10/44;G06V10/52;G06V10/82;G06N3/0455;G06N3/0464
优先权:
专利状态码:在审-实质审查的生效
法律状态:2024.04.09#实质审查的生效;2024.03.22#公开
摘要:本发明公开了一种无人机航拍车辆检测与道路分割的多任务实现方法,涉及图像处理方法技术领域。所述方法包括如下步骤:获取无人机航拍视频,处理并构建多任务数据集;对数据集进行增强处理和分辨率调整以满足多任务网络输入要求;构建车辆检测与道路分割的多任务网络模型,通过编码器网络将输入图像进行编码。将编码输入车辆检测网络,得到检测候选框,使用非极大值抑制算法对候选框进一步处理,得到车辆检测结果。同时将编码输入道路分割网络得到分割结果;通过公开数据集对模型预训练,获取预训练权重;通过自建多任务数据集进行训练,得到最终多任务模型;采用评估指标对模型性能进行评估。该方法具有高精度、参数量和计算量低等优点。
主权项:1.一种无人机航拍车辆检测与道路分割的多任务实现方法,其特征在于包括如下步骤:S1:获取无人机航拍视频,对视频进行处理,获得图像数据,对图像进行标注,并对标注结果进行划分,构建多任务数据集;S2:对多任务数据集进行数据增强处理,并调整数据集中图像大小,使其符合多任务网络输入大小的要求;S3:构建多任务网络,通过多任务网络对经过处理后的图像进行特征提取;S4:多任务网络对图像进行处理后,车辆检测网络和道路分割网络输出相应的特征张量;S5:对多任务网络的编码器网络、车辆检测解码网络和道路分割解码网络分别进行预训练,得到预训练权重;S6:通过预训练权重,在多任务数据集上优化无人机航拍图像车辆检测与道路分割多任务网络联合损失函数,对多任务网络进行训练,使用NMS算法对车辆检测输出候选框进行筛选,得到最终训练结果,使用评估指标对结果进行评估。
全文数据:
权利要求:
百度查询: 河北翔拓航空科技有限公司;石家庄铁道大学 无人机航拍车辆检测与道路分割的多任务实现方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。