haotian/duan8

History

Ryo Mizote 4ae3e8743f Fix a bug(wrong CMakeLists syntax) (#377 )		2021-01-28 12:58:42 +08:00
..
CMakeLists.txt	Fix a bug(wrong CMakeLists syntax) (#377 )	2021-01-28 12:58:42 +08:00
gen_wts.py	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
holder.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
ibnnet.cpp	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
ibnnet.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
InferenceEngine.cpp	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
InferenceEngine.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
layers.cpp	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
layers.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
logging.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
main.cpp	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
README.md	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
utils.cpp	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00
utils.h	IBN-Net (#362 )	2021-01-19 22:06:25 +08:00

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