grounding-dino-base

Grounding DINO – base variant (model ID IDEA-Research/grounding-dino-base ) is a state‑of‑the‑art open‑set object detector that combines a Vision Transformer (ViT) backbone with a text encoder to perform

IDEA-Research 3.3M downloads apache-2.0 Zero-Shot Object Detection Top 100
Frameworkstransformerspytorchsafetensors
Tagsgrounding-dinozero-shot-object-detectionvision
Downloads
3.3M
License
apache-2.0
Pipeline
Zero-Shot Object Detection
Author
IDEA-Research

Run grounding-dino-base locally on a Q4KM hard drive

Accelerate your AI workflow with Q4KM hard drives pre‑loaded with Grounding DINO . These high‑speed SSDs ship with the model checkpoint, tokenizer, and sample inference scripts ready to run...

Shop Q4KM Drives

Technical Overview

Grounding DINO – base variant (model ID IDEA-Research/grounding-dino-base) is a state‑of‑the‑art open‑set object detector that combines a Vision Transformer (ViT) backbone with a text encoder to perform zero‑shot detection. Unlike traditional closed‑set detectors that can only recognize a fixed list of categories, Grounding DINO can locate and label any object described by a natural‑language query, making it ideal for dynamic, real‑world scenarios where new classes appear on‑the‑fly.

Key capabilities include:

  • Zero‑shot object detection – detect arbitrary objects using plain text prompts (e.g., “a cat.” “a remote control.”).
  • Open‑set reasoning – the model does not require a predefined label set, enabling flexible downstream pipelines.
  • High‑resolution visual grounding – precise bounding‑box regression aligned with textual concepts.
  • Efficient inference – the base variant balances accuracy and speed, suitable for both research and production.

Architecture highlights:

  • Backbone: DINO (Deformable DETR with a ViT‑L/14 encoder) that provides strong visual features.
  • Text encoder: a frozen CLIP‑style transformer that converts lower‑cased, period‑terminated queries into a shared embedding space.
  • Cross‑modal grounding module: learns to align visual tokens with text embeddings, producing class‑agnostic object proposals that are then filtered by confidence thresholds.
  • Losses: a combination of Hungarian matching, classification, and grounding losses that encourage both spatial precision and semantic relevance.

Intended use cases range from rapid prototyping of visual‑language applications to production‑grade image analysis where labeling budgets are limited. Typical scenarios include content moderation, robotics perception, e‑commerce visual search, and assistive technologies that need to understand user‑provided descriptions on the fly.

Benchmark Performance

Grounding DINO is evaluated primarily on the COCO dataset using the zero‑shot protocol, where the model must detect objects from categories unseen during training. The base variant reports an impressive 52.5 AP (average precision) on COCO zero‑shot, a substantial leap over earlier open‑set detectors. This metric is crucial because it reflects the model’s ability to generalize to novel classes while maintaining precise localization.

Compared to contemporaries such as GLIP and OWL‑ViT, Grounding DINO’s AP is higher while using a comparable amount of parameters, indicating a more effective grounding strategy. The model also achieves strong box‑threshold and text‑threshold trade‑offs (e.g., 0.4 and 0.3 respectively) that can be tuned for precision‑recall balance in production pipelines.

Hardware Requirements

For inference, the base grounding‑dino model fits comfortably on a single modern GPU. Rough estimates are:

  • VRAM: 8 GB is the minimum for a batch size of 1 at 640 × 640 resolution; 12 GB+ is recommended for larger images or batch processing.
  • GPU: NVIDIA RTX‑30 series (e.g., RTX 3080) or AMD equivalents with CUDA/ROCm support provide optimal throughput.
  • CPU: Any recent x86‑64 processor; inference speed is primarily GPU‑bound, but a multi‑core CPU (8 + threads) helps with data preprocessing and post‑processing.
  • Storage: The model checkpoint (~1 GB in safetensors format) plus auxiliary tokenizer files require < 2 GB of disk space.
  • Performance: On an RTX 3080, a single image (640 × 640) processes in ~120 ms (≈8 FPS) with torch‑no‑grad mode, making real‑time applications feasible.

Use Cases

Grounding DINO’s zero‑shot capability unlocks a wide range of applications:

  • Content moderation: Detect prohibited objects (e.g., weapons, nudity) by simply updating the text prompt list without retraining.
  • Robotics & autonomous agents: Enable robots to locate user‑specified items (“a red mug”) in cluttered environments.
  • E‑commerce visual search: Allow shoppers to upload an image and query “a leather backpack” to retrieve matching products.
  • Assistive technology: Provide visually impaired users with spoken descriptions of objects they point to, using dynamic queries.
  • Digital asset management: Auto‑tag large image libraries with custom vocabularies on demand.

Integration is straightforward via the AutoProcessor and AutoModelForZeroShotObjectDetection classes from the 🤗 Transformers library, enabling deployment in Python, C++, or JavaScript through ONNX export.

Training Details

Grounding DINO was trained on a combination of COCO‑style detection data and large‑scale image‑text pairs (e.g., OpenImages, Visual Genome) to learn a shared visual‑language embedding space. The training pipeline follows a two‑stage approach:

  • Pre‑training: A frozen CLIP text encoder is aligned with a ViT backbone using contrastive loss on image‑text pairs.
  • Grounded detection fine‑tuning: The model is further trained on annotated bounding boxes with associated captions, employing Hungarian matching, classification, and grounding losses.

The authors reported training on a cluster of 8 × NVIDIA A100 GPUs for roughly 48 hours, using a batch size of 16 and a learning rate schedule that decays after 12 epochs. The base variant contains ~86 M parameters, making it more lightweight than the large variant while preserving high zero‑shot AP. Fine‑tuning on domain‑specific data is supported via the same AutoModelForZeroShotObjectDetection interface, allowing practitioners to adapt the model to specialized vocabularies or higher‑resolution inputs.

Licensing Information

The model card lists the license as Apache‑2.0, a permissive open‑source license that grants broad rights to use, modify, and distribute the software. However, the License field in the repository metadata is marked “unknown,” which may indicate that the underlying weights or certain assets have not been explicitly re‑licensed. In practice:

  • Commercial use: Apache‑2.0 permits commercial deployment, provided you retain the license notice and do not use trademarks without permission.
  • Restrictions: If any component remains “unknown,” you should verify with the author (IDEA‑Research) before embedding the model in a product that is sold or offered as a service.
  • Attribution: You must include a copy of the Apache‑2.0 license and credit the original authors (Shilong Liu et al.) in any distributed derivative work.

For safety, review the model files and the discussion forum for any community‑reported licensing nuances.

Pre-loaded AI models. Ready to run.

Skip the downloads. Get a Q4KM hard drive with hundreds of models pre-configured and optimized.

Shop Q4KM Hard Drives