haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
wang-xinyu 2021-01-31 16:37:09 +08:00
parent 95bea3e7fa
commit 32d960a47c
5 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -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

2
yolov3-spp/README.md
View File

@ -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/

2
yolov3-tiny/README.md
View File

@ -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

2
yolov3/README.md
View File

@ -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/}

2
yolov4/README.md
View File

@ -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/}