Model Details

View as Markdown

Model Architecture

Architecture Type: Transformer
Network Architecture: SAM-like

Input

Input Type(s): Computed Tomography (CT) Image
Input Format(s): (Neuroimaging Informatics Technology Initiative) NIfTI
Input Parameters: Three-Dimensional (3D)
Other Properties Related to Input: Array of Class/Point Information

Output

Output Type(s): Image
Output Format: NIfTI
Output Parameters: 3D

Software Integration

Latest model version: 0.5.7

Runtime Engine(s): MONAI Core v.1.4

Supported Hardware Microarchitecture Compatibility:

  • NVIDIA Ampere
  • NVIDIA Hopper

[Preferred/Supported] Operating System(s): Linux

Inference

Engine: Triton
Test Hardware: A100, H100, L40s, RTX 6000 Ada

Supported Anatomy

You can find a comprehensive list of all the classes available in the model in the VISTA-3D labels.json.

Ethical Considerations

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 supporting 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. Please report security vulnerabilities or NVIDIA AI Concerns here.

License

The NIM container and model are governed by the NVIDIA software and model evaluation license agreement.