├── LICENSE ├── README.md ├── dataset ├── ade20k.py ├── camvid.py ├── cityscapes.py ├── coco_stuff_164k.py ├── list │ ├── CamVid │ │ ├── camvid_test_list.txt │ │ ├── camvid_train_list.txt │ │ ├── camvid_trainval_list.txt │ │ └── camvid_val_list.txt │ ├── cityscapes │ │ ├── test.lst │ │ ├── train++.lst │ │ ├── train+.lst │ │ ├── train.lst │ │ ├── trainval.lst │ │ └── val.lst │ ├── coco_stuff_164k │ │ ├── coco_stuff_164k_train.txt │ │ └── coco_stuff_164k_val.txt │ └── voc │ │ ├── train_aug.txt │ │ └── val.txt └── voc.py ├── eval.py ├── eval_scripts └── eval.sh ├── losses ├── __init__.py ├── kd.py ├── ocmgd.py └── task.py ├── main.png ├── models ├── __init__.py ├── base_models │ ├── __init__.py │ ├── hrnet.py │ ├── mobilenetv2.py │ └── resnet.py ├── deeplabv3.py ├── deeplabv3_mobile.py ├── model_zoo.py ├── psp_mobile.py ├── pspnet.py ├── segbase.py └── segformer.py ├── test.py ├── train_baseline.py ├── train_kd.py ├── train_ocmgd.py ├── train_scripts ├── ade20k │ └── deeplabv3_res18.sh ├── camvid │ ├── deeplabv3_res18.sh │ └── pspnet_res18.sh ├── cityscape │ ├── deeplabv3_r101_deeplabv3_r18_ocmgd.sh │ └── deeplabv3_r101_mbv2_r18_v2_ocmgd.sh ├── coco_stuff │ └── deeplabv3_res18.sh └── voc │ ├── deeplabv3_res18.sh │ └── pspnet_res18.sh ├── tsne ├── save_embeddings.py ├── tsne.sh └── tsne_visual.py └── utils ├── __init__.py ├── distributed.py ├── flops.py ├── logger.py ├── sagan.py ├── score.py └── visualize.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/README.md -------------------------------------------------------------------------------- /dataset/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/ade20k.py -------------------------------------------------------------------------------- /dataset/camvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/camvid.py -------------------------------------------------------------------------------- /dataset/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/cityscapes.py -------------------------------------------------------------------------------- /dataset/coco_stuff_164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/coco_stuff_164k.py -------------------------------------------------------------------------------- /dataset/list/CamVid/camvid_test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/CamVid/camvid_test_list.txt -------------------------------------------------------------------------------- /dataset/list/CamVid/camvid_train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/CamVid/camvid_train_list.txt -------------------------------------------------------------------------------- /dataset/list/CamVid/camvid_trainval_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/CamVid/camvid_trainval_list.txt -------------------------------------------------------------------------------- /dataset/list/CamVid/camvid_val_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/CamVid/camvid_val_list.txt -------------------------------------------------------------------------------- /dataset/list/cityscapes/test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/test.lst -------------------------------------------------------------------------------- /dataset/list/cityscapes/train++.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/train++.lst -------------------------------------------------------------------------------- /dataset/list/cityscapes/train+.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/train+.lst -------------------------------------------------------------------------------- /dataset/list/cityscapes/train.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/train.lst -------------------------------------------------------------------------------- /dataset/list/cityscapes/trainval.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/trainval.lst -------------------------------------------------------------------------------- /dataset/list/cityscapes/val.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/cityscapes/val.lst -------------------------------------------------------------------------------- /dataset/list/coco_stuff_164k/coco_stuff_164k_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/coco_stuff_164k/coco_stuff_164k_train.txt -------------------------------------------------------------------------------- /dataset/list/coco_stuff_164k/coco_stuff_164k_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/coco_stuff_164k/coco_stuff_164k_val.txt -------------------------------------------------------------------------------- /dataset/list/voc/train_aug.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/voc/train_aug.txt -------------------------------------------------------------------------------- /dataset/list/voc/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/list/voc/val.txt -------------------------------------------------------------------------------- /dataset/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/dataset/voc.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/eval.py -------------------------------------------------------------------------------- /eval_scripts/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/eval_scripts/eval.sh -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/losses/kd.py -------------------------------------------------------------------------------- /losses/ocmgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/losses/ocmgd.py -------------------------------------------------------------------------------- /losses/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/losses/task.py -------------------------------------------------------------------------------- /main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/main.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | """Model Zoo""" 2 | -------------------------------------------------------------------------------- /models/base_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/base_models/hrnet.py -------------------------------------------------------------------------------- /models/base_models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/base_models/mobilenetv2.py -------------------------------------------------------------------------------- /models/base_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/base_models/resnet.py -------------------------------------------------------------------------------- /models/deeplabv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/deeplabv3.py -------------------------------------------------------------------------------- /models/deeplabv3_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/deeplabv3_mobile.py -------------------------------------------------------------------------------- /models/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/model_zoo.py -------------------------------------------------------------------------------- /models/psp_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/psp_mobile.py -------------------------------------------------------------------------------- /models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/pspnet.py -------------------------------------------------------------------------------- /models/segbase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/segbase.py -------------------------------------------------------------------------------- /models/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/models/segformer.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/test.py -------------------------------------------------------------------------------- /train_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_baseline.py -------------------------------------------------------------------------------- /train_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_kd.py -------------------------------------------------------------------------------- /train_ocmgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_ocmgd.py -------------------------------------------------------------------------------- /train_scripts/ade20k/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/ade20k/deeplabv3_res18.sh -------------------------------------------------------------------------------- /train_scripts/camvid/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/camvid/deeplabv3_res18.sh -------------------------------------------------------------------------------- /train_scripts/camvid/pspnet_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/camvid/pspnet_res18.sh -------------------------------------------------------------------------------- /train_scripts/cityscape/deeplabv3_r101_deeplabv3_r18_ocmgd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/cityscape/deeplabv3_r101_deeplabv3_r18_ocmgd.sh -------------------------------------------------------------------------------- /train_scripts/cityscape/deeplabv3_r101_mbv2_r18_v2_ocmgd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/cityscape/deeplabv3_r101_mbv2_r18_v2_ocmgd.sh -------------------------------------------------------------------------------- /train_scripts/coco_stuff/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/coco_stuff/deeplabv3_res18.sh -------------------------------------------------------------------------------- /train_scripts/voc/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/voc/deeplabv3_res18.sh -------------------------------------------------------------------------------- /train_scripts/voc/pspnet_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/train_scripts/voc/pspnet_res18.sh -------------------------------------------------------------------------------- /tsne/save_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/tsne/save_embeddings.py -------------------------------------------------------------------------------- /tsne/tsne.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/tsne/tsne.sh -------------------------------------------------------------------------------- /tsne/tsne_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/tsne/tsne_visual.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/flops.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/sagan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/sagan.py -------------------------------------------------------------------------------- /utils/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/score.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSVAI/Af-DCD/HEAD/utils/visualize.py --------------------------------------------------------------------------------