CamDiff: Camouflage Image Augmentation via Diffusion Model

Xue-Jing Luo,Shuo Wang,Zongwei Wu,Christos Sakaridis,Yun Cheng,Deng-Ping Fan,Luc Van Gool
DOI: https://doi.org/10.48550/arXiv.2304.05469
2023-04-12
Abstract:The burgeoning field of camouflaged object detection (COD) seeks to identify objects that blend into their surroundings. Despite the impressive performance of recent models, we have identified a limitation in their robustness, where existing methods may misclassify salient objects as camouflaged ones, despite these two characteristics being contradictory. This limitation may stem from lacking multi-pattern training images, leading to less saliency robustness. To address this issue, we introduce CamDiff, a novel approach inspired by AI-Generated Content (AIGC) that overcomes the scarcity of multi-pattern training images. Specifically, we leverage the latent diffusion model to synthesize salient objects in camouflaged scenes, while using the zero-shot image classification ability of the Contrastive Language-Image Pre-training (CLIP) model to prevent synthesis failures and ensure the synthesized object aligns with the input prompt. Consequently, the synthesized image retains its original camouflage label while incorporating salient objects, yielding camouflage samples with richer characteristics. The results of user studies show that the salient objects in the scenes synthesized by our framework attract the user's attention more; thus, such samples pose a greater challenge to the existing COD models. Our approach enables flexible editing and efficient large-scale dataset generation at a low cost. It significantly enhances COD baselines' training and testing phases, emphasizing robustness across diverse domains. Our newly-generated datasets and source code are available at <a class="link-external link-https" href="https://github.com/drlxj/CamDiff" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is that in the Camouflaged Object Detection (COD) task, existing methods perform poorly when dealing with images that contain both salient objects and camouflaged objects. Specifically, current COD models may misclassify salient objects as camouflaged objects by mistake, although these two features are actually contradictory to each other. This problem may stem from the lack of multi - mode training images, resulting in insufficient performance of the model in terms of saliency robustness. To solve this problem, the paper introduces CamDiff, a new method based on the diffusion model, aiming to enhance the diversity of camouflage scenes by synthesizing salient objects. Specifically, CamDiff utilizes the Latent Diffusion Model (LDM) to synthesize salient objects in camouflage scenes and uses the Contrastive Language - Image Pre - training (CLIP) model to prevent synthesis failures and ensure that the synthesized objects are consistent with the input prompts. In this way, the synthesized images retain the original camouflage labels while incorporating salient objects, generating camouflage samples with more abundant characteristics. Through this method, CamDiff can not only flexibly edit images but also efficiently generate large - scale datasets, thereby significantly improving the performance of COD baseline models in the training and testing stages, especially in terms of robustness in different fields. ### Main contributions of the paper: 1. **Introduction of CamDiff**: This framework can generate salient objects on camouflage scenes while retaining the original labels. This makes it possible to combine contrastive patterns in real images without additional learning and annotation costs. 2. **Experimental verification**: The authors created a new test set (Diff - COD) and tested the performance of existing state - of - the - art COD methods on this test set. The results show that current COD methods are not robust enough when dealing with scenes containing salient objects. 3. **Generation of a new training set**: A new training set (Diff - COD training set) was generated through CamDiff and used to train existing COD models. Experimental results show that this new training set can improve the model's robustness to saliency. ### Summary: The paper provides a new perspective for understanding the concept of camouflage and lays the foundation for promoting this rapidly developing field by introducing CamDiff, a powerful camouflage synthesis tool.