01
恶劣天气下森林目标检测的虚拟数据集扩充
问题背景:森林场景在雨雪等恶劣天气下的真实标注数据稀缺,直接训练的视觉模型在恶劣天气下感知性能骤降,而人工采集标注成本极高。
技术方案
- 在 Unreal Engine 5 中截取不同天气下的森林场景作为基础素材
- 微调 CycleGAN + Stable Diffusion XL + ControlNet,将晴天森林图转化为雨/雪天森林图,扩充数据集
- 对 YOLO26、Mask R-CNN、Mask2Former 分别用扩充前后数据集对比训练,在真实雨/雪森林数据上评测