LIP: Local Importance-based Pooling
PyTorch implementations of LIP (ICCV 2019).
This codebase is under construction and will contain:
- the implementation of LIP based on PyTorch primitives,
- ImageNet training and testing code,
- CUDA implementation of LIP.
[8.13] We uploaded the code of LIP-ResNets and their ImageNet pretrained models (LIP-ResNet-50 & 101).
[8.17] Fixed the missing
init_lr key and the possible in_place
mul_ operation problem (reported in PyTorch 1.1).
- Python 3.6
- PyTorch 1.0+
- tensorboard and tensorboardX
You can download ImageNet pretrained models here.
Please refer to imagenet/README.md.