├── .gitignore ├── LICENSE ├── README.md ├── experiments ├── densenet121_sw │ ├── run.sh │ └── run_slurm.sh ├── resnet50 │ ├── run.sh │ └── run_slurm.sh └── resnet50_sw │ ├── run.sh │ └── run_slurm.sh ├── imagenet.py ├── models ├── __init__.py ├── backbones │ ├── __init__.py │ ├── densenet.py │ └── resnet.py ├── ops │ ├── __init__.py │ ├── switchwhiten.py │ └── sync_switchwhiten.py └── utils │ ├── __init__.py │ └── norm.py ├── requirements.txt └── utils ├── __init__.py ├── common_utils.py └── distributed_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/README.md -------------------------------------------------------------------------------- /experiments/densenet121_sw/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/densenet121_sw/run.sh -------------------------------------------------------------------------------- /experiments/densenet121_sw/run_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/densenet121_sw/run_slurm.sh -------------------------------------------------------------------------------- /experiments/resnet50/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/resnet50/run.sh -------------------------------------------------------------------------------- /experiments/resnet50/run_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/resnet50/run_slurm.sh -------------------------------------------------------------------------------- /experiments/resnet50_sw/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/resnet50_sw/run.sh -------------------------------------------------------------------------------- /experiments/resnet50_sw/run_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/experiments/resnet50_sw/run_slurm.sh -------------------------------------------------------------------------------- /imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/imagenet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .backbones import * -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/backbones/__init__.py -------------------------------------------------------------------------------- /models/backbones/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/backbones/densenet.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/ops/__init__.py: -------------------------------------------------------------------------------- 1 | from .switchwhiten import * -------------------------------------------------------------------------------- /models/ops/switchwhiten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/ops/switchwhiten.py -------------------------------------------------------------------------------- /models/ops/sync_switchwhiten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/ops/sync_switchwhiten.py -------------------------------------------------------------------------------- /models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .norm import build_norm_layer 2 | -------------------------------------------------------------------------------- /models/utils/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/models/utils/norm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torchvision 2 | tensorboardX 3 | -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/utils/common_utils.py -------------------------------------------------------------------------------- /utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingangPan/Switchable-Whitening/HEAD/utils/distributed_utils.py --------------------------------------------------------------------------------