PET-DINO: Unifying Visual Cues into Grounding DINO with Prompt-Enriched Training
Organizations: YouTu Lab, Tencent · Huazhong University of Science and Technology · Kling Team, Kuaishou Technology
Abstract
Open-Set Object Detection (OSOD) enables recognition of novel categories beyond fixed classes but faces challenges in aligning text representations with complex visual concepts and the scarcity of image-text pairs for rare categories. This results in suboptimal performance in specialized domains or with complex objects. Recent visual-prompted methods partially address these issues but often involve complex multi-modal designs and multi-stage optimizations, prolonging the development cycle. Additionally, effective training strategies for data-driven OSOD models remain largely unexplored. To address these challenges, we propose PET-DINO, a universal detector supporting both text and visual prompts. Our Alignment-Friendly Visual Prompt Generation (AFVPG) module builds upon an advanced text-prompted detector, addressing the limitations of text representation guidance and reducing the development cycle. We introduce two prompt-enriched training strategies: Intra-Batch Parallel Prompting (IBP) at the iteration level and Dynamic Memory-Driven Prompting (DMD) at the overall training level. These strategies enable simultaneous modeling of multiple prompt routes, facilitating parallel alignment with diverse real-world usage scenarios. Comprehensive experiments demonstrate that PET-DINO exhibits competitive zero-shot object detection capabilities across various prompt-based detection protocols. These strengths can be attributed to inheritance-based philosophy and prompt-enriched training strategies, which play a critical role in building an effective generic object detector. Project page: https://fuweifuvtoo.github.io/pet-dino.
Figures & tables
| Method | Open | Prompt | Backbone | VLM sup. | Data | Training Data | COCO | LVIS | ODinW35 |
|---|---|---|---|---|---|---|---|---|---|
| Source | Type | Size | |||||||
| T-Rex2 [ 10 ] | N | Visual-I | Swin-T | CLIP | 3.1M | O365, OpenImages, HierText, CrowdHuman, SA-1B | 56.6 | 59.3 | 37.7 |
| CP-DETR-T [ 1 ] | N | Visual-I | Swin-T | CLIP | 3.3M | O365, OpenImages, GoldG, V3Det | 61.8 | 64.1 | 41.0 |
| PET-DINO | Y | Visual-I | Swin-T | - | 0.6M | O365 | 64.0 | 61.8 | 38.8 |
| PET-DINO | Y | Visual-I | Swin-T | - | 2.5M | O365, OpenImages, V3Det | 64.3 | 64.5 | 48.3 |
| T-Rex2 [ 10 ] | N | Visual-I | Swin-L | CLIP | 3.1M | O365, OpenImages, HierText, CrowdHuman, SA-1B | 58.5 | 62.5 | 39.7 |
| Method | Open | Prompt | Backbone | VLM sup. | Data | Training Data | COCO | LVIS | ODinW35 |
|---|---|---|---|---|---|---|---|---|---|
| Source | Type | Size | |||||||
| T-Rex2 [ 10 ] | N | Visual-G | Swin-T | CLIP | 3.1M | O365, OpenImages, HierText, CrowdHuman, SA-1B | 38.8 | 37.4 | 23.6 |
| YOLOE-v11-S [ 27 ] | Y | Visual-G | YOLO-v11-S | MobileCLIP | 1.4M | O365, GoldG | - | 26.2 | - |
| PET-DINO | Y | Visual-G | Swin-T | - | 0.6M | O365 | 40.3 | 29.6 | 20.4 |
| PET-DINO | Y | Visual-G | Swin-T | - | 2.5M | O365, OpenImages, V3Det | 38.4 | 31.5 | 25.5 |
| T-Rex2 [ 10 ] | N | Visual-G | Swin-L | CLIP | 3.1M | O365, OpenImages, HierText, CrowdHuman, SA-1B | 46.5 | 47.6 | 27.8 |
| Method | Prompt | Backbone | Training Data | COCO | LVIS | ODinW35 |
|---|---|---|---|---|---|---|
| Type | ||||||
| Grounding DINO [ 20 ] | Text | Swin-T | O365, GoldG, Cap4M | 48.4 | 27.4 | 22.3 |
| MM-Grounding-DINO [ 36 ] | Text | Swin-T | O365, GoldG, V3Det | 50.6 | 40.5 | 21.4 |
| PET-DINO | Text | Swin-T | O365 † | 49.8 ↓0.8 | 37.8 ↓2.7 | 20.6 ↓0.8 |
| Grounding DINO [ 20 ] | Text | Swin-L | O365, OpenImages, GoldG | 52.5 | 33.9 | 26.1 |
| MM-Grounding-DINO [ 36 ] | Text | Swin-L | O365V2, OpenImage, GoldG | 53.0 | 36.7 | 23.9 |
| Method | Prompt | Backbone | Training Data | COCO |
|---|---|---|---|---|
| Type | ||||
| X-Decoder T [ 39 ] | Text | Focal-T | COCO+CC3M+… | 43.6 |
| OpenSeed T [ 34 ] | Text | Swin-T | COCO+O365 | 52.0 |
| OpenSeed L [ 34 ] | Text | Swin-L | COCO+O365 | 58.2 |
| MI Grounding-S [ 35 ] | Image | ViT-L | COCO+LVIS | 54.7 |
| DINOv [ 14 ] | Visual-G | Swin-T | COCO+SA-1B | 45.2 |
| T-Rex2 VP encoder | AFVPG | COCO Text | COCO Visual-I | COCO Visual-G |
|---|---|---|---|---|
| ✓ | 49.6 | 59.2 | 37.6 | |
| ✓ | 49.8 | 64.0 ↑4.8 | 40.3 ↑2.7 |
| Row | AFVPG | IBP | DMD | COCO Text | COCO Visual-I | COCO Visual-G |
|---|---|---|---|---|---|---|
| 0 | 50.6 | - | - | |||
| 1 | ✓ | 49.7 | 67.0 | 12.5 | ||
| 2 | ✓ | ✓ | 49.8 | 63.5 | 24.7 | |
| 3 | ✓ | ✓ | 49.6 | 63.2 | 37.2 | |
| 4 | ✓ | ✓ | ✓ | 49.8 | 64.0 | 40.3 |
| Method | Training Data | COCO | COCO | |
|---|---|---|---|---|
| MM-Grounding-DINO | PET-DINO | visual-I | visual-G | |
| PET-DINO-T | - | O365 | 62.1 | 32.7 |
| (from scratch) | ||||
| PET-DINO-T | O365, GoldG, V3Det | O365 | 64.0 | 40.3 |
| (from pre-trained) | ||||
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
Object Detection
Object detection locates and classifies objects in images or video, balancing localization accuracy, robustness, and inference speed.
See all 624 papers