Model reference · open weights
C-RADIOv2-B is an open-weight embedding model from NVIDIA. C-RADIOv2-B (FP32) weighs 196 MB; the smallest configuration that runs it is RTX 3060 12 GB.
What it is
| Released by | NVIDIA |
|---|---|
| Type | Embedding models |
| Task | Image embed |
| Parameters (lead) | 98M |
| Runs with | transformers |
| Released | 2025-01-13 |
| Popularity | 672 downloads / month |
| Weights | 196 MB (C-RADIOv2-B (FP32), file size) |
| Licence | Its own licence terms |
What it runs on
Weights 196 MB (file size) · overhead about 1.1 GB.
| Card | Runs | Counted memory |
|---|---|---|
| RTX 3060 12 GB | fits | 11.6 GB |
| RTX 4060 Ti 16 GB | fits | 15.4 GB |
| RTX 3090 24 GB | fits | 23.4 GB |
| RTX 4090 24 GB | fits | 23.4 GB |
| RTX 5090 32 GB | fits | 31.0 GB |
| L40S 48 GB | fits | 44.0 GB |
| A100 80 GB | fits | 78.2 GB |
| H100 80 GB | fits | 78.1 GB |
| RTX PRO 6000 Blackwell 96 GB | fits | 93.8 GB |
| DGX Spark (GB10) 128 GB unified | fits | 107 GB |
| H200 141 GB | fits | 138 GB |
| B200 180 GB | fits | 176 GB |
Estimates, not measurements: the weights are the build's file size. No cache grows with use; a batch of inputs needs working memory of its own. Counted memory is 92 % of what CUDA reports for the card.
From the model card
[Github] [CVPR 2025] [CVPR 2024]
This model performs visual feature extraction. For instance, RADIO generates image embeddings that can be used by a downstream model to classify images.
C-RADIOv2 models are available in multiple sizes:
C-RADIOv2 was trained for 1M steps (400k more steps than v1), using inverse frequency sampling for data balancing, and PHI Standardization for teacher distribution balancing.
This model is ready for commercial/non-commercial use.
GOVERNING TERMS: Use of this model is governed by the NVIDIA Open Model License Agreement.
Global.
The embeddings generated by this model are expected to be used by a downstream application. For example:
Huggingface: 03/26/2025 via RADIO Collection of Models.
Architecture Type: Neural Network Network Architecture: Vision Transformer
Input Type(s): Image Input Format(s): Red, Green, Blue (RGB) Input Parameters: Two Dimensional (2D) Other Properties Related to Input: Image resolutions up to 2048x2028 in increments of 16 pixels
Output Type(s): Embeddings Output Format: Tensor Output Parameters: 2D Other Properties Related to Output: Downstream model required to leverage image features
RADIO will return a tuple with two tensors.
The summary is similar to the cls_token in ViT and is meant to represent the general concept of the entire image.
It has shape (B,C) with B being the batch dimension, and C being some number of channels.
The spatial_features represent more localized content which should be suitable for dense tasks such as semantic segmentation, or for integration into an LLM.
import torch
from PIL import Image
from transformers import AutoModel, CLIPImageProcessor
hf_repo = "nvidia/C-RADIOv2-B"
image_processor = CLIPImageProcessor.from_pretrained(hf_repo)
model = AutoModel.from_pretrained(hf_repo, trust_remote_code=True)
model.eval().cuda()
image = Image.open('./assets/radio.png').convert('RGB')
pixel_values = image_processor(images=image, return_tensors='pt', do_resize=True).pixel_values
pixel_values = pixel_values.cuda()
summary, features = model(pixel_values)
Spatial features have shape (B,T,D) with T being the flattened spatial tokens, and D being the channels for spatial features. Note that C!=D in general.
Converting to a spatial tensor format can be done using the downsampling size of the model, combined with the input tensor shape. For RADIO, the patch size is 16.
from einops import rearrange
spatial_features = rearrange(spatial_features, 'b (h w) d -> b d h w', h=x.shape[-2] // patch_size, w=x.shape[-1] // patch_size)
The resulting tensor will have shape (B,D,H,W), as is typically seen with computer vision models.
Runtime Engine(s):
Supported Hardware Microarchitecture Compatibility:
[Preferred/Supported] Operating System(s):
Links:
NV-CC-Img-Text-Dataset
Link: ImageNet
Properties: This dataset spans 1000 object classes and contains 1,281,167 training images, 50,000 validation images and 100,000 test images.
Engine: PyTorch Test Hardware: A100
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
For more detailed information on ethical considerations for this model, please see the Model Card++ Explainability, Bias, Safety & Security, and Privacy Subcards below.
Please report security vulnerabilities or NVIDIA AI Concerns here.
Field | Response :---------------------------------------------
Quoted from the model card on Hugging Face — the full card is behind the Hugging Face link above.