PyTorch-SE-ResNet
SE-ResNet PyTorch版本
我将其用于Place365和Challenge AI场景分类。
实作网
SE-ResNet(50、101、152)
SE-ResNeXt(50、101、152)
其他框架可以很容易地实现来修改model/model.py 。
Caffe源代码在
引文
如果您在研究中使用挤压和激发网络,请引用以下文章:
@article{hu2017,
title={Squeeze-and-Excitation Networks},
author={Jie Hu and Li Shen and Gang Sun},
journal={arXiv preprint arXiv:1709.01507},
year={2017}
}
2021-08-31 11:17:05
9KB
Python
1