Skip to content

Files

Latest commit

4ae3e87 · Jan 28, 2021

History

History
This branch is 21 commits ahead of, 151 commits behind wang-xinyu/tensorrtx:master.

ibnnet

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jan 28, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021
Jan 19, 2021

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