├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── equation.svg ├── include ├── checks.h ├── cuda_utils.cuh ├── dispatch.h ├── inplace_abn.h └── utils.h ├── inplace_abn.png ├── inplace_abn ├── __init__.py ├── _backend.pyi ├── abn.py ├── functions.py └── group.py ├── licenses.csv ├── requirements.txt ├── scripts ├── dataset │ ├── __init__.py │ ├── dataset.py │ ├── sampler.py │ └── transform.py ├── experiments │ ├── densenet264_ipabn_lr_256.json │ ├── resnet101_ipabn-sync_lr_512.json │ ├── resnet34_ipabn-sync_lr_512.json │ ├── resnet50_ipabn-sync_lr_512.json │ ├── resnext101_ipabn-sync_lr_256.json │ ├── resnext101_ipabn_lr_512.json │ ├── resnext101_stdbn_lr_256.json │ ├── resnext152_ipabn_lr_256.json │ └── wider_resnet38_ipabn_lr_256.json ├── imagenet │ ├── __init__.py │ ├── config.py │ ├── transforms.py │ └── utils.py ├── models │ ├── __init__.py │ ├── densenet.py │ ├── resnet.py │ ├── resnext.py │ ├── util.py │ └── wider_resnet.py ├── modules │ ├── __init__.py │ ├── deeplab.py │ ├── dense.py │ ├── misc.py │ └── residual.py ├── requirements.txt ├── test_imagenet.py ├── test_vistas.py ├── test_vistas_single_gpu.py └── train_imagenet.py ├── setup.cfg ├── setup.py └── src ├── inplace_abn.cpp ├── inplace_abn_cpu.cpp ├── inplace_abn_cuda.cu ├── inplace_abn_kernels.cuh └── utils.cpp /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/README.md -------------------------------------------------------------------------------- /equation.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/equation.svg -------------------------------------------------------------------------------- /include/checks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/include/checks.h -------------------------------------------------------------------------------- /include/cuda_utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/include/cuda_utils.cuh -------------------------------------------------------------------------------- /include/dispatch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/include/dispatch.h -------------------------------------------------------------------------------- /include/inplace_abn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/include/inplace_abn.h -------------------------------------------------------------------------------- /include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/include/utils.h -------------------------------------------------------------------------------- /inplace_abn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn.png -------------------------------------------------------------------------------- /inplace_abn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn/__init__.py -------------------------------------------------------------------------------- /inplace_abn/_backend.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn/_backend.pyi -------------------------------------------------------------------------------- /inplace_abn/abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn/abn.py -------------------------------------------------------------------------------- /inplace_abn/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn/functions.py -------------------------------------------------------------------------------- /inplace_abn/group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/inplace_abn/group.py -------------------------------------------------------------------------------- /licenses.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/licenses.csv -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/dataset/dataset.py -------------------------------------------------------------------------------- /scripts/dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/dataset/sampler.py -------------------------------------------------------------------------------- /scripts/dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/dataset/transform.py -------------------------------------------------------------------------------- /scripts/experiments/densenet264_ipabn_lr_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/densenet264_ipabn_lr_256.json -------------------------------------------------------------------------------- /scripts/experiments/resnet101_ipabn-sync_lr_512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnet101_ipabn-sync_lr_512.json -------------------------------------------------------------------------------- /scripts/experiments/resnet34_ipabn-sync_lr_512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnet34_ipabn-sync_lr_512.json -------------------------------------------------------------------------------- /scripts/experiments/resnet50_ipabn-sync_lr_512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnet50_ipabn-sync_lr_512.json -------------------------------------------------------------------------------- /scripts/experiments/resnext101_ipabn-sync_lr_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnext101_ipabn-sync_lr_256.json -------------------------------------------------------------------------------- /scripts/experiments/resnext101_ipabn_lr_512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnext101_ipabn_lr_512.json -------------------------------------------------------------------------------- /scripts/experiments/resnext101_stdbn_lr_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnext101_stdbn_lr_256.json -------------------------------------------------------------------------------- /scripts/experiments/resnext152_ipabn_lr_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/resnext152_ipabn_lr_256.json -------------------------------------------------------------------------------- /scripts/experiments/wider_resnet38_ipabn_lr_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/experiments/wider_resnet38_ipabn_lr_256.json -------------------------------------------------------------------------------- /scripts/imagenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/imagenet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/imagenet/config.py -------------------------------------------------------------------------------- /scripts/imagenet/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/imagenet/transforms.py -------------------------------------------------------------------------------- /scripts/imagenet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/imagenet/utils.py -------------------------------------------------------------------------------- /scripts/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/__init__.py -------------------------------------------------------------------------------- /scripts/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/densenet.py -------------------------------------------------------------------------------- /scripts/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/resnet.py -------------------------------------------------------------------------------- /scripts/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/resnext.py -------------------------------------------------------------------------------- /scripts/models/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/util.py -------------------------------------------------------------------------------- /scripts/models/wider_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/models/wider_resnet.py -------------------------------------------------------------------------------- /scripts/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/modules/__init__.py -------------------------------------------------------------------------------- /scripts/modules/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/modules/deeplab.py -------------------------------------------------------------------------------- /scripts/modules/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/modules/dense.py -------------------------------------------------------------------------------- /scripts/modules/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/modules/misc.py -------------------------------------------------------------------------------- /scripts/modules/residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/modules/residual.py -------------------------------------------------------------------------------- /scripts/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/requirements.txt -------------------------------------------------------------------------------- /scripts/test_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/test_imagenet.py -------------------------------------------------------------------------------- /scripts/test_vistas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/test_vistas.py -------------------------------------------------------------------------------- /scripts/test_vistas_single_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/test_vistas_single_gpu.py -------------------------------------------------------------------------------- /scripts/train_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/scripts/train_imagenet.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [metadata] 2 | license_files = LICENSE -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/setup.py -------------------------------------------------------------------------------- /src/inplace_abn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/src/inplace_abn.cpp -------------------------------------------------------------------------------- /src/inplace_abn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/src/inplace_abn_cpu.cpp -------------------------------------------------------------------------------- /src/inplace_abn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/src/inplace_abn_cuda.cu -------------------------------------------------------------------------------- /src/inplace_abn_kernels.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/src/inplace_abn_kernels.cuh -------------------------------------------------------------------------------- /src/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mapillary/inplace_abn/HEAD/src/utils.cpp --------------------------------------------------------------------------------