From 8a472a1c6d1f0ec487e3c0a22ddf40f9541d1a40 Mon Sep 17 00:00:00 2001 From: Wang Xinyu Date: Mon, 23 May 2022 10:30:57 +0800 Subject: [PATCH] Update README.md --- README.md | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index bb2fd53..77aedb0 100644 --- a/README.md +++ b/README.md @@ -10,6 +10,8 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a ## News +- `23 May 2022`. [yhpark](https://github.com/yester31): Real-ESRGAN, Practical Algorithms for General Image/Video Restoration. +- `15 Mar 2022`. [sky_hole](https://github.com/wdhao): Swin Transform - Semantic Segmentation. - `19 Oct 2021`. [liuqi123123](https://github.com/liuqi123123) added cuda preprossing for yolov5, preprocessing + inference is 3x faster when batchsize=8. - `18 Oct 2021`. [xupengao](https://github.com/xupengao): YOLOv5 updated to v6.0, supporting n/s/m/l/x/n6/s6/m6/l6/x6. - `31 Aug 2021`. [FamousDirector](https://github.com/FamousDirector): update retinaface to support TensorRT 8.0. @@ -23,8 +25,6 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a - `28 Apr 2021`. [aditya-dl](https://github.com/aditya-dl): mobilenetv2, alexnet, densenet121, mobilenetv3 with python API. - `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. ## Tutorials @@ -41,7 +41,8 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a ## Test Environment -1. GTX1080 / Ubuntu16.04 / cuda10.0 / cudnn7.6.5 / tensorrt7.0.0 / nvinfer7.0.0 / opencv3.3 +1. TensorRT 7.x +2. TensorRT 8.x(Some of the models support 8.x) ## How to run @@ -90,6 +91,8 @@ Following models are implemented. |[centernet](./centernet)| CenterNet DLA-34, pytorch from [xingyizhou/CenterNet](https://github.com/xingyizhou/CenterNet) | |[efficientnet](./efficientnet)| EfficientNet b0-b8 and l2, pytorch from [lukemelas/EfficientNet-PyTorch](https://github.com/lukemelas/EfficientNet-PyTorch) | |[detr](./detr)| DEā«¶TR, pytorch from [facebookresearch/detr](https://github.com/facebookresearch/detr) | +|[swin-transformer](./swin-transformer)| Swin Transform - Semantic Segmentation, only support Swin-T. The Pytorch implementation is [microsoft/Swin-Transformer](https://github.com/microsoft/Swin-Transformer.git) | +|[real-esrgan](./real-esrgan)| Real-ESRGAN. The Pytorch implementation is [real-esrgan](https://github.com/xinntao/Real-ESRGAN) | ## Model Zoo