hyaihjq / tensorrt_inference

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
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
seresnext	PyTorch	PyTorch example
Yolov4	darknet	darknet and object detection example
yolov5	PyTorch	PyTorch and object detection example