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 |