Awesome
NonLocalNetwork and Sequeeze-Excitation Network
Paper:
- Non-local Neural Networks (https://arxiv.org/abs/1711.07971)
- Squeeze-and-Excitation Networks (https://arxiv.org/abs/1709.01507)
Train nonlocal with cifar100
python train_nonlocal.py --cfg cfgs/nonlocal_cifar100.yaml
Train sequeeze-excitation with tiny-imagenet
python train_seresnet.py --cfg cfgs/seresnet_tinyimagenet.yaml
Reference
[1] SENet.mxnet