YuLillll commited on
Commit
428296c
Β·
verified Β·
1 Parent(s): b869ec3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +91 -3
README.md CHANGED
@@ -1,3 +1,91 @@
1
- ---
2
- license: unknown
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: unknown
3
+ base_model:
4
+ - black-forest-labs/FLUX.1-Fill-dev
5
+ - black-forest-labs/FLUX.1-dev
6
+ - black-forest-labs/FLUX.1-Redux-dev
7
+ ---
8
+ # [NeurIPS 2025] Domain-RAG: Retrieval-Guided Compositional Image Generation for Cross-Domain Few-Shot Object Detection
9
+
10
+ [πŸ”₯ Paper (NeurIPS 2025)](https://arxiv.org/abs/2506.05872) | [🌐 Project Page](https://yuli-cs.net/papers/domain-rag) | [πŸ“¦ Dataset Scripts](#dataset-preparation) | [🧠 Model Zoo](#pretrained-models) | [πŸš€ Quick Start](#quick-start) | [πŸŽ₯ Video](#video) | [πŸ“Š Evaluation](#evaluation) | [πŸ“ž Contact](#contact)
11
+
12
+ ---
13
+
14
+ **Domain-RAG** is a novel retrieval-augmented generative framework designed for **Cross-Domain Few-Shot Object Detection (CD-FSOD)**. We leverage large-scale vision-language models (GroundingDINO), a curated COCO-style retrieval corpus, and Flux-based background generation to synthesize diverse, domain-aware training data that enhances FSOD generalization under domain shift.
15
+
16
+ <p align="center">
17
+ <img src="assets/framework.svg" alt="DomainRAG Pipeline" width="700"/>
18
+ </p>
19
+
20
+ ---
21
+
22
+ ## ✨ Highlights
23
+
24
+ - πŸ” **Retrieval-Augmented Generation**: retrieve semantically similar source images for novel-class prompts.
25
+ - 🎨 **Flux-Redux Integration**: compose diverse backgrounds with target foregrounds for domain-aligned generation.
26
+ - πŸ“¦ **Support for Multiple Target Domains**: ArTAXOr, Clipart1k, DIOR, DeepFish, UODD, NEU-DET, and more.
27
+ - πŸ§ͺ **Strong Benchmarks**: surpasses GroundingDINO baseline in 1-shot and 5-shot CD-FSOD across 6 datasets.
28
+
29
+ ---
30
+
31
+ ## πŸ”§ Installation
32
+
33
+ ```bash
34
+ git clone https://github.com/LiYu0524/Domain-RAG.git
35
+ cd Domain-RAG
36
+ conda create -n domainrag python=3.10
37
+ conda activate domainrag
38
+ pip install -r requirements.txt
39
+ ```
40
+
41
+
42
+ ## Pretrained Models
43
+
44
+ we will relase the fine-tuned grounding-dino model soon
45
+
46
+ ## Dataset Preparation
47
+
48
+ You can prepare CDFSOD with [CDVITO](https://github.com/lovelyqian/CDFSOD-benchmark?tab=readme-ov-file)
49
+
50
+ ## Quick start
51
+
52
+ You can refer to `./domainrag.sh`
53
+
54
+
55
+
56
+ ## Video
57
+
58
+ Walkthrough video(Chinese version): [Watch here](https://www.bilibili.com/video/BV1YznKzkEEK/?spm_id_from=333.337.search-card.all.click&vd_source=23bede4ceb3dc1ea2ffc645933850555)
59
+
60
+ ## Contact
61
+
62
+ For questions and collaboration, please contact:
63
+
64
+ - Yu Li : `<liyu24@m.fudan.edu.cn>`
65
+
66
+
67
+ ## Citation
68
+
69
+ If you find **Domain-RAG** useful in your research, please cite:
70
+
71
+ ```bibtex
72
+ @inproceedings{li2025domainrag,
73
+ author={Li, Yu and Qiu, Xingyu and Fu, Yuqian and Chen, Jie and Qian, Tianwen and Zheng, Xu and Paudel, Danda Pani and Fu, Yanwei and Huang, Xuanjing and Van Gool, Luc and others},
74
+ booktitle = {Advances in Neural Information Processing Systems},
75
+ title = {Domain-RAG: Retrieval-Guided Compositional Image Generation for Cross-Domain Few-Shot Object Detection},
76
+ year = {2025}
77
+ }
78
+
79
+ ```
80
+
81
+ If you find **CD-Vito** useful in your research, please cite:
82
+ ```bibtex
83
+ @inproceedings{fu2024cross,
84
+ title={Cross-domain few-shot object detection via enhanced open-set object detector},
85
+ author={Fu, Yuqian and Wang, Yu and Pan, Yixuan and Huai, Lian and Qiu, Xingyu and Shangguan, Zeyu and Liu, Tong and Fu, Yanwei and Van Gool, Luc and Jiang, Xingqun},
86
+ booktitle={European Conference on Computer Vision},
87
+ pages={247--264},
88
+ year={2024},
89
+ organization={Springer}
90
+ }
91
+ ```