| mlp | the very basic model for starters, properly documented |
| lenet | the simplest, as a “hello world” of this project |
| alexnet | easy to implement, all layers are supported in tensorrt |
| googlenet | GoogLeNet (Inception v1) |
| inception | Inception v3, v4 |
| mnasnet | MNASNet with depth multiplier of 0.5 from the paper |
| mobilenet | MobileNet v2, v3-small, v3-large |
| resnet | resnet-18, resnet-50 and resnext50-32x4d are implemented |
| senet | se-resnet50 |
| shufflenet | ShuffleNet v2 with 0.5x output channels |
| squeezenet | SqueezeNet 1.1 model |
| vgg | VGG 11-layer model |
| yolov3-tiny | weights and pytorch implementation from ultralytics/yolov3 |
| yolov3 | darknet-53, weights and pytorch implementation from ultralytics/yolov3 |
| yolov3-spp | darknet-53, weights and pytorch implementation from ultralytics/yolov3 |
| yolov4 | CSPDarknet53, weights from AlexeyAB/darknet, pytorch implementation from ultralytics/yolov3 |
| yolov5 | yolov5 v1.0-v6.0, pytorch implementation from ultralytics/yolov5 |
| retinaface | resnet50 and mobilnet0.25, weights from biubug6/Pytorch_Retinaface |
| arcface | LResNet50E-IR, LResNet100E-IR and MobileFaceNet, weights from deepinsight/insightface |
| retinafaceAntiCov | mobilenet0.25, weights from deepinsight/insightface, retinaface anti-COVID-19, detect face and mask attribute |
| dbnet | Scene Text Detection, weights from BaofengZan/DBNet.pytorch |
| crnn | pytorch implementation from meijieru/crnn.pytorch |
| ufld | pytorch implementation from Ultra-Fast-Lane-Detection, ECCV2020 |
| hrnet | hrnet-image-classification and hrnet-semantic-segmentation, pytorch implementation from HRNet-Image-Classification and HRNet-Semantic-Segmentation |
| psenet | PSENet Text Detection, tensorflow implementation from liuheng92/tensorflow_PSENet |
| ibnnet | IBN-Net, pytorch implementation from XingangPan/IBN-Net, ECCV2018 |
| unet | U-Net, pytorch implementation from milesial/Pytorch-UNet |
| repvgg | RepVGG, pytorch implementation from DingXiaoH/RepVGG |
| lprnet | LPRNet, pytorch implementation from xuexingyu24/License_Plate_Detection_Pytorch |
| refinedet | RefineDet, pytorch implementation from luuuyi/RefineDet.PyTorch |
| densenet | DenseNet-121, from torchvision.models |
| rcnn | FasterRCNN and MaskRCNN, model from detectron2 |
| tsm | TSM: Temporal Shift Module for Efficient Video Understanding, ICCV2019 |
| scaled-yolov4 | yolov4-csp, pytorch from WongKinYiu/ScaledYOLOv4 |
| centernet | CenterNet DLA-34, pytorch from xingyizhou/CenterNet |
| efficientnet | EfficientNet b0-b8 and l2, pytorch from lukemelas/EfficientNet-PyTorch |
| detr | DE⫶TR, pytorch from facebookresearch/detr |
| swin-transformer | Swin Transformer - Semantic Segmentation, only support Swin-T. The Pytorch implementation is microsoft/Swin-Transformer |
| real-esrgan | Real-ESRGAN. The Pytorch implementation is real-esrgan |
| superpoint | SuperPoint. The Pytorch model is from magicleap/SuperPointPretrainedNetwork |
