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

  Copy

  python gen_wts.py a
  

  a file 'resnet50-ibna.wts' will be generated.

  // for ibn-b

  Copy

  python gen_wts.py b
  

  a file 'resnet50-ibnb.wts' will be generated.

• 2. cmake and make
  Copy

  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

  Copy

  ./ibnnet -s  // serialize model to plan file
  ./ibnnet -d  // deserialize plan file and run inference