46 lines
998 B
Markdown
46 lines
998 B
Markdown
# IBN-Net
|
|
|
|
An implementation of IBN-Net, proposed in ["Two at Once: Enhancing Learning and Generalization Capacities via IBN-Net"](https://arxiv.org/abs/1807.09441), ECCV2018 by Xingang Pan, Ping Luo, Jianping Shi, Xiaoou Tang.
|
|
|
|
For the Pytorch implementation, you can refer to [IBN-Net](https://github.com/XingangPan/IBN-Net)
|
|
|
|
## Features
|
|
- InstanceNorm2d
|
|
- bottleneck_ibn
|
|
- Resnet50-IBNA
|
|
- Resnet50-IBNB
|
|
- Multi-thread inference
|
|
|
|
## How to Run
|
|
|
|
* 1. generate .wts
|
|
|
|
// for ibn-a
|
|
```
|
|
python gen_wts.py a
|
|
```
|
|
a file 'resnet50-ibna.wts' will be generated.
|
|
|
|
// for ibn-b
|
|
```
|
|
python gen_wts.py b
|
|
```
|
|
a file 'resnet50-ibnb.wts' will be generated.
|
|
* 2. cmake and make
|
|
|
|
```
|
|
mkdir build
|
|
cd build
|
|
cmake ..
|
|
make
|
|
```
|
|
* 3. build engine and run classification
|
|
|
|
// put resnet50-ibna.wts/resnet50-ibnb.wts into tensorrtx/ibnnet
|
|
|
|
// go to tensorrtx/ibnnet
|
|
```
|
|
./ibnnet -s // serialize model to plan file
|
|
./ibnnet -d // deserialize plan file and run inference
|
|
```
|
|
|