Name		Name	Last commit message	Last commit date
parent directory ..
CMakeLists.txt		CMakeLists.txt
InferenceEngine.cpp		InferenceEngine.cpp
InferenceEngine.h		InferenceEngine.h
README.md		README.md
gen_wts.py		gen_wts.py
holder.h		holder.h
ibnnet.cpp		ibnnet.cpp
ibnnet.h		ibnnet.h
layers.cpp		layers.cpp
layers.h		layers.h
logging.h		logging.h
main.cpp		main.cpp
utils.cpp		utils.cpp
utils.h		utils.h

README.md

IBN-Net

An implementation of IBN-Net, proposed in "Two at Once: Enhancing Learning and Generalization Capacities via IBN-Net", ECCV2018 by Xingang Pan, Ping Luo, Jianping Shi, Xiaoou Tang.

For the Pytorch implementation, you can refer to 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.
1. cmake and make
```
mkdir build
cd build
cmake ..
make
```
1. 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
```