AV2T-Gen: Aerial Visible to Thermal Generation with Environment and Vehicle State Guidance¶
会议: ECCV2026
Paper: https://eccv.ecva.net/virtual/2026/poster/3919
PDF: https://media.eventhosts.cc/Conferences/ECCV2026/pdfs/3703.pdf
代码: https://github.com/NPU-CVPG/AV2T
领域: 自动驾驶 / 航拍跨模态图像生成
关键词: 可见光到热红外、环境条件、车辆运动状态、基础模型引导、潜空间扩散
一句话总结¶
AV2T-Gen 在 Instruct-Pix2Pix 式扩散生成中加入环境提示和带运动状态的目标条件,使航拍 RGB 到热红外转换不只模仿外观,在四场景联合评测中将 ThermalGen 的 FID 从 83.16 降至 67.45,并提升合成红外图像上的车辆检测效果。
研究背景与动机¶
可见光和热红外不是同一场景的两套配色。RGB 主要反映可见光照明与反射,热红外表现还受物体温度、材料特性和大气条件影响。因此,两辆外观相似的车,可能因为运动状态或所处环境不同而呈现不同的热特征。把 RGB-T 转换只当作风格迁移,容易生成轮廓正确、热表现却不符合场景的图像。另一方面,同步且标定良好的航拍 RGB-T 图像对采集成本高,生成数据又确实能服务跨模态匹配和目标检测,这使转换质量具有实际价值。
已有 GAN 和扩散方法逐渐改善了清晰度与结构保持,F-ViTA 还用 SAM、Grounding DINO、CLIP 等基础模型提供语义和边界引导。但“知道这里是一辆车”不等于“知道这辆车此时应当如何发热”,只依赖单帧外观也很难可靠判断车辆是否在运动。作者于是引入两类外部证据:由航拍序列获得的车辆状态,以及由拍摄位置和时间关联的温度、湿度、气压、风速等环境信息,并构建包含这些元数据的配对数据集。
这里的目标是减少 RGB 到热红外映射中的条件缺失,而不是用一个生成网络替代完整热物理模型。核心 idea:用环境文本说明全局成像条件,用目标掩码及带运动状态的标签定位局部热特征,再将两者共同注入潜空间扩散去噪过程。
方法详解¶
整体框架¶
输入包含可见光图像、车辆状态标注和环境信息,输出为对应的热红外图像。训练时还需要配对真实热红外图像作为监督;部署时不需要真实热红外目标,但车辆状态依赖序列分析,环境条件依赖额外元数据,并非纯单帧 RGB 输入。
处理分为条件获取、上下文融合和条件去噪三部分。冻结的 VAE 将可见光图像编码为结构条件,将真实热红外图像编码为训练目标;环境提示经 CLIP 编码,目标标签与 SAM 掩码经可训练 MLP 融合,再一起指导 UNet 生成热红外潜变量。最后由 VAE 解码回图像空间。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
A["航拍 RGB 序列<br/>位置与拍摄时间"] --> B["环境与状态获取"]
B --> C["掩码与语义融合"]
B -->|环境提示经 CLIP| D["条件潜空间去噪"]
C -->|目标上下文| D
A -->|RGB 经冻结 VAE| D
N["带噪热红外潜变量<br/>训练来自真实图像"] --> D
D --> E["冻结 VAE 解码<br/>生成热红外图像"]
关键设计¶
1. 环境与状态获取:补上单帧 RGB 无法确定的条件
车辆状态并非由提示词随意指定。作者先训练航拍视角的 YOLOv11 可见光车辆检测器,再利用时间序列判断同一辆车是否移动,最终产生 stationary、moving、unknown 三类状态。unknown 保留了无法确定的情况,但正文没有给出轨迹关联、相机运动补偿或运动阈值的完整算法,细节被放在补充材料中,不能把这些环节补写成已知实现。这个输入的作用是为车辆热特征提供额外线索,而不是直接测出发动机温度;运动标签和真实热状态之间仍可能存在偏差。
环境分支根据 GPS 与拍摄时间,从开放气象平台取得相应的时间、温度、湿度、气压、风速,将离散观测插值为连续值,再组织成文本提示,经 CLIP 得到环境提示嵌入。使用文本接口便于接入原有图像编辑扩散模型,但这些气象量仍是学习到的条件,不是直接代入辐射传输方程的物理参数。作者将湿度、气压解释为大气衰减的代理变量,将语义关联解释为材料发射率的先验;这种解释具有合理性,但不等于实现了可标定的物理求解器。
2. 掩码与语义融合:让车辆状态作用于正确的目标区域
仅在全局提示里写“有移动车辆”,不能明确指定场景中的哪辆车应产生相应热特征。AV2T-Gen 使用 Ram-Grounding 获取通用目标标签和边界框,与自动车辆状态标注合并形成完整标签及对应位置;这些框交给 SAM 生成物体掩码。SAM 掩码提供边界和空间定位,CLIP 编码后的完整标签提供目标类别与状态语义。这里既包含车辆,也能包含建筑等背景物体,因此上下文并非只覆盖检测框中的车辆。
掩码嵌入与标签嵌入先拼接,再由可训练 MLP 转为联合上下文;这一分支随后与环境提示嵌入一起作为条件输入。原文式 (3) 将条件组织方式写为:
其中 \(E_p\) 是环境提示嵌入,\(E_m\) 是掩码嵌入,\(E_l\) 是包含状态信息的标签嵌入。括号表示共同构成条件输入,正文没有明确最终融合究竟采用逐元素相加还是特定 token 拼接布局,也没有给出掩码编码器的完整维度,笔记不自行补定。关键区别是保留“哪个目标、在哪里、是什么状态”的关联,使全局环境条件与局部目标条件能分别发挥作用。
3. 条件潜空间去噪:在保持 RGB 结构的同时学习热表现
骨干沿用 Instruct-Pix2Pix 的 VAE 和扩散思路。冻结的 VAE 编码器把可见光输入压缩为潜变量 \(I_E\),保留布局与目标形状;真实热红外图像编码为 \(Z\),训练时向热红外潜变量加高斯噪声。带噪潜变量与 \(I_E\) 拼接成为 UNet 输入,联合条件 \(E_c\) 参与去噪,UNet 内的 QKV 注意力用于联系空间结构和文本语义。去噪产生的热红外潜变量再经冻结解码器还原图像,因此 RGB 的结构约束与热红外的输出分布并不是直接共享像素值。
这一设计让掩码边界和标签语义能够在多步生成中持续提供约束,而不只是输出后的修补。推理从随机热红外潜变量开始,在 RGB 与上下文条件下去噪;训练使用的真实热红外图像不能被理解为推理输入。正文式 (4) 只是去噪映射的简写,未明确采样器、噪声调度、步数或注意力各支路的具体张量组织,不能据此声称使用某种特定 DDIM 配置。基础模型保持冻结,作者说明可训练部分为 MLP 和 UNet;全系统参数量不能当成可训练参数量。
一个完整示例¶
设一段航拍序列中同一停车区域有移动车辆和静止车辆,这是解释流程的示例,不是新增实验。先由序列分析给各车附上 moving 或 stationary;无法判断的目标保留 unknown。位置与时间关联的温度、湿度、气压、风速组成共享环境提示,Ram-Grounding 与状态标注提供目标标签和框,SAM 再把各目标的空间区域分开。
MLP 接收这些区域和相应语义,UNet 同时看到全局气象条件、逐目标状态和 RGB 布局,生成一幅位置对应的热红外图像。若移除状态,模型只能依据外观推断车辆热特征;若移除环境提示,不同环境下相似外观可能被映射为平均热表现。两类消融正是用来检验这两个条件是否带来独立收益,不能将示例解读为“移动车辆必然更热”的物理定律。
损失函数 / 训练策略¶
训练监督包括预测热红外潜变量与真实潜变量之间的 L1 损失、对生成图像和真实图像应用 Sobel 后的边缘 L1 损失,以及 VGG 感知损失。潜变量监督约束整体转换,边缘项抑制场景结构与车辆边界退化,感知项补充图像级细节。按正文的三项文字说明与权重,目标可整理为:
缓存式 (5) 的第三项连接符及系数标记存在抽取缺损,上式是依据紧邻正文明确列出的三项和权重整理,不是逐字符复刻;精确排版以原文为准。正文未给出 VGG 使用层、特征归一化或优化器设置,不能补造。训练为 100 epochs,使用 8 张 24GB RTX 4090,训练与测试分辨率均为 \(640\times512\)。
实验关键数据¶
主实验¶
自建配对数据集包含变电站、学校、广场、驾校四个航拍场景,共 29,672 对图像,各场景按 9:1 划分。正文称训练/验证,表 2 称训练/测试,以下沿用“评测划分”避免混淆。下表摘录原文表 1 的 All scenes 列,即四场景联合训练与评测,不是留一场景外推结果。PSNR 越高越好,单位为 dB;SSIM 越高越好,FID 和 LPIPS 越低越好。
| 方法 | PSNR | SSIM | FID | LPIPS |
|---|---|---|---|---|
| DiffV2IR | 22.35 | 0.620 | 108.54 | 0.39 |
| PID | 22.33 | 0.621 | 106.13 | 0.36 |
| sRGB-TIR | 22.34 | 0.619 | 110.83 | 0.39 |
| ThermalGen | 23.94 | 0.734 | 83.16 | 0.37 |
| AV2T-Gen | 24.08 | 0.750 | 67.45 | 0.29 |
相对 ThermalGen,PSNR 仅增加 0.14 dB,而 FID 减少 15.71、LPIPS 减少 0.08,收益更明显地体现在感知与分布指标上。相同数据划分并不意味着相同信息预算:本文额外使用环境及状态元数据,比较支持“增加这些条件的系统更好”,并非严格隔离骨干架构的比较。
消融实验¶
下表来自原文表 7,所有配置均保留潜变量损失,每行只移除所列项;移除项的顺序与图 6 的说明一致。
| 配置 | PSNR | SSIM | FID | LPIPS |
|---|---|---|---|---|
| 去掉感知损失 | 23.14 | 0.716 | 72.19 | 0.30 |
| 去掉边缘损失 | 22.90 | 0.722 | 68.00 | 0.30 |
| 去掉车辆状态 | 22.93 | 0.732 | 73.83 | 0.30 |
| 去掉环境提示 | 22.29 | 0.700 | 77.66 | 0.31 |
| 完整模型 | 24.08 | 0.750 | 67.45 | 0.29 |
移除环境提示使 PSNR 下降 1.79 dB、FID 增加 10.21,是该表影响最大的移除项;移除车辆状态使 PSNR 下降 1.15 dB、FID 增加 6.38。边缘损失对 PSNR 的影响比对 FID 更显著,说明各项改善的质量侧面并不相同。表 8 进一步逐个移除气象量,作者得到温度、湿度、风速、气压依次递减的重要性排序,但这不构成各变量具有因果可识别性的证明。
下游与泛化¶
原文表 9 在各方法生成的热红外图像上使用同一检测器,下表摘录与 ThermalGen 的比较,指标单位均为百分数。
| 方法 | mAP | mAP50 | mAP75 | 小目标 mAP |
|---|---|---|---|---|
| ThermalGen | 11.7 | 29.4 | 6.9 | 10.7 |
| AV2T-Gen | 15.6 | 35.9 | 11.1 | 14.2 |
mAP 增加 3.9 个百分点,说明生成质量改善也体现在检测器可利用的特征上。注意这是“真实红外训练检测器、合成红外评测”的方向,不应混同于合成数据训练带来的真实域收益;原文表 10 的 Setup B 才采用合成训练、真实测试,其 AV2T-Gen moving/static mAP50 分别为 78.4/73.1。
LLVIP 跨数据集结果见原文表 4:零样本下 AV2T-Gen 与 ThermalGen 的 PSNR 为 13.82/12.58,微调后为 23.68/23.12。留一场景评测中,本文 PSNR 相对对应场景内结果下降 1.18 至 1.52 dB,支持一定迁移能力,但不是无损泛化。表 4 的可见行未包含 F-ViTA,不能据正文措辞补写它的比较数值。
亮点与洞察¶
- 条件设计针对跨模态映射中真正缺失的信息,而不只是增加网络层。序列提供车辆运动证据,气象元数据提供全局背景,使“外观相同却热表现不同”有了可学习的解释变量。
- 掩码与状态标签成对输入,比孤立的全局文本更容易保持对象归属。这个思路可用于其他需要逐对象属性控制的跨模态生成,但仍需验证掩码错误和标签不确定性如何传播。
- 消融、跨场景迁移及下游检测构成不同层面的证据。它们分别回答条件是否有用、是否依赖固定场景、生成结果是否具备任务价值,不能只看一项 FID 就概括全部结论。
局限与展望¶
- 物理真实性仍是间接证据。 作者明确不求解完整辐射传输方程,正文也没有给出绝对温度误差或辐射定标结果;PSNR、SSIM、FID、LPIPS 和检测指标不足以证明温度预测准确。可补充受控环境变化和辐射测量评测。
- 输入条件可能含误差。 运动不等于发动机热状态,停车后的余热、遮挡和相机运动均可能影响标签;正文未提供状态算法完整细节。作者提出将实时传感器和更细粒度跟踪历史用于后续改进。
- 数据划分和公平性有边界。 连续序列采用 9:1 划分,但正文未明确是否按拍摄时间块隔离,不能排除相邻帧相关性的影响;额外元数据和冻结基础模型也构成系统资源优势。留一场景与 LLVIP 实验有帮助,但未完全解决这些问题。
- 推理开销较大且硬件口径不统一。 全系统约 15.5B 参数,单图总耗时约 10.00 秒,扩散部分为 9.45 秒。正文将总耗时归于 RTX 4090,而表 3 标注 RTX L40,不能将二者当成一致的硬件实测;作者定位本就是离线数据生成,而非实时机载推理。
- 缓存与报告存在缺口。 补充材料未包含在本次缓存中,损失公式有抽取缺损,部分基线叙述与表格不完全对应。本笔记保留可核对结果,不补写采样超参、缺失基线或不明实现。
相关工作与启发¶
- 对比 Instruct-Pix2Pix: 本文借用潜空间图像编辑框架,但把编辑条件具体化为热红外任务所需的环境与车辆状态。贡献重点在条件建模和数据,而非提出全新的扩散骨干。
- 对比 F-ViTA: 二者都利用基础模型获得语义与空间引导,本文强调显式环境变量和逐车辆状态。当前缓存没有表列的 F-ViTA 定量结果,因此这里只比较方法,不声称已量化证明优于该方法。
- 对比 PID 与 ThermalGen: PID 强调物理相关约束,ThermalGen 采用风格解耦的流式生成;AV2T-Gen 则把可观测外部条件接入扩散。可进一步设计相同元数据、相同训练预算的比较,区分生成骨干和条件信息各自的贡献。
评分¶
- 新颖性: 4/5。环境元数据与逐车状态的联合控制有明确任务针对性,骨干和基础组件主要复用已有方法。
- 实验充分度: 4/5。包含主比较、组件消融、跨数据集、留一场景和检测验证,但物理定标与信息预算公平性仍不足。
- 写作质量: 3/5。总体流程易于理解,但硬件口径、基线叙述和部分实现信息需要澄清。
- 价值: 4/5。对离线航拍 RGB-T 数据构建有实际意义,尚不能替代真实热传感器或直接用于实时部署。