用语言驱动的双层风格混合提升单一领域检测器的跨域泛化能力
Language-Driven Dual Style Mixing for Single-Domain Generalized Object Detection
- 通过视觉语言模型语义生成风格多样化图像,实现语义可控的数据增强
- 在图像和特征层面进行双路径风格混合,无需依赖特定检测框架
- 适用于一阶段、两阶段及基于Transformer的主流检测器,跨域效果显著
将单一领域训练的物体检测器泛化到多个未见领域是一项挑战。现有方法通常采用图像或特征增强来提升源域多样性,从而增强检测器鲁棒性。基于视觉-语言模型(VLM)的增强技术已被证明有效,但要求检测器主干网络与VLM图像编码器结构一致,限制了框架选择。为此,我们提出语言驱动的双风格混合(LDDS)方法,充分利用VLM中的语义信息实现源域多样化。首先,构建提示词将VLM中嵌入的风格语义传递至图像转换网络,生成带有明确语义信息的风格多样化图像。其次,提出在图像层面混合多样化图像与源域图像,有效挖掘语义信息用于数据增强,且不依赖特定增强策略。最后,提出双通道架构下的特征级风格混合,使特征增强具有模型无关性,可无缝适配主流检测框架,包括一阶段、两阶段及基于Transformer的检测器。大量实验表明,该方法在多个基准数据集上表现优异,涵盖真实图像转卡通、正常场景转恶劣天气等任务。代码与预训练模型将公开于https://github.com/qinhongda8/LDDS。
原文摘要 · Abstract (English)
Generalizing an object detector trained on a single domain to multiple unseen domains is a challenging task. Existing methods typically introduce image or feature augmentation to diversify the source domain to raise the robustness of the detector. Vision-Language Model (VLM)-based augmentation techniques have been proven to be effective, but they require that the detector's backbone has the same structure as the image encoder of VLM, limiting the detector framework selection. To address this problem, we propose Language-Driven Dual Style Mixing (LDDS) for single-domain generalization, which diversifies the source domain by fully utilizing the semantic information of the VLM. Specifically, we first construct prompts to transfer style semantics embedded in the VLM to an image translation network. This facilitates the generation of style diversified images with explicit semantic information. Then, we propose image-level style mixing between the diversified images and source domain images. This effectively mines the semantic information for image augmentation without relying on specific augmentation selections. Finally, we propose feature-level style mixing in a double-pipeline manner, allowing feature augmentation to be model-agnostic and can work seamlessly with the mainstream detector frameworks, including the one-stage, two-stage, and transformer-based detectors. Extensive experiments demonstrate the effectiveness of our approach across various benchmark datasets, including real to cartoon and normal to adverse weather tasks. The source code and pre-trained models will be publicly available at https://github.com/qinhongda8/LDDS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。