NGC Supported Models

MediaTek Genio supports the following pretrained ONNX models from NGC:

NGC Supported Pretrained ONNX Models on Genio

Category

Architecture

Model

File

Object Detection

DetectNet_V2

PeopleNet

resnet34_peoplenet.onnx

resnet34_peoplenet_int8.onnx

LPDNet

LPDNet_usa_pruned_tao5.onnx

Pose Estimation

Visual Geometry Group (CNN)

BodyPoseNet

model.onnx

Object Recognition

ResNet

Retail Object Recognition

retail_object_detection.onnx

PeopleSemSegNet

Vanilla UNet

PeopleSemSegNet

peoplesemsegnet_vanilla_unet_dynamic_etlt_int8_fp16.onnx

Shuffle UNet

peoplesemsegnet_shuffleseg.onnx

PeopleSemSegNet AMR

Vanilla UNet

PeopleSemSegNet_AMR

peoplesemsegnet_amr.onnx

Shuffle UNet

peoplesemsegnet_amr_rel.onnx

Object Reidentification

ResNet

ReIdentificationNet

resnet50_market1501_aicity156.onnx

Action Recognition

ResNet

Action Recognition Net

resnet18_2d_rgb_hmdb5_32.onnx