├── .gitignore ├── README.md ├── custom_datasets.py ├── imgs ├── framework.png └── performance.png ├── kmeans.py ├── loader.py ├── main.py ├── models ├── __init__.py ├── densenet.py ├── dla.py ├── dla_simple.py ├── dpn.py ├── efficientnet.py ├── googlenet.py ├── lenet.py ├── mobilenet.py ├── mobilenetv2.py ├── pnasnet.py ├── preact_resnet.py ├── regnet.py ├── resnet.py ├── resnext.py ├── senet.py ├── shufflenet.py ├── shufflenetv2.py └── vgg.py ├── requirements.txt ├── rotation.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | scripts/ 2 | __pycache__ 3 | test.yaml 4 | run.md 5 | temp -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/README.md -------------------------------------------------------------------------------- /custom_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/custom_datasets.py -------------------------------------------------------------------------------- /imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/imgs/framework.png -------------------------------------------------------------------------------- /imgs/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/imgs/performance.png -------------------------------------------------------------------------------- /kmeans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/kmeans.py -------------------------------------------------------------------------------- /loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/loader.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/dla.py -------------------------------------------------------------------------------- /models/dla_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/dla_simple.py -------------------------------------------------------------------------------- /models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/dpn.py -------------------------------------------------------------------------------- /models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/efficientnet.py -------------------------------------------------------------------------------- /models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/googlenet.py -------------------------------------------------------------------------------- /models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/lenet.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/pnasnet.py -------------------------------------------------------------------------------- /models/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/preact_resnet.py -------------------------------------------------------------------------------- /models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/regnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/resnext.py -------------------------------------------------------------------------------- /models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/senet.py -------------------------------------------------------------------------------- /models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/shufflenet.py -------------------------------------------------------------------------------- /models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/shufflenetv2.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/models/vgg.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/requirements.txt -------------------------------------------------------------------------------- /rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/rotation.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvlab-research/BAL/HEAD/utils.py --------------------------------------------------------------------------------