Skip to content

taxuezcy/tensorrt_inference

 
 

Repository files navigation

TensorRT Models Deploy from ONNX

Install Depends

see INSTALL.md

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
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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 66.1%
  • Python 28.4%
  • CMake 5.0%
  • C 0.5%