├── .gitignore ├── LICENSE ├── README.md ├── backbones ├── __init__.py └── aognet │ ├── AOG.py │ ├── __init__.py │ ├── aognet_singlescale.py │ ├── config.py │ ├── operator_basic.py │ └── operator_singlescale.py ├── configs ├── aognet_cifar100_1M.yaml ├── aognet_imagenet_12M.yaml └── aognet_imagenet_40M.yaml ├── examples ├── kill_all_python.sh ├── test_fp16.sh └── train_fp16.sh ├── images ├── teaser-imagenet-dissection.png └── teaser.png ├── requirements.txt └── tools ├── __init__.py ├── get_cifar.py ├── main_fp16.py └── smoothing.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/README.md -------------------------------------------------------------------------------- /backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /backbones/aognet/AOG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/backbones/aognet/AOG.py -------------------------------------------------------------------------------- /backbones/aognet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /backbones/aognet/aognet_singlescale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/backbones/aognet/aognet_singlescale.py -------------------------------------------------------------------------------- /backbones/aognet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/backbones/aognet/config.py -------------------------------------------------------------------------------- /backbones/aognet/operator_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/backbones/aognet/operator_basic.py -------------------------------------------------------------------------------- /backbones/aognet/operator_singlescale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/backbones/aognet/operator_singlescale.py -------------------------------------------------------------------------------- /configs/aognet_cifar100_1M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/configs/aognet_cifar100_1M.yaml -------------------------------------------------------------------------------- /configs/aognet_imagenet_12M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/configs/aognet_imagenet_12M.yaml -------------------------------------------------------------------------------- /configs/aognet_imagenet_40M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/configs/aognet_imagenet_40M.yaml -------------------------------------------------------------------------------- /examples/kill_all_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/examples/kill_all_python.sh -------------------------------------------------------------------------------- /examples/test_fp16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/examples/test_fp16.sh -------------------------------------------------------------------------------- /examples/train_fp16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/examples/train_fp16.sh -------------------------------------------------------------------------------- /images/teaser-imagenet-dissection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/images/teaser-imagenet-dissection.png -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/images/teaser.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python 2 | thop 3 | yacs 4 | scipy 5 | -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/get_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/tools/get_cifar.py -------------------------------------------------------------------------------- /tools/main_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/tools/main_fp16.py -------------------------------------------------------------------------------- /tools/smoothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNets/HEAD/tools/smoothing.py --------------------------------------------------------------------------------