ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

无人机低空交通巡检:航拍道路状态分类系统实战

无人机低空交通巡检:航拍道路状态分类系统实战

1. 低空巡检场景下的道路状态识别:这个项目到底在做什么

第一次接触“无人机低空交通巡检”这个概念,很多人脑子里浮现的画面可能是那种大型活动现场的航拍直播,或者影视剧里追车戏的上帝视角。但真正落到交通管理这个领域,它要解决的核心问题其实非常具体:用无人机挂载的摄像头,在低空对城市道路进行周期性或事件驱动的航拍,然后把拍回来的画面自动分类成“正常通行”“交通拥堵”“交通事故现场”等状态,辅助交管部门快速研判和调度。

这个项目的本质,是一个航拍视角下的道路场景分类系统。它不追求识别每一辆车的车牌,也不要求精确统计车流量,而是要在第一时间回答一个关键问题:这段路现在是什么状况?是畅通、缓行、堵死,还是发生了事故?这个判断结果会直接影响交警的远程研判效率和出警决策。

适合参考这个项目的人,大致有这么几类:做智慧城市或交通信息化产品的开发者,想了解航拍视角下的视觉分类方案怎么落地;做无人机行业应用的团队,需要给巡检航线增加自动分析能力;还有做边缘计算设备部署的工程师,关心模型怎么在机载或地面站上跑起来。哪怕你只是对“无人机+AI”这个组合感兴趣,这个项目的技术路径也能给你一个比较完整的参考框架。

我之所以觉得这个方向值得认真拆解,是因为它和常见的固定摄像头交通监控有本质区别。固定摄像头的视角是稳定的、光照变化相对可控、背景基本不变,而无人机航拍面临的是动态视角、高度变化、光照剧烈波动、地物遮挡复杂等一系列挑战。这就决定了它的技术方案不能照搬地面监控那一套,需要在数据采集、标注策略、模型选型、部署方式上做针对性的设计。

2. 整体方案设计与技术选型思路

2.1 为什么选择“分类”而不是“检测”

很多人一上来就想做目标检测,把画面里的每辆车都框出来,然后统计数量判断拥堵。这个思路在固定摄像头场景下可行,但在无人机航拍场景下会碰到几个硬钉子。

首先是尺度变化问题。无人机飞行高度从几十米到几百米不等,同一辆车在不同高度下的像素尺寸可能相差十几倍。检测模型要同时覆盖这么大的尺度范围,训练难度和标注成本都会急剧上升。其次是遮挡问题。城市道路上有树冠遮挡、建筑阴影、高架桥覆盖,航拍画面里经常出现车辆被部分或完全遮挡的情况,检测模型很容易漏检。最后是实时性要求。交通巡检场景往往需要在边缘设备上实时出结果,检测模型的推理开销比分类模型大得多。

所以这个项目选择图像分类作为核心任务,把整张航拍图或裁剪后的道路区域作为一个样本,直接输出场景类别标签。这样做的好处是标注成本低(只需要给整张图打一个标签)、模型轻量(分类网络通常比检测网络小一个数量级)、推理速度快(适合边缘部署)。代价是丢失了车辆级别的空间信息,但对于“判断道路状态”这个目标来说,这个代价是可以接受的。

2.2 类别体系怎么定

类别体系的设计直接决定了模型能回答什么问题。根据项目标题里提到的需求,我建议至少包含以下几个大类:

类别编号类别名称典型特征应用场景
0畅通车辆稀疏、间距大、无明显排队常规巡检
1缓行车辆较多、速度慢、有短排队预警提示
2拥堵车辆密集、排队长度长、几乎静止调度决策
3事故现场有车辆异常停靠、可能有散落物、警车或救护车在场紧急研判
4施工占道有施工围挡、锥桶、工程车辆路况提示
5正常但特殊如大型活动散场、临时管制特殊场景

这里有个经验之谈:类别不要定得太细。我见过一些团队一开始定了十几个类别,结果发现类间差异太小,标注人员自己都分不清,模型训练出来混淆矩阵一塌糊涂。建议先从4到6个大类起步,等数据积累够了再考虑细分。

另外,事故现场这个类别需要特别注意。它的样本量天然就少,而且不同事故的视觉表现差异很大——有的是两车追尾,有的是单车撞护栏,有的是侧翻。如果直接把它和拥堵放在一起做多分类,模型很容易因为样本不均衡而偏向多数类。常见的做法是先做二分类(正常 vs 异常),再在异常里做细分,或者用异常检测的思路来处理。

2.3 数据采集与标注策略

航拍数据的采集有几个关键参数需要控制:

  • 飞行高度:建议在80到150米之间。太低视野窄,覆盖不了足够长的路段;太高车辆像素太少,拥堵和事故的视觉特征会丢失。
  • 拍摄角度:以俯视和斜俯视为主,角度控制在30度到90度之间。完全垂直俯视有利于减少透视畸变,但斜俯视更符合人眼对交通场景的认知习惯。
  • 光照条件:要覆盖晴天、阴天、黄昏、夜间(如果有夜航能力)等多种光照。特别是黄昏和夜间,事故现场的车灯、警灯是重要的视觉线索。
  • 路段类型:城市快速路、主干道、交叉路口、匝道口都要覆盖。不同路段类型的拥堵模式不一样。

标注方面,我强烈建议先做区域裁剪再打标签。整张航拍图里可能同时包含畅通路段和拥堵路段,如果直接给整图打一个标签,会引入大量噪声。做法是沿着道路走向把画面切成若干段,每段单独判断状态。裁剪的时候要注意保留一定的上下文,不要把路段切得太碎,否则模型学不到拥堵的全局特征。

注意:标注一致性是这类项目最大的坑。同一个画面,不同标注人员可能给出不同判断。建议制定详细的标注规范,明确“排队长度超过多少米算拥堵”“车辆间距小于多少算缓行”等量化标准,并且定期做标注质量抽检。

3. 核心模型选型与训练细节

3.1 骨干网络怎么选

分类任务的骨干网络选择,核心权衡是精度、速度、参数量三者之间的平衡。考虑到这个项目大概率要部署在无人机机载设备或地面边缘服务器上,我建议从以下几个方向考虑:

轻量级方案:MobileNetV3、EfficientNet-Lite、ShuffleNetV2。这类网络参数量在几百万级别,推理速度快,适合算力受限的场景。MobileNetV3-Small在骁龙级别的移动芯片上可以做到实时推理,精度对于4到6分类任务来说通常够用。

中等规模方案:ResNet-50、EfficientNet-B0/B1、ConvNeXt-Tiny。如果边缘设备有GPU或者算力稍充裕,这类网络能提供更好的特征表达能力,特别是对于事故现场这种类间差异细微的场景。

Transformer方案:ViT、Swin-Tiny。Transformer在全局建模能力上有优势,对于判断“整条路的拥堵态势”这种需要全局信息的任务可能有帮助。但ViT对数据量要求高,如果训练样本不够多,容易过拟合。

我的建议是:先用MobileNetV3或EfficientNet-Lite跑通基线,看看精度是否满足业务要求。如果不够,再往上换更大的模型。不要一上来就上大模型,训练成本和部署成本都会成倍增加。

3.2 数据增强策略

航拍图像的数据增强和常规图像分类不太一样,需要针对航拍特点做设计:

  • 旋转:航拍视角下,道路走向是任意的,所以0到360度随机旋转是合理且必要的增强。
  • 缩放:模拟不同飞行高度带来的尺度变化,缩放范围建议在0.7到1.3倍之间。
  • 亮度与对比度扰动:模拟不同光照条件,亮度变化范围可以大一些,正负40%左右。
  • 色彩抖动:模拟不同天气和季节的色温变化。
  • 随机裁剪:模拟不同取景范围,但要注意裁剪后不能丢失关键的道路状态信息。
  • 马赛克增强:把四张图拼成一张,增加样本多样性,同时让模型学习不同场景之间的边界。

需要避免的增强:垂直翻转。航拍图垂直翻转后,天空跑到下面,地物关系完全乱了,这种增强会引入错误的学习信号。大角度透视变换也要慎用,因为航拍本身的透视关系是有物理意义的,过度变换会让模型学到不真实的几何关系。

3.3 损失函数与类别不均衡处理

事故现场、施工占道这类样本天然就少,类别不均衡是绕不开的问题。几种常用的处理方式:

重采样:对少数类样本进行过采样,或者对多数类样本进行欠采样。过采样容易导致过拟合,欠采样会丢失信息,实践中可以结合使用。

类别权重:在交叉熵损失里给少数类更高的权重。权重的设定可以按类别频率的倒数来算,也可以根据业务重要性手动调整。比如事故现场的漏检代价很高,可以给它更高的权重。

Focal Loss:降低易分类样本的权重,让模型聚焦在难分类样本上。对于事故现场这种“难样本”,Focal Loss通常能带来明显提升。

标签平滑:缓解模型对训练标签的过度自信,提升泛化能力。对于标注边界模糊的类别(比如缓行和拥堵之间),标签平滑特别有用。

我自己的经验是:先算一下各类别的样本数量,如果最大类和最小类的比例超过10比1,就必须做处理。处理方式优先选类别权重加Focal Loss的组合,实现简单且效果稳定。

4. 实操流程与关键环节实现

4.1 数据预处理流水线

整个预处理流程可以拆成以下几个步骤:

import cv2 import numpy as np from torch.utils.data import Dataset class AerialRoadDataset(Dataset): def __init__(self, image_paths, labels, transform=None): self.image_paths = image_paths self.labels = labels self.transform = transform def __len__(self): return len(self.image_paths) def __getitem__(self, idx): img = cv2.imread(self.image_paths[idx]) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) label = self.labels[idx] if self.transform: img = self.transform(image=img)['image'] return img, label

预处理的关键点:

  • 尺寸统一:建议统一到224x224或256x256。如果原始航拍图分辨率很高,可以先做区域裁剪再缩放,避免直接缩放导致细节丢失。
  • 归一化:使用ImageNet的均值和标准差做归一化,如果从头训练也可以用数据集自身的统计量。
  • 道路区域提取:如果画面里包含大量非道路区域(建筑、绿地、天空),可以考虑先用语义分割或简单的颜色阈值把道路区域提取出来,再做分类。这样可以减少背景干扰。

4.2 训练配置与调参

以下是一套经过验证的训练配置,可以作为起点:

配置项推荐值说明
优化器AdamW比SGD更容易调,适合中小数据集
学习率1e-3 到 1e-4用余弦退火调度
权重衰减1e-4防止过拟合
Batch Size32 到 64根据显存调整
训练轮数50 到 100配合早停策略
混合精度开启加速训练,节省显存
梯度裁剪开启防止梯度爆炸

调参的几个经验:

  • 学习率是最重要的超参数。如果训练loss震荡不下降,先降学习率。如果loss下降太慢,适当提高学习率。
  • Batch Size不要太小。小于16的时候,BatchNorm的统计量会不稳定,影响训练效果。
  • 早停策略:监控验证集上的F1分数,如果连续10个epoch没有提升就停止训练。不要只看准确率,类别不均衡时准确率会骗人。

4.3 模型部署与推理优化

训练好的模型要落到实际巡检场景里,部署方式主要有两种:

机载部署:把模型部署在无人机的机载计算设备上,实时处理摄像头画面。这种方式响应最快,但对算力和功耗要求高。常用的推理框架有TensorRT、NCNN、TFLite。优化手段包括量化(FP16或INT8)、剪枝、知识蒸馏。

地面站部署:无人机把画面实时回传到地面站,由地面服务器做推理。这种方式算力充裕,可以跑更大的模型,但依赖通信链路。适合对实时性要求不那么极端的场景。

推理优化的几个实用技巧:

  • INT8量化:在精度损失可控的前提下,推理速度可以提升2到4倍。建议用训练后量化先试,如果精度掉太多再考虑量化感知训练。
  • 输入分辨率调整:如果224x224的精度够用,就不要用448x448。分辨率翻倍,计算量翻四倍。
  • 模型剪枝:去掉冗余的通道和层,参数量可以减少30%到50%,精度损失通常在1到2个百分点以内。

提示:部署前一定要做端到端测试,用真实航拍视频流跑一遍,看看帧率、延迟、内存占用是否满足要求。实验室里的精度指标和实际场景的表现往往有差距。

5. 常见问题与排查技巧实录

5.1 模型把拥堵误判成事故怎么办

这是最典型的一类错误。拥堵和事故在视觉上确实有相似之处:车辆密集、速度慢、有排队。区分的关键在于异常信号:事故现场通常有车辆姿态异常(横穿、侧翻)、有散落物、有警车或救护车、有人员走动。如果模型分不清,可以从以下几个方向排查:

  • 检查标注质量:是不是有些拥堵样本被误标成了事故?反过来也要查。
  • 增加难例挖掘:把模型分错的样本单独拿出来,人工复核后加入训练集。
  • 引入辅助特征:除了RGB图像,可以考虑加入光流特征(判断车辆是否在移动)、车辆姿态特征等。
  • 调整类别权重:如果事故现场的漏检代价高,就给它更高的权重,让模型更倾向于预测事故。

5.2 不同光照条件下精度波动大

航拍图像的光照变化比地面图像剧烈得多。中午的强光、黄昏的暖色调、夜间的低照度,都会让模型的表现不稳定。解决办法:

  • 数据层面:确保训练集覆盖各种光照条件,特别是目标部署场景的主要光照条件。
  • 增强层面:加大亮度和色彩扰动的范围,模拟极端光照。
  • 模型层面:考虑使用对光照变化更鲁棒的特征,比如颜色不变性特征,或者在网络里加入光照归一化层。
  • 后处理层面:如果连续多帧的预测结果不一致,可以用时序平滑来稳定输出。

5.3 边缘设备上推理速度不达标

这是部署阶段最常见的问题。排查思路:

问题现象可能原因解决方向
单帧推理超过100ms模型太大换轻量骨干、剪枝、量化
内存占用过高输入分辨率太大降低分辨率、减少batch size
功耗超标没有用硬件加速启用GPU/NPU、用专用推理框架
帧率不稳定内存泄漏或线程竞争检查代码、固定线程数

我踩过的一个坑是:在开发板上直接用PyTorch推理,速度慢得离谱。后来换成TensorRT,同样的模型速度提升了5倍多。所以部署前一定要选对推理框架,不要直接用训练框架跑推理。

5.4 常见问题速查表

问题排查方向快速验证方法
训练loss不下降学习率、数据标注、模型结构用一个小数据集过拟合测试
验证集精度远低于训练集过拟合增加数据增强、加正则化
某些类别精度特别低样本不足、类间混淆看混淆矩阵、补充样本
推理结果抖动时序不一致加时序平滑、多帧投票
部署后精度下降预处理不一致对比训练和推理的预处理代码

6. 实际落地中的经验与扩展方向

这个项目从技术验证到实际落地,中间还有不少工程上的细节需要打磨。我说几个自己体会比较深的点。

第一,数据闭环比模型选型更重要。很多团队花大量时间调模型结构,但实际效果提升有限。真正带来质变的是持续收集真实场景数据,把模型分错的样本找出来,人工复核后加入训练集。这个闭环跑起来之后,模型精度会稳步提升。

第二,多帧时序信息值得利用。单帧图像判断拥堵和事故有时候确实困难,但如果结合前后几帧的信息,比如车辆是否在移动、排队长度是否在变化,判断会准确得多。实现上可以用LSTM或Transformer对帧序列建模,也可以用简单的投票机制。

第三,和业务系统对接要考虑输出格式。模型输出的不只是一个类别标签,还需要包含置信度、时间戳、地理位置等信息,方便业务系统做后续处理。如果置信度低于某个阈值,应该标记为“待人工确认”,而不是强行给出一个判断。

第四,扩展方向可以往多任务走。除了场景分类,还可以同时做车辆检测、道路分割、异常事件检测等任务。多任务学习可以共享特征提取网络,提升整体效率。当然,多任务学习的训练难度也更大,需要仔细设计损失函数的权重。

最后分享一个实操中的小技巧:在标注阶段就引入模型辅助。先用一个初步训练的模型对未标注数据做预标注,人工只需要修正错误的部分。这样可以大幅提升标注效率,特别是对于事故现场这种需要仔细判断的类别。预标注的精度不需要很高,哪怕只有60%的准确率,也能节省大量人工。

RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进