see INSTALL.md
models | framework | instruction |
---|---|---|
lenet | PyTorch | An example from model training to TensorRT model deploy |
alexnet | MXNet Gluon | MXNet Gluon example |
seresnext | PyTorch | PyTorch example |
arcface | MXNet Symbol | MXNet Symbol and face recognition example |
RetinaFace | MXNet Symbol | MXNet Symbol and face detection example |
Yolov4 | darknet | darknet and object detection example |
yolov5 | PyTorch | PyTorch and object detection example |