tensorrt_inference
tensorrt_inference copied to clipboard
TensorRT Models Deploy from ONNX
Install Depends
see INSTALL.md
Build from Docker
docker build -t tensorrt_inference:0.1.0_rc .
Supported Models
| models | framework | instruction |
|---|---|---|
| lenet | PyTorch | An example from model training to TensorRT model deploy |
| alexnet | MXNet Gluon | MXNet Gluon example |
| arcface | MXNet Symbol | MXNet Symbol and face recognition example |
| CenterFace | ONNX | rewrite ONNX model and face detection example |
| efficientnet | Keras | Keras to ONNX example |
| face_alignment | MXNet Symbol | MXNet Symbol and face key points detection example |
| fast-reid | PyTorch | PyTorch and pedestrian reid example |
| FCN | GluonCV | MXNet GluonCV semantic segmentation example |
| gender-age | MXNet Symbol | MXNet Symbol and face gender and age recognize example |
| ghostnet | PyTorch | PyTorch example |
| MiniFASNet | PyTorch | PyTorch face anti spoofing example |
| mmpose | PyTorch | PyTorch person key points detect example |
| nanodet | PyTorch | PyTorchlightweight anchor-free object detection example |
| RetinaFace | MXNet Symbol | MXNet Symbol and face detection example |
| ScaledYOLOv4 | PyTorch | YOLOv4 large with PyTorch implementation |
| scrfd | PyTorch | PyTorch scrfd face detection example |
| seresnext | PyTorch | PyTorch example |
| Swin-Transformer | timm | timm image classification example |
| yolor | PyTorch | PyTorch and object detection example |
| Yolov4 | darknet | darknet and object detection example |
| yolov5 | PyTorch | PyTorch and object detection example |
| YOLOv6 | PyTorch | PyTorch and object detection example |
| yolov7 | PyTorch | PyTorch and object detection example |