diff --git a/README.md b/README.md index f60df94..f8c949a 100644 --- a/README.md +++ b/README.md @@ -10,6 +10,7 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a ## News +- `31 Jan 2021`. RepVGG added by [upczww](https://github.com/upczww). - `29 Jan 2021`. U-Net added by [YuzhouPeng](https://github.com/YuzhouPeng). - `24 Jan 2021`. IBN-Net added by [TCHeish](https://github.com/TCHeish), PSENet optimized, YOLOv5 v4.0 INT8, etc. - `8 Jan 2021`. YOLOv5 s/m/l/x updated to v4.0. @@ -24,7 +25,6 @@ All the models are implemented in pytorch/mxnet/tensorflown first, and export a - `13 Sep 2020`. Add crnn, and got 1000fps on GTX1080. - `28 Aug 2020`. [BaofengZan](https://github.com/BaofengZan) added a tutorial for compiling and running tensorrtx on windows. - `16 Aug 2020`. [upczww](https://github.com/upczww) added a python wrapper for yolov5. -- `28 May 2020`. arcface LResNet50E-IR model from [deepinsight/insightface](https://github.com/deepinsight/insightface) implemented. We got 333fps on GTX1080. ## Tutorials @@ -80,6 +80,7 @@ Following models are implemented. |[psenet](./psenet)| PSENet Text Detection, tensorflow implementation from [liuheng92/tensorflow_PSENet](https://github.com/liuheng92/tensorflow_PSENet) | |[ibnnet](./ibnnet)| IBN-Net, pytorch implementation from [XingangPan/IBN-Net](https://github.com/XingangPan/IBN-Net), ECCV2018 | |[unet](./unet)| U-Net, pytorch implementation from [milesial/Pytorch-UNet](https://github.com/milesial/Pytorch-UNet) | +|[repvgg](./repvgg)| RepVGG, pytorch implementation from [DingXiaoH/RepVGG](https://github.com/DingXiaoH/RepVGG) | ## Model Zoo diff --git a/yolov3-spp/README.md b/yolov3-spp/README.md index b701517..1bd914f 100644 --- a/yolov3-spp/README.md +++ b/yolov3-spp/README.md @@ -20,7 +20,7 @@ The Pytorch implementation is [ultralytics/yolov3](https://github.com/ultralytic ``` git clone https://github.com/wang-xinyu/tensorrtx.git -git clone https://github.com/ultralytics/yolov3.git +git clone -b archive https://github.com/ultralytics/yolov3.git // download its weights 'yolov3-spp-ultralytics.pt' // copy gen_wts.py from tensorrtx/yolov3-spp/ to ultralytics/yolov3/ // go to ultralytics/yolov3/ diff --git a/yolov3-tiny/README.md b/yolov3-tiny/README.md index ab81118..502175d 100644 --- a/yolov3-tiny/README.md +++ b/yolov3-tiny/README.md @@ -7,7 +7,7 @@ The Pytorch implementation is [ultralytics/yolov3](https://github.com/ultralytic ``` 1. generate yolov3-tiny.wts from pytorch implementation with yolov3-tiny.cfg and yolov3-tiny.weights, or download .wts from model zoo -git clone https://github.com/ultralytics/yolov3.git +git clone -b archive https://github.com/ultralytics/yolov3.git // download its weights 'yolov3-tiny.pt' or 'yolov3-tiny.weights' // put tensorrtx/yolov3-tiny/gen_wts.py into ultralytics/yolov3 and run python gen_wts.py yolov3-tiny.weights diff --git a/yolov3/README.md b/yolov3/README.md index a891f02..ee4eafc 100644 --- a/yolov3/README.md +++ b/yolov3/README.md @@ -19,7 +19,7 @@ This branch is using tensorrt7 API, there is also a yolov3 implementation using ``` git clone https://github.com/wang-xinyu/tensorrtx.git -git clone https://github.com/ultralytics/yolov3.git +git clone -b archive https://github.com/ultralytics/yolov3.git // download its weights 'yolov3.pt' or 'yolov3.weights' cp {tensorrtx}/yolov3/gen_wts.py {ultralytics/yolov3/} cd {ultralytics/yolov3/} diff --git a/yolov4/README.md b/yolov4/README.md index 552233b..6ef32c5 100644 --- a/yolov4/README.md +++ b/yolov4/README.md @@ -18,7 +18,7 @@ The Pytorch implementation is from [ultralytics/yolov3](https://github.com/ultra ``` git clone https://github.com/wang-xinyu/tensorrtx.git -git clone https://github.com/ultralytics/yolov3.git +git clone -b archive https://github.com/ultralytics/yolov3.git // download yolov4.weights from https://github.com/AlexeyAB/darknet#pre-trained-models cp {tensorrtx}/yolov4/gen_wts.py {ultralytics/yolov3/} cd {ultralytics/yolov3/}