GPU-optimized AI, Machine Learning, & HPC Software

SearchSearch thousands of GPU-optimized Containers, pretrained Models, SDKs, and Helm charts—ready to accelerate AI, digital twins, and HPC from cloud to edge.

Displaying 58 results

NVIDIA

PeopleSemSegnet

Semantic segmentation of persons in an image.

Model

NVIDIA

Action Recognition Net

5 class action recognition network to recognize what people do in an image.

Model

NVIDIA

Mask Grounding DINO CC

Open vocabulary multi-modal instance segmentation model trained on commercial data.

Model

Pre-trained SegFormer ImageNet weights

Pre-trained SegFormer weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

Grounding DINO

Open vocabulary multi-modal object detection model trained on commercial data.

Model

NVIDIA

Re-Identification Transformer

SWIN Transformer based Re-Identification network to generate embeddings for identifying persons in different scenes.

Model

NVIDIA

LPDNet

Object Detection network to detect license plates in an image of a car.

Model

—

ReIdentificationNet

Re-Identification network to generate embeddings for identifying persons in different scenes.

Model

NVIDIA

Optical Character Detection

Network to detect characters in an image.

Model

—

Pre-trained DINO NvImageNet weights

Pre-trained DINO weights trained on NvImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

Mask2Former

Binary instance segmentation model trained on COCO data.

Model

—

Pre-trained FAN based NVImageNet Classification weights

Pre-trained FAN weights trained on NVImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

PointPillarNet

Model to detect one or more objects from a LIDAR point cloud file and return 3D bounding boxes.

Model

—

Optical Inspection

Model to detect defects in soldered components on a Printed Circuit Board.

Model

—

Pre-trained FAN based ImageNet Classification weights

Pre-trained FAN weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

Pre-trained Deformable DETR NvImageNet weights

Pre-trained deformable_detr weights trained on NvImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

BodyPose3DNet

3D human pose estimation network to predict 34 keypoints in 3D of a person in an image.

Model

NVIDIA

BEVFusion for 3D Object Detection

BEVFusion model to detect 3D objects from point cloud and RGB data.

Model

NVIDIA

Pose Classification

Pose classification network to classify poses of people from their skeletons.

Model

NVIDIA

Optical Character Recognition

Model to recognise characters from a preceding OCDNet model.

Model

—

Visual ChangeNet Classification

Visual ChangeNet - Classification Models

Model

NVIDIA

CitySemSegFormer

Semantic segmentation of persons in an image.

Model

—

Retail Object Recognition

Embedding generator model to recognize objects on a checkout counter.

Model

—

DINO

Model trained on COCO2017 to detect and classify objects.

Model

1-24 of 58 items