Models
This page contains pre-defined model configs. Copy the model name to use it with the JetNet tools.
Classification Models
| Class | Name | Config |
|---|---|---|
| TorchvisionModel | jetnet.torchvision.RESNET18_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET18_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET18_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET18_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.RESNET34_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET34_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET34_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET34_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.RESNET50_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET50_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET50_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET50_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.RESNET101_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET101_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET101_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET101_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.RESNET152_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET152_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET152_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.RESNET152_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.DENSENET121_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET121_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET121_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET121_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.DENSENET161_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET161_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET161_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET161_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.DENSENET169_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET169_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET169_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET169_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.DENSENET201_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET201_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET201_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.DENSENET201_IMAGENET_TRT_INT8 | json |
| TorchvisionModel | jetnet.torchvision.MOBILENET_V2_IMAGENET | json |
| TorchvisionModelTRT | jetnet.torchvision.MOBILENET_V2_IMAGENET_TRT | json |
| TorchvisionModelTRT | jetnet.torchvision.MOBILENET_V2_IMAGENET_TRT_FP16 | json |
| TorchvisionModelTRT | jetnet.torchvision.MOBILENET_V2_IMAGENET_TRT_INT8 | json |
Detection Models
| Class | Name | Config |
|---|---|---|
| YOLOX | jetnet.yolox.YOLOX_L | json |
| YOLOXTRT | jetnet.yolox.YOLOX_L_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_L_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_L_TRT_INT8 | json |
| YOLOX | jetnet.yolox.YOLOX_M | json |
| YOLOXTRT | jetnet.yolox.YOLOX_M_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_M_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_M_TRT_INT8 | json |
| YOLOX | jetnet.yolox.YOLOX_S | json |
| YOLOXTRT | jetnet.yolox.YOLOX_S_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_S_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_S_TRT_INT8 | json |
| YOLOX | jetnet.yolox.YOLOX_X | json |
| YOLOXTRT | jetnet.yolox.YOLOX_X_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_X_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_X_TRT_INT8 | json |
| YOLOX | jetnet.yolox.YOLOX_TINY | json |
| YOLOXTRT | jetnet.yolox.YOLOX_TINY_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_TINY_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_TINY_TRT_INT8 | json |
| YOLOX | jetnet.yolox.YOLOX_NANO | json |
| YOLOXTRT | jetnet.yolox.YOLOX_NANO_TRT | json |
| YOLOXTRT | jetnet.yolox.YOLOX_NANO_TRT_FP16 | json |
| YOLOXTRT | jetnet.yolox.YOLOX_NANO_TRT_INT8 | json |
| MMDet | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO | json |
| Torch2trtModel | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO_TRT_FP16 | json |
| Torch2trtModel | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO_TRT | json |
Instance Segmentation Models
| Class | Name | Config |
|---|---|---|
| MMDet | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO | json |
| Torch2trtModel | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO_TRT_FP16 | json |
| Torch2trtModel | jetnet.mmdet.MASK_RCNN_R50_FPN_1X_COCO_TRT | json |
Pose Models
| Class | Name | Config |
|---|---|---|
| TRTPose | jetnet.trt_pose.RESNET18_BODY_224X224 | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_BODY_224X224_TRT | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_BODY_224X224_TRT_FP16 | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_BODY_224X224_TRT_INT8 | json |
| TRTPose | jetnet.trt_pose.DENSENET121_BODY_256X256 | json |
| Torch2trtModel | jetnet.trt_pose.DENSENET121_BODY_256X256_TRT | json |
| Torch2trtModel | jetnet.trt_pose.DENSENET121_BODY_256X256_TRT_FP16 | json |
| Torch2trtModel | jetnet.trt_pose.DENSENET121_BODY_256X256_TRT_INT8 | json |
| TRTPose | jetnet.trt_pose.RESNET18_HAND_224X224 | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_HAND_224X224_TRT | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_HAND_224X224_TRT_FP16 | json |
| Torch2trtModel | jetnet.trt_pose.RESNET18_HAND_224X224_TRT_INT8 | json |
Text Detection Models
| Class | Name | Config |
|---|---|---|
| EasyOCR | jetnet.easyocr.EASYOCR_EN | json |
| EasyOCRTRT | jetnet.easyocr.EASYOCR_EN_TRT | json |
| EasyOCRTRT | jetnet.easyocr.EASYOCR_EN_TRT_FP16 | json |
| EasyOCRTRT | jetnet.easyocr.EASYOCR_EN_TRT_INT8_FP16 | json |
| MMOCR | jetnet.mmocr.MMOCR_DB_R18_CRNN | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_CRNN_TRT | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_CRNN_TRT_FP16 | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_CRNN_TRT_INT8_FP16 | json |
| MMOCR | jetnet.mmocr.MMOCR_DB_R18_ROBUSTSCANNER | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_ROBUSTSCANNER_TRT | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_ROBUSTSCANNER_TRT_FP16 | json |
| MMOCRTRT | jetnet.mmocr.MMOCR_DB_R18_ROBUSTSCANNER_TRT_INT8_FP16 | json |