├── CE.png ├── README.md ├── devkit ├── __init__.py ├── core │ ├── __init__.py │ ├── dist_utils.py │ ├── lr_scheduler.py │ └── utils.py ├── dataset │ ├── .ipynb_checkpoints │ │ └── imagenet_dataset-checkpoint.py │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ └── imagenet_dataset.cpython-36.pyc │ └── imagenet_dataset.py └── ops │ ├── .ipynb_checkpoints │ └── CE_module-checkpoint.py │ ├── CE_module.py │ ├── __init__.py │ ├── switchable_norm.py │ ├── syncbn_layer.py │ └── syncsn_layer.py └── imagenet ├── __init__.py ├── configs ├── .ipynb_checkpoints │ └── config_resnetv1ce50_step_moving_average-checkpoint.yaml └── config_resnetv1ce50_step_moving_average.yaml ├── models ├── .ipynb_checkpoints │ └── resnet_v1_ce-checkpoint.py ├── __init__.py ├── resnet_v1_ce.py ├── resnet_v1_sn.py └── resnet_v2_sn.py ├── test.sh ├── train_imagenet.py └── train_val.sh /CE.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/CE.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/README.md -------------------------------------------------------------------------------- /devkit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /devkit/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/core/__init__.py -------------------------------------------------------------------------------- /devkit/core/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/core/dist_utils.py -------------------------------------------------------------------------------- /devkit/core/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/core/lr_scheduler.py -------------------------------------------------------------------------------- /devkit/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/core/utils.py -------------------------------------------------------------------------------- /devkit/dataset/.ipynb_checkpoints/imagenet_dataset-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/dataset/.ipynb_checkpoints/imagenet_dataset-checkpoint.py -------------------------------------------------------------------------------- /devkit/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /devkit/dataset/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/dataset/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /devkit/dataset/__pycache__/imagenet_dataset.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/dataset/__pycache__/imagenet_dataset.cpython-36.pyc -------------------------------------------------------------------------------- /devkit/dataset/imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/dataset/imagenet_dataset.py -------------------------------------------------------------------------------- /devkit/ops/.ipynb_checkpoints/CE_module-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/.ipynb_checkpoints/CE_module-checkpoint.py -------------------------------------------------------------------------------- /devkit/ops/CE_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/CE_module.py -------------------------------------------------------------------------------- /devkit/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/__init__.py -------------------------------------------------------------------------------- /devkit/ops/switchable_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/switchable_norm.py -------------------------------------------------------------------------------- /devkit/ops/syncbn_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/syncbn_layer.py -------------------------------------------------------------------------------- /devkit/ops/syncsn_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/devkit/ops/syncsn_layer.py -------------------------------------------------------------------------------- /imagenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imagenet/configs/.ipynb_checkpoints/config_resnetv1ce50_step_moving_average-checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/configs/.ipynb_checkpoints/config_resnetv1ce50_step_moving_average-checkpoint.yaml -------------------------------------------------------------------------------- /imagenet/configs/config_resnetv1ce50_step_moving_average.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/configs/config_resnetv1ce50_step_moving_average.yaml -------------------------------------------------------------------------------- /imagenet/models/.ipynb_checkpoints/resnet_v1_ce-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/models/.ipynb_checkpoints/resnet_v1_ce-checkpoint.py -------------------------------------------------------------------------------- /imagenet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/models/__init__.py -------------------------------------------------------------------------------- /imagenet/models/resnet_v1_ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/models/resnet_v1_ce.py -------------------------------------------------------------------------------- /imagenet/models/resnet_v1_sn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/models/resnet_v1_sn.py -------------------------------------------------------------------------------- /imagenet/models/resnet_v2_sn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/models/resnet_v2_sn.py -------------------------------------------------------------------------------- /imagenet/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/test.sh -------------------------------------------------------------------------------- /imagenet/train_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/train_imagenet.py -------------------------------------------------------------------------------- /imagenet/train_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/CENet/HEAD/imagenet/train_val.sh --------------------------------------------------------------------------------