Feature Extraction
Transformers
Safetensors
custom_code
C-RADIO / README.md
gheinrich's picture
Update README.md
fa9358c verified
|
raw
history blame
2.75 kB

Model Overview

Description:

This model performs visual feature extraction. For instance, RADIO generates image embeddings that can be used by a downstream model to classify images.

This model is for research and development only.

License/Terms of Use

License

References:

Paper

Model Architecture:

Architecture Type: Neural Network
Network Architecture: Vision Transformer

Input:

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:

Output Type(s): Embeddings
Output Format: Tensor
Output Parameters: 2D
Other Properties Related to Output: Downstream model required to leverage image features

Software Integration:

Runtime Engine(s):

  • TAO- 24.10

Supported Hardware Microarchitecture Compatibility:

  • NVIDIA Ampere
  • NVIDIA Blackwell
  • NVIDIA Jetson
  • NVIDIA Hopper
  • NVIDIA Lovelace
  • NVIDIA Pascal
  • NVIDIA Turing
  • NVIDIA Volta

[Preferred/Supported] Operating System(s):

  • Linux
  • Linux 4 Tegra
  • QNX
  • Windows

Model Version(s):

C-RADIO.

Link: https://huggingface.co/nvidia/C-RADIO

Training, Testing, and Evaluation Datasets:

Training Dataset:

NV-CC-Img-Text-Dataset
** Data Collection Method by dataset

  • Automated
    ** Labeling Method by dataset
  • Not Applicable (no labels are needed)
    Properties: 700 Million Images

Evaluation Dataset:

Link: ImageNet
** Data Collection Method by dataset

  • Automated
    ** Labeling Method by dataset
  • Human

Properties: This dataset spans 1000 object classes and contains 1,281,167 training images, 50,000 validation images and 100,000 test images.

Inference:

Engine: PyTorch
Test Hardware: A100

Ethical Considerations (For NVIDIA Models Only):

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.

Please report security vulnerabilities or NVIDIA AI Concerns here.