├── .gitignore ├── LICENSE ├── README.md ├── configs ├── aognet-12m-an-imagenet-200e-lbsmth-mixup.yaml ├── aognet-12m-an-imagenet.yaml ├── aognet-12m-imagenet-200e-lbsmth-mixup.yaml ├── aognet-12m-imagenet.yaml ├── aognet-40m-an-imagenet-200e-lbsmth-mixup.yaml ├── aognet-40m-an-imagenet.yaml ├── aognet-40m-imagenet-200e-lbsmth-mixup.yaml ├── aognet-40m-imagenet.yaml ├── mobilenetv2-an-imagenet.yaml ├── resnet-101-an-imagenet-200e-lbsmth-mixup.yaml ├── resnet-101-an-imagenet.yaml ├── resnet-101-imagenet-200e-lbsmth-mixup.yaml ├── resnet-101-imagenet.yaml ├── resnet-50-an-imagenet-200e-lbsmth-mixup.yaml ├── resnet-50-an-imagenet.yaml ├── resnet-50-imagenet-200e-lbsmth-mixup.yaml └── resnet-50-imagenet.yaml ├── images ├── an-comparison.png ├── teaser-imagenet-dissection.png └── teaser.png ├── models ├── __init__.py ├── aognet │ ├── AOG.py │ ├── __init__.py │ ├── aognet.py │ ├── operator_basic.py │ └── operator_singlescale.py ├── config.py ├── mobilenet.py └── resnet.py ├── requirements.txt ├── scripts ├── test_fp16.sh └── train_fp16.sh └── tools ├── __init__.py ├── main_fp16.py └── smoothing.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/README.md -------------------------------------------------------------------------------- /configs/aognet-12m-an-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-12m-an-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/aognet-12m-an-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-12m-an-imagenet.yaml -------------------------------------------------------------------------------- /configs/aognet-12m-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-12m-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/aognet-12m-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-12m-imagenet.yaml -------------------------------------------------------------------------------- /configs/aognet-40m-an-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-40m-an-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/aognet-40m-an-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-40m-an-imagenet.yaml -------------------------------------------------------------------------------- /configs/aognet-40m-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-40m-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/aognet-40m-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/aognet-40m-imagenet.yaml -------------------------------------------------------------------------------- /configs/mobilenetv2-an-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/mobilenetv2-an-imagenet.yaml -------------------------------------------------------------------------------- /configs/resnet-101-an-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-101-an-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/resnet-101-an-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-101-an-imagenet.yaml -------------------------------------------------------------------------------- /configs/resnet-101-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-101-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/resnet-101-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-101-imagenet.yaml -------------------------------------------------------------------------------- /configs/resnet-50-an-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-50-an-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/resnet-50-an-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-50-an-imagenet.yaml -------------------------------------------------------------------------------- /configs/resnet-50-imagenet-200e-lbsmth-mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-50-imagenet-200e-lbsmth-mixup.yaml -------------------------------------------------------------------------------- /configs/resnet-50-imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/configs/resnet-50-imagenet.yaml -------------------------------------------------------------------------------- /images/an-comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/images/an-comparison.png -------------------------------------------------------------------------------- /images/teaser-imagenet-dissection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/images/teaser-imagenet-dissection.png -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/images/teaser.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/aognet/AOG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/aognet/AOG.py -------------------------------------------------------------------------------- /models/aognet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/aognet/aognet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/aognet/aognet.py -------------------------------------------------------------------------------- /models/aognet/operator_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/aognet/operator_basic.py -------------------------------------------------------------------------------- /models/aognet/operator_singlescale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/aognet/operator_singlescale.py -------------------------------------------------------------------------------- /models/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/config.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/models/resnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python 2 | thop 3 | yacs 4 | scipy 5 | -------------------------------------------------------------------------------- /scripts/test_fp16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/scripts/test_fp16.sh -------------------------------------------------------------------------------- /scripts/train_fp16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/scripts/train_fp16.sh -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/main_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/tools/main_fp16.py -------------------------------------------------------------------------------- /tools/smoothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iVMCL/AOGNet-v2/HEAD/tools/smoothing.py --------------------------------------------------------------------------------