基于域适应与类别对比的夜间目标检测方法

Nighttime Object Detection Method Based on Domain Adaptation and Category Contrast

夜间目标检测任务中,目标能见度低,难以对图像数据进行大量标注,导致使用监督方法进行大规模数据训练的困难,而当使用较小规模标注数据训练时监督方法容易产生过拟合,预测准确性差。针对这些问题,本文提出一种无监督域适应夜间目标检测模型,使用有标注白昼图像和无标注夜间图像训练。模型使用了昼夜图像增强方法,减小昼夜域间隙并提升夜间训练数据的复杂性以丰富特征学习;将多尺度通道注意力引入Faster-RCNN模型,提升感知多尺度特征的能力;使用类别对比学习方法获得具有鉴别性和域间不变性的类别特征。在城市交通数据集BDD100K和SODA10M上进行的实验表明,本文方法性能优于常用的域适应目标检测方法。

In the task of nighttime object detection,the visibility of the object is low,and it is difficult to annotate a large amount of image data,which makes it difficult to use supervised methods for large-scale data training.And when using a small-scale annotated data for training supervised methods tend to overfit on the training data,resulting in poor prediction accuracy.To address these issues,an unsupervised domain adaptation nighttime object detection model using labeled daytime images and unlabeled nighttime images for training is proposed in this paper.Day-night image enhancement methods is employed in the model to reduce the domain gap and enhance the complexity of nighttime training data to enrich feature learning;multi-scale channel attention is introduced to the Faster-RCNN model to enhance its ability to perceive multi-scale features;and class-level contrastive learning is used to obtain discriminative and domain-invariant class features.Experiments conducted on the urban traffic datasets BDD100K and SODA10M demonstrate that the performance of the proposed method exceeds that of commonly used domain adaptation target detection methods.