SENet-PyTorch

这是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