Update README.md

README.md

@@ -4,7 +4,6 @@ license_name: nvidia-open-model-license
 license_link: https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf
 ---
 
-
 # Model Overview
 
 ## Description
@@ -68,6 +67,41 @@ Huggingface: 03/26/2025 via [RADIO Collection of Models](https://huggingface.co/
 **Output Parameters:** 2D <br>
 **Other Properties Related to Output:** Downstream model required to leverage image features <br>
 
+## Usage:
+
+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.
+
+```python
+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.
+
+```Python
+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.
+
 ## Software Integration
 
 **Runtime Engine(s):**
@@ -192,4 +226,3 @@ Model Application(s):                               |  Generation of visual embe
 Describe the life critical impact (if present).   |  Not Applicable
 Use Case Restrictions:                              |  Abide by NVIDIA Open Model License Agreement
 Model and dataset restrictions:            |  The Principle of least privilege (PoLP) is applied limiting access for dataset generation and model development. Restrictions enforce dataset access during training, and dataset license constraints adhered to.
-