├── .gitignore ├── LICENSE ├── README.md ├── configs ├── cifar10.yaml ├── cifar100.yaml ├── iNaturalist2017.yaml └── iNaturalist2018.yaml ├── lib ├── backbone │ ├── __init__.py │ ├── resnet.py │ └── resnet_cifar.py ├── config │ ├── __init__.py │ └── default.py ├── core │ ├── __init__.py │ ├── combiner.py │ ├── evaluate.py │ └── function.py ├── data_transform │ ├── __init__.py │ └── transform_wrapper.py ├── dataset │ ├── __init__.py │ ├── baseset.py │ ├── iNaturalist.py │ └── imbalance_cifar.py ├── loss │ ├── __init__.py │ └── loss.py ├── modules │ ├── __init__.py │ ├── classifier_ops.py │ └── pooling_ops.py ├── net │ ├── __init__.py │ └── network.py └── utils │ ├── __init__.py │ ├── lr_scheduler.py │ ├── registry.py │ └── utils.py ├── main ├── _init_paths.py ├── train.py └── valid.py ├── requirements.txt └── tools └── convert_from_iNat.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/README.md -------------------------------------------------------------------------------- /configs/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/configs/cifar10.yaml -------------------------------------------------------------------------------- /configs/cifar100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/configs/cifar100.yaml -------------------------------------------------------------------------------- /configs/iNaturalist2017.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/configs/iNaturalist2017.yaml -------------------------------------------------------------------------------- /configs/iNaturalist2018.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/configs/iNaturalist2018.yaml -------------------------------------------------------------------------------- /lib/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/backbone/__init__.py -------------------------------------------------------------------------------- /lib/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/backbone/resnet.py -------------------------------------------------------------------------------- /lib/backbone/resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/backbone/resnet_cifar.py -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/config/__init__.py -------------------------------------------------------------------------------- /lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/config/default.py -------------------------------------------------------------------------------- /lib/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/core/combiner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/core/combiner.py -------------------------------------------------------------------------------- /lib/core/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/core/evaluate.py -------------------------------------------------------------------------------- /lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/core/function.py -------------------------------------------------------------------------------- /lib/data_transform/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/data_transform/transform_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/data_transform/transform_wrapper.py -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/dataset/__init__.py -------------------------------------------------------------------------------- /lib/dataset/baseset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/dataset/baseset.py -------------------------------------------------------------------------------- /lib/dataset/iNaturalist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/dataset/iNaturalist.py -------------------------------------------------------------------------------- /lib/dataset/imbalance_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/dataset/imbalance_cifar.py -------------------------------------------------------------------------------- /lib/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/loss/__init__.py -------------------------------------------------------------------------------- /lib/loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/loss/loss.py -------------------------------------------------------------------------------- /lib/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/modules/__init__.py -------------------------------------------------------------------------------- /lib/modules/classifier_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/modules/classifier_ops.py -------------------------------------------------------------------------------- /lib/modules/pooling_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/modules/pooling_ops.py -------------------------------------------------------------------------------- /lib/net/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/net/__init__.py -------------------------------------------------------------------------------- /lib/net/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/net/network.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/utils/lr_scheduler.py -------------------------------------------------------------------------------- /lib/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/utils/registry.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /main/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/main/_init_paths.py -------------------------------------------------------------------------------- /main/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/main/train.py -------------------------------------------------------------------------------- /main/valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/main/valid.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/convert_from_iNat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/BBN/HEAD/tools/convert_from_iNat.py --------------------------------------------------------------------------------