一种基于CycleGAN的夜间桥梁交通检测数据增强方法

本发明公开了一种基于CycleGAN的夜间桥梁交通检测数据增强方法。该方法包括以下步骤：对日间车流图片数据进行完整标注以获取具有完整标签的真日间车流图片数据，并对夜间车流图片数据进行部分标注以获取具有部分标签的真夜间车流图片数据；基于CycleGAN构建夜间桥梁交通检测数据增强模型，并对夜间桥梁交通检测数据增强模型进行训练；对训练后的夜间桥梁交通检测数据增强模型进行多尺度筛选；根据具有完整标签的真实日间车流图片数据和最终的夜间桥梁交通检测数据增强模型，获取增强的夜间桥梁交通检测数据。本发明不仅提高YOLOv5目标检测模型的识别精度和泛化能力，达到了生成式数据增强的效果，而且标注数据的整体迁移也减少了夜间车流数据标注的人力成本。