├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README.md ├── augmentations.py ├── cifar.py ├── cifar10-scripts ├── allconv │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── densenet │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── resnext │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh └── wideresnet │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── cifar100-scripts ├── allconv │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── densenet │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── resnext │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh └── wideresnet │ ├── run-cn.sh │ ├── run-cnsn-augmix.sh │ ├── run-cnsn-consist.sh │ ├── run-cnsn.sh │ └── run-sn.sh ├── imagenet-scripts ├── run-cn.sh ├── run-cnsn-augmix.sh ├── run-cnsn-consist.sh ├── run-cnsn.sh └── run-sn.sh ├── imagenet.py ├── models ├── cifar │ ├── allconv_cnsn.py │ ├── densenet_cnsn.py │ ├── resnext_cnsn.py │ └── wideresnet_cnsn.py ├── cnsn.py └── imagenet │ ├── resnet_cnsn.py │ └── resnet_ibn_cnsn.py ├── segmentation ├── Readme.md ├── config │ └── gtav │ │ ├── gtav_fcn50.yaml │ │ └── gtav_fcn50_cnsn.yaml ├── data │ ├── cityscapes │ │ ├── cityscapes_colors.txt │ │ └── cityscapes_names.txt │ └── gtav │ │ ├── gtav_colors.txt │ │ └── gtav_names.txt ├── model │ ├── cnsn.py │ ├── cnsn_resnet.py │ ├── fcn.py │ ├── psanet.py │ ├── pspnet.py │ ├── resnet.py │ └── utils.py ├── tool │ ├── train.py │ ├── train.sh │ ├── train_cnsn.py │ └── train_cnsn.sh └── util │ ├── config.py │ ├── dataset.py │ ├── transform.py │ └── util.py ├── utils.py └── vis └── cnsn-framework.png /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | segmentation/dataset/ 3 | *_bkp.py 4 | exp/ 5 | 6 | -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/README.md -------------------------------------------------------------------------------- /augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/augmentations.py -------------------------------------------------------------------------------- /cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar.py -------------------------------------------------------------------------------- /cifar10-scripts/allconv/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/allconv/run-cn.sh -------------------------------------------------------------------------------- /cifar10-scripts/allconv/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/allconv/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar10-scripts/allconv/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/allconv/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar10-scripts/allconv/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/allconv/run-cnsn.sh -------------------------------------------------------------------------------- /cifar10-scripts/allconv/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/allconv/run-sn.sh -------------------------------------------------------------------------------- /cifar10-scripts/densenet/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/densenet/run-cn.sh -------------------------------------------------------------------------------- /cifar10-scripts/densenet/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/densenet/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar10-scripts/densenet/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/densenet/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar10-scripts/densenet/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/densenet/run-cnsn.sh -------------------------------------------------------------------------------- /cifar10-scripts/densenet/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/densenet/run-sn.sh -------------------------------------------------------------------------------- /cifar10-scripts/resnext/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/resnext/run-cn.sh -------------------------------------------------------------------------------- /cifar10-scripts/resnext/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/resnext/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar10-scripts/resnext/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/resnext/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar10-scripts/resnext/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/resnext/run-cnsn.sh -------------------------------------------------------------------------------- /cifar10-scripts/resnext/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/resnext/run-sn.sh -------------------------------------------------------------------------------- /cifar10-scripts/wideresnet/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/wideresnet/run-cn.sh -------------------------------------------------------------------------------- /cifar10-scripts/wideresnet/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/wideresnet/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar10-scripts/wideresnet/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/wideresnet/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar10-scripts/wideresnet/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/wideresnet/run-cnsn.sh -------------------------------------------------------------------------------- /cifar10-scripts/wideresnet/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar10-scripts/wideresnet/run-sn.sh -------------------------------------------------------------------------------- /cifar100-scripts/allconv/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/allconv/run-cn.sh -------------------------------------------------------------------------------- /cifar100-scripts/allconv/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/allconv/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar100-scripts/allconv/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/allconv/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar100-scripts/allconv/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/allconv/run-cnsn.sh -------------------------------------------------------------------------------- /cifar100-scripts/allconv/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/allconv/run-sn.sh -------------------------------------------------------------------------------- /cifar100-scripts/densenet/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/densenet/run-cn.sh -------------------------------------------------------------------------------- /cifar100-scripts/densenet/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/densenet/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar100-scripts/densenet/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/densenet/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar100-scripts/densenet/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/densenet/run-cnsn.sh -------------------------------------------------------------------------------- /cifar100-scripts/densenet/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/densenet/run-sn.sh -------------------------------------------------------------------------------- /cifar100-scripts/resnext/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/resnext/run-cn.sh -------------------------------------------------------------------------------- /cifar100-scripts/resnext/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/resnext/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar100-scripts/resnext/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/resnext/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar100-scripts/resnext/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/resnext/run-cnsn.sh -------------------------------------------------------------------------------- /cifar100-scripts/resnext/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/resnext/run-sn.sh -------------------------------------------------------------------------------- /cifar100-scripts/wideresnet/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/wideresnet/run-cn.sh -------------------------------------------------------------------------------- /cifar100-scripts/wideresnet/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/wideresnet/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /cifar100-scripts/wideresnet/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/wideresnet/run-cnsn-consist.sh -------------------------------------------------------------------------------- /cifar100-scripts/wideresnet/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/wideresnet/run-cnsn.sh -------------------------------------------------------------------------------- /cifar100-scripts/wideresnet/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/cifar100-scripts/wideresnet/run-sn.sh -------------------------------------------------------------------------------- /imagenet-scripts/run-cn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet-scripts/run-cn.sh -------------------------------------------------------------------------------- /imagenet-scripts/run-cnsn-augmix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet-scripts/run-cnsn-augmix.sh -------------------------------------------------------------------------------- /imagenet-scripts/run-cnsn-consist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet-scripts/run-cnsn-consist.sh -------------------------------------------------------------------------------- /imagenet-scripts/run-cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet-scripts/run-cnsn.sh -------------------------------------------------------------------------------- /imagenet-scripts/run-sn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet-scripts/run-sn.sh -------------------------------------------------------------------------------- /imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/imagenet.py -------------------------------------------------------------------------------- /models/cifar/allconv_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/cifar/allconv_cnsn.py -------------------------------------------------------------------------------- /models/cifar/densenet_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/cifar/densenet_cnsn.py -------------------------------------------------------------------------------- /models/cifar/resnext_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/cifar/resnext_cnsn.py -------------------------------------------------------------------------------- /models/cifar/wideresnet_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/cifar/wideresnet_cnsn.py -------------------------------------------------------------------------------- /models/cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/cnsn.py -------------------------------------------------------------------------------- /models/imagenet/resnet_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/imagenet/resnet_cnsn.py -------------------------------------------------------------------------------- /models/imagenet/resnet_ibn_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/models/imagenet/resnet_ibn_cnsn.py -------------------------------------------------------------------------------- /segmentation/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/Readme.md -------------------------------------------------------------------------------- /segmentation/config/gtav/gtav_fcn50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/config/gtav/gtav_fcn50.yaml -------------------------------------------------------------------------------- /segmentation/config/gtav/gtav_fcn50_cnsn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/config/gtav/gtav_fcn50_cnsn.yaml -------------------------------------------------------------------------------- /segmentation/data/cityscapes/cityscapes_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/data/cityscapes/cityscapes_colors.txt -------------------------------------------------------------------------------- /segmentation/data/cityscapes/cityscapes_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/data/cityscapes/cityscapes_names.txt -------------------------------------------------------------------------------- /segmentation/data/gtav/gtav_colors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/data/gtav/gtav_colors.txt -------------------------------------------------------------------------------- /segmentation/data/gtav/gtav_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/data/gtav/gtav_names.txt -------------------------------------------------------------------------------- /segmentation/model/cnsn.py: -------------------------------------------------------------------------------- 1 | ../../models/cnsn.py -------------------------------------------------------------------------------- /segmentation/model/cnsn_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/cnsn_resnet.py -------------------------------------------------------------------------------- /segmentation/model/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/fcn.py -------------------------------------------------------------------------------- /segmentation/model/psanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/psanet.py -------------------------------------------------------------------------------- /segmentation/model/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/pspnet.py -------------------------------------------------------------------------------- /segmentation/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/resnet.py -------------------------------------------------------------------------------- /segmentation/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/model/utils.py -------------------------------------------------------------------------------- /segmentation/tool/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/tool/train.py -------------------------------------------------------------------------------- /segmentation/tool/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/tool/train.sh -------------------------------------------------------------------------------- /segmentation/tool/train_cnsn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/tool/train_cnsn.py -------------------------------------------------------------------------------- /segmentation/tool/train_cnsn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/tool/train_cnsn.sh -------------------------------------------------------------------------------- /segmentation/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/util/config.py -------------------------------------------------------------------------------- /segmentation/util/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/util/dataset.py -------------------------------------------------------------------------------- /segmentation/util/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/util/transform.py -------------------------------------------------------------------------------- /segmentation/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/segmentation/util/util.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/utils.py -------------------------------------------------------------------------------- /vis/cnsn-framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/crossnorm-selfnorm/HEAD/vis/cnsn-framework.png --------------------------------------------------------------------------------