├── .gitignore ├── README.md ├── __init__.py ├── appendix ├── VOCdevkit │ ├── longtail2012 │ │ ├── class_freq.pkl │ │ ├── class_split.pkl │ │ └── img_id.txt │ └── terse_gt_2012.pkl └── coco │ ├── longtail2017 │ ├── class_freq.pkl │ ├── class_split.pkl │ └── img_id.pkl │ ├── longtail2017class_split.pkl │ ├── longtail2017img_id.pkl │ └── terse_gt_2017.pkl ├── assets ├── dataset.png ├── intro.png └── overall.png ├── configs ├── coco │ └── LT_resnet50_pfc_DB.py └── voc │ └── LT_resnet50_pfc_DB.py ├── mllt ├── __init__.py ├── apis │ ├── __init__.py │ ├── env.py │ └── train.py ├── core │ ├── __init__.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── class_names.py │ │ ├── display.py │ │ ├── eval_hooks.py │ │ ├── eval_tools.py │ │ └── mean_ap.py │ └── utils │ │ ├── __init__.py │ │ ├── dist_utils.py │ │ └── misc.py ├── datasets │ ├── Lvis.py │ ├── __init__.py │ ├── builder.py │ ├── coco.py │ ├── custom.py │ ├── dataset_wrappers.py │ ├── datasets_split.py │ ├── extra_aug.py │ ├── loader │ │ ├── __init__.py │ │ ├── build_loader.py │ │ └── sampler.py │ ├── miniImagenet.py │ ├── registry.py │ ├── transforms.py │ ├── utils.py │ ├── voc.py │ └── xml_style.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── backbone_collection.py │ │ ├── inception3.py │ │ ├── resnet.py │ │ └── resnext.py │ ├── builder.py │ ├── classifiers │ │ ├── __init__.py │ │ ├── base.py │ │ └── simple.py │ ├── heads │ │ ├── __init__.py │ │ ├── cls_head.py │ │ ├── cos_head.py │ │ ├── gcn_head.py │ │ └── weldon_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── cross_entropy_loss.py │ │ ├── focal_loss.py │ │ ├── resample_loss.py │ │ └── utils.py │ ├── necks │ │ ├── __init__.py │ │ ├── fpn.py │ │ ├── mlp.py │ │ └── pfc.py │ ├── plugins │ │ ├── __init__.py │ │ └── generalized_attention.py │ ├── registry.py │ └── utils │ │ ├── __init__.py │ │ ├── conv_module.py │ │ ├── conv_ws.py │ │ ├── norm.py │ │ ├── scale.py │ │ └── weight_init.py └── utils │ ├── __init__.py │ └── registry.py └── tools ├── __init__.py ├── create_longtail_dataset.py ├── dist_feature.sh ├── dist_test.sh ├── record.py ├── save_feat.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /appendix/VOCdevkit/longtail2012/class_freq.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/VOCdevkit/longtail2012/class_freq.pkl -------------------------------------------------------------------------------- /appendix/VOCdevkit/longtail2012/class_split.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/VOCdevkit/longtail2012/class_split.pkl -------------------------------------------------------------------------------- /appendix/VOCdevkit/longtail2012/img_id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/VOCdevkit/longtail2012/img_id.txt -------------------------------------------------------------------------------- /appendix/VOCdevkit/terse_gt_2012.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/VOCdevkit/terse_gt_2012.pkl -------------------------------------------------------------------------------- /appendix/coco/longtail2017/class_freq.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/longtail2017/class_freq.pkl -------------------------------------------------------------------------------- /appendix/coco/longtail2017/class_split.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/longtail2017/class_split.pkl -------------------------------------------------------------------------------- /appendix/coco/longtail2017/img_id.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/longtail2017/img_id.pkl -------------------------------------------------------------------------------- /appendix/coco/longtail2017class_split.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/longtail2017class_split.pkl -------------------------------------------------------------------------------- /appendix/coco/longtail2017img_id.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/longtail2017img_id.pkl -------------------------------------------------------------------------------- /appendix/coco/terse_gt_2017.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/appendix/coco/terse_gt_2017.pkl -------------------------------------------------------------------------------- /assets/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/assets/dataset.png -------------------------------------------------------------------------------- /assets/intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/assets/intro.png -------------------------------------------------------------------------------- /assets/overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/assets/overall.png -------------------------------------------------------------------------------- /configs/coco/LT_resnet50_pfc_DB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/configs/coco/LT_resnet50_pfc_DB.py -------------------------------------------------------------------------------- /configs/voc/LT_resnet50_pfc_DB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/configs/voc/LT_resnet50_pfc_DB.py -------------------------------------------------------------------------------- /mllt/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /mllt/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/apis/__init__.py -------------------------------------------------------------------------------- /mllt/apis/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/apis/env.py -------------------------------------------------------------------------------- /mllt/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/apis/train.py -------------------------------------------------------------------------------- /mllt/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/__init__.py -------------------------------------------------------------------------------- /mllt/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mllt/core/evaluation/class_names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/class_names.py -------------------------------------------------------------------------------- /mllt/core/evaluation/display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/display.py -------------------------------------------------------------------------------- /mllt/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mllt/core/evaluation/eval_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/eval_tools.py -------------------------------------------------------------------------------- /mllt/core/evaluation/mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/evaluation/mean_ap.py -------------------------------------------------------------------------------- /mllt/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/utils/__init__.py -------------------------------------------------------------------------------- /mllt/core/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/utils/dist_utils.py -------------------------------------------------------------------------------- /mllt/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/core/utils/misc.py -------------------------------------------------------------------------------- /mllt/datasets/Lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/Lvis.py -------------------------------------------------------------------------------- /mllt/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/__init__.py -------------------------------------------------------------------------------- /mllt/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/builder.py -------------------------------------------------------------------------------- /mllt/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/coco.py -------------------------------------------------------------------------------- /mllt/datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/custom.py -------------------------------------------------------------------------------- /mllt/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mllt/datasets/datasets_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/datasets_split.py -------------------------------------------------------------------------------- /mllt/datasets/extra_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/extra_aug.py -------------------------------------------------------------------------------- /mllt/datasets/loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/loader/__init__.py -------------------------------------------------------------------------------- /mllt/datasets/loader/build_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/loader/build_loader.py -------------------------------------------------------------------------------- /mllt/datasets/loader/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/loader/sampler.py -------------------------------------------------------------------------------- /mllt/datasets/miniImagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/miniImagenet.py -------------------------------------------------------------------------------- /mllt/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/registry.py -------------------------------------------------------------------------------- /mllt/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/transforms.py -------------------------------------------------------------------------------- /mllt/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/utils.py -------------------------------------------------------------------------------- /mllt/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/voc.py -------------------------------------------------------------------------------- /mllt/datasets/xml_style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/datasets/xml_style.py -------------------------------------------------------------------------------- /mllt/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/__init__.py -------------------------------------------------------------------------------- /mllt/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/backbones/__init__.py -------------------------------------------------------------------------------- /mllt/models/backbones/backbone_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/backbones/backbone_collection.py -------------------------------------------------------------------------------- /mllt/models/backbones/inception3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/backbones/inception3.py -------------------------------------------------------------------------------- /mllt/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/backbones/resnet.py -------------------------------------------------------------------------------- /mllt/models/backbones/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/backbones/resnext.py -------------------------------------------------------------------------------- /mllt/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/builder.py -------------------------------------------------------------------------------- /mllt/models/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/classifiers/__init__.py -------------------------------------------------------------------------------- /mllt/models/classifiers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/classifiers/base.py -------------------------------------------------------------------------------- /mllt/models/classifiers/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/classifiers/simple.py -------------------------------------------------------------------------------- /mllt/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/heads/__init__.py -------------------------------------------------------------------------------- /mllt/models/heads/cls_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/heads/cls_head.py -------------------------------------------------------------------------------- /mllt/models/heads/cos_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/heads/cos_head.py -------------------------------------------------------------------------------- /mllt/models/heads/gcn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/heads/gcn_head.py -------------------------------------------------------------------------------- /mllt/models/heads/weldon_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/heads/weldon_head.py -------------------------------------------------------------------------------- /mllt/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/__init__.py -------------------------------------------------------------------------------- /mllt/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/accuracy.py -------------------------------------------------------------------------------- /mllt/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /mllt/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/focal_loss.py -------------------------------------------------------------------------------- /mllt/models/losses/resample_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/resample_loss.py -------------------------------------------------------------------------------- /mllt/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/losses/utils.py -------------------------------------------------------------------------------- /mllt/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/necks/__init__.py -------------------------------------------------------------------------------- /mllt/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/necks/fpn.py -------------------------------------------------------------------------------- /mllt/models/necks/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/necks/mlp.py -------------------------------------------------------------------------------- /mllt/models/necks/pfc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/necks/pfc.py -------------------------------------------------------------------------------- /mllt/models/plugins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/plugins/__init__.py -------------------------------------------------------------------------------- /mllt/models/plugins/generalized_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/plugins/generalized_attention.py -------------------------------------------------------------------------------- /mllt/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/registry.py -------------------------------------------------------------------------------- /mllt/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/__init__.py -------------------------------------------------------------------------------- /mllt/models/utils/conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/conv_module.py -------------------------------------------------------------------------------- /mllt/models/utils/conv_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/conv_ws.py -------------------------------------------------------------------------------- /mllt/models/utils/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/norm.py -------------------------------------------------------------------------------- /mllt/models/utils/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/scale.py -------------------------------------------------------------------------------- /mllt/models/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/models/utils/weight_init.py -------------------------------------------------------------------------------- /mllt/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/utils/__init__.py -------------------------------------------------------------------------------- /mllt/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/mllt/utils/registry.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/create_longtail_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/create_longtail_dataset.py -------------------------------------------------------------------------------- /tools/dist_feature.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/dist_feature.sh -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/record.py -------------------------------------------------------------------------------- /tools/save_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/save_feat.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wutong16/DistributionBalancedLoss/HEAD/tools/train.py --------------------------------------------------------------------------------