Low-light few-shot object detection via curve contrast enhancement and flow-encoder-based variational autoencoder
摘要
Aiming at the problem of insufficient samples in low-light object detection in some environments, a low-light few-shot object detection method based on curve contrast enhancement and flow-encoder-based variational autoencoder (CCEFVAE) is proposed. Our approach involves designing a CCE module to enhance the detailed features and contrast of low-light images by deriving a relationship expression between the enhanced image and the low-light image through the recursive relationship of high-order curves. The lumination estimation module in the CCE module estimates the parameters of the expression to calculate the pixel values of the enhanced image. Moreover, we propose an FVAE module to improve the decoupling of support features by combining the flow model encoder with the variational autoencoder, facilitating subsequent feature aggregation and classification. To ensure the consistency of the loss function of the flow model with the few-shot object detection loss, we design a negative Jacobian determinant transformation function. This enables direct addition of the two losses, allowing for unified optimization. Experimental results demonstrate that our proposed algorithm outperforms mainstream few-shot object detection models by an average of 13.1–23% in average after training on the low-light dataset (ExDark), and shows an average improvement of 5.8% compared to the state-of-the-art (SOTA) few-shot object detection model VFA. When trained on the normal lighting dataset (PASCAL VOC), the proposed algorithm exhibits a 1.7% improvement in average compared to VFA.