这是SENet的PyTorch实施(在ImageNet数据集上进行训练)
论文:
用法
准备数据
该代码以ImageNet数据集为例。 您可以下载ImageNet数据集,并将其放入如下位置。 由于内存限制,我仅提供ILSVRC2012_dev_kit_t12 ,换句话说,您需要下载ILSVRC2012_img_train和ILSVRC2012_img_val 。
├── train.py # train script
├── se_resnet.py # network of se_resnet
├── se_resnext.py # network of se_resnext
├── read_ImageNetData.py # ImageNet dataset read script
├── ImageData # train and validation data
├── IL
1