diff --git a/README.md b/README.md index cc7c7c9..a05938d 100644 --- a/README.md +++ b/README.md @@ -10,6 +10,10 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a ## News +- `26 Apr 2021`. [makaveli10](https://github.com/makaveli10) add Inceptionv4. +- `25 Apr 2021`. YOLOv5 updated to v5.0, supporting s/m/l/x/s6/m6/l6/x6. +- `23 Apr 2021`. [irvingzhang0512](https://github.com/irvingzhang0512) add TSM: Temporal Shift Module for Efficient Video Understanding, ICCV2019. +- `23 Apr 2021`. [freedenS](https://github.com/freedenS) implement MaskRCNN, till now the MOST complicated model in this repo. - `16 Apr 2021`. [irvingzhang0512](https://github.com/irvingzhang0512) implement lenet and resnet50 with Python API, [freedenS](https://github.com/freedenS) implement FasterRCNN with five plugins, cheers! - `2 Apr 2021`. [mingyu6yang](https://github.com/mingyu6yang) added a python wrapper for retinaface, [makaveli10](https://github.com/makaveli10) added DenseNet-121. - `17 Mar 2021`. [wuzuowuyou](https://github.com/wuzuowuyou) added refinedet, which utilized libtorch to do postprocessing. @@ -21,10 +25,6 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a - `27 Dec 2020`. HRNet-Semantic-Segmentation added by [BaofengZan](https://github.com/BaofengZan). - `4 Dec 2020`. DBNet dynamic input shape support by [BaofengZan](https://github.com/BaofengZan), YOLOv3 int8, PSENet(tensorflow) text detection by [upczww](https://github.com/upczww). - `19 Nov 2020`. YOLOv3-SPP supports dynamic input shape, including a dynamic plugin. -- `17 Nov 2020`. [AlfengYuan](https://github.com/AlfengYuan) added a Dockerfile. -- `7 Nov 2020`. All models migrated to trt7 API, and clean up the master branch. -- `29 Oct 2020`. First INT8 quantization implementation! Please check retinaface. -- `23 Oct 2020`. Add a .wts model zoo for quick evaluation. ## Tutorials @@ -56,20 +56,19 @@ Following models are implemented. |[lenet](./lenet) | the simplest, as a "hello world" of this project | |[alexnet](./alexnet)| easy to implement, all layers are supported in tensorrt | |[googlenet](./googlenet)| GoogLeNet (Inception v1) | -|[inception](./inceptionv3)| Inception v3 | +|[inception](./inception)| Inception v3, v4 | |[mnasnet](./mnasnet)| MNASNet with depth multiplier of 0.5 from the paper | -|[mobilenetv2](./mobilenetv2)| MobileNet V2 | -|[mobilenetv3](./mobilenetv3)| V3-small, V3-large. | +|[mobilenet](./mobilenet)| MobileNet v2, v3-small, v3-large | |[resnet](./resnet)| resnet-18, resnet-50 and resnext50-32x4d are implemented | |[senet](./senet)| se-resnet50 | -|[shufflenet](./shufflenetv2)| ShuffleNetV2 with 0.5x output channels | +|[shufflenet](./shufflenetv2)| ShuffleNet v2 with 0.5x output channels | |[squeezenet](./squeezenet)| SqueezeNet 1.1 model | |[vgg](./vgg)| VGG 11-layer model | |[yolov3-tiny](./yolov3-tiny)| weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) | |[yolov3](./yolov3)| darknet-53, weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) | |[yolov3-spp](./yolov3-spp)| darknet-53, weights and pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) | |[yolov4](./yolov4)| CSPDarknet53, weights from [AlexeyAB/darknet](https://github.com/AlexeyAB/darknet#pre-trained-models), pytorch implementation from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) | -|[yolov5](./yolov5)| yolov5-s/m/l/x v1.0 v2.0 v3.0 v3.1, pytorch implementation from [ultralytics/yolov5](https://github.com/ultralytics/yolov5) | +|[yolov5](./yolov5)| yolov5 v1.0-v5.0, pytorch implementation from [ultralytics/yolov5](https://github.com/ultralytics/yolov5) | |[retinaface](./retinaface)| resnet50 and mobilnet0.25, weights from [biubug6/Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface) | |[arcface](./arcface)| LResNet50E-IR, weights from [deepinsight/insightface](https://github.com/deepinsight/insightface) | |[retinafaceAntiCov](./retinafaceAntiCov)| mobilenet0.25, weights from [deepinsight/insightface](https://github.com/deepinsight/insightface), retinaface anti-COVID-19, detect face and mask attribute | @@ -84,7 +83,8 @@ Following models are implemented. |[lprnet](./lprnet)| LPRNet, pytorch implementation from [xuexingyu24/License_Plate_Detection_Pytorch](https://github.com/xuexingyu24/License_Plate_Detection_Pytorch) | |[refinedet](./refinedet)| RefineDet, pytorch implementation from [luuuyi/RefineDet.PyTorch](https://github.com/luuuyi/RefineDet.PyTorch) | |[densenet](./densenet)| DenseNet-121, from torchvision.models | -|[rcnn](./rcnn)| FasterRCNN, model from [detectron2](https://github.com/facebookresearch/detectron2) | +|[rcnn](./rcnn)| FasterRCNN and MaskRCNN, model from [detectron2](https://github.com/facebookresearch/detectron2) | +|[tsm](./tsm)| TSM: Temporal Shift Module for Efficient Video Understanding, ICCV2019 | ## Model Zoo