基于成对图像数据生成与知识蒸馏的雨天目标检测方法

本申请公开了一种基于成对图像数据生成与知识蒸馏的雨天目标检测方法，属于自动驾驶目标检测技术领域。该方法包括：获取道路场景真实图像和语义分割掩码；将语义分割掩码与天气条件文本信息输入到训练好的改进扩散模型，得到雨天条件下的道路场景生成图像；在目标检测模型中引入CoT模块与ODConv模块进行优化，然后采用动态加权聚合损失函数对目标检测模型进行改进，得到优化后的目标检测模型；进行知识蒸馏，得到雨天目标检测增强模型；使用雨天目标检测增强模型进行目标检测。本发明提供的方法在雨天场景中表现出色，具备良好的泛化能力，能够在各种复杂场景中提供更高精度的目标检测，有效提升了检测准确率和稳定性。