├── .gitignore ├── README.md ├── dataloaders ├── __init__.py ├── cityscapes_dataloader.py ├── nyu_v2_dataloader.py └── taskonomy_dataloader.py ├── envs ├── __init__.py ├── base_env.py └── blockdrop_env.py ├── figures └── model.jpg ├── models ├── __init__.py ├── base.py ├── deeplab_resnet.py └── util.py ├── nyu_v2_samples ├── depth │ ├── 0061.npy │ ├── 0295.npy │ ├── 0774.npy │ └── 1314.npy ├── img │ ├── 0061.png │ ├── 0295.png │ ├── 0774.png │ └── 1314.png ├── normal_mask │ ├── 0061.png │ ├── 0295.png │ ├── 0774.png │ └── 1314.png └── seg │ ├── 0061.png │ ├── 0295.png │ ├── 0774.png │ └── 1314.png ├── re-train.py ├── test.py ├── test_sample.py ├── train.py ├── utils ├── __init__.py ├── flops_benchmark.py └── util.py └── yamls ├── __init__.py └── adashare ├── cityscapes_2task.yml ├── cityscapes_2task_test.yml ├── nyu_v2_2task.yml ├── nyu_v2_2task_test.yml ├── nyu_v2_3task.yml ├── nyu_v2_3task_test.yml └── taskonomy_5task.yml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/README.md -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataloaders/cityscapes_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/dataloaders/cityscapes_dataloader.py -------------------------------------------------------------------------------- /dataloaders/nyu_v2_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/dataloaders/nyu_v2_dataloader.py -------------------------------------------------------------------------------- /dataloaders/taskonomy_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/dataloaders/taskonomy_dataloader.py -------------------------------------------------------------------------------- /envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /envs/base_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/envs/base_env.py -------------------------------------------------------------------------------- /envs/blockdrop_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/envs/blockdrop_env.py -------------------------------------------------------------------------------- /figures/model.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/figures/model.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/models/base.py -------------------------------------------------------------------------------- /models/deeplab_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/models/deeplab_resnet.py -------------------------------------------------------------------------------- /models/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/models/util.py -------------------------------------------------------------------------------- /nyu_v2_samples/depth/0061.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/depth/0061.npy -------------------------------------------------------------------------------- /nyu_v2_samples/depth/0295.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/depth/0295.npy -------------------------------------------------------------------------------- /nyu_v2_samples/depth/0774.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/depth/0774.npy -------------------------------------------------------------------------------- /nyu_v2_samples/depth/1314.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/depth/1314.npy -------------------------------------------------------------------------------- /nyu_v2_samples/img/0061.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/img/0061.png -------------------------------------------------------------------------------- /nyu_v2_samples/img/0295.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/img/0295.png -------------------------------------------------------------------------------- /nyu_v2_samples/img/0774.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/img/0774.png -------------------------------------------------------------------------------- /nyu_v2_samples/img/1314.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/img/1314.png -------------------------------------------------------------------------------- /nyu_v2_samples/normal_mask/0061.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/normal_mask/0061.png -------------------------------------------------------------------------------- /nyu_v2_samples/normal_mask/0295.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/normal_mask/0295.png -------------------------------------------------------------------------------- /nyu_v2_samples/normal_mask/0774.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/normal_mask/0774.png -------------------------------------------------------------------------------- /nyu_v2_samples/normal_mask/1314.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/normal_mask/1314.png -------------------------------------------------------------------------------- /nyu_v2_samples/seg/0061.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/seg/0061.png -------------------------------------------------------------------------------- /nyu_v2_samples/seg/0295.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/seg/0295.png -------------------------------------------------------------------------------- /nyu_v2_samples/seg/0774.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/seg/0774.png -------------------------------------------------------------------------------- /nyu_v2_samples/seg/1314.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/nyu_v2_samples/seg/1314.png -------------------------------------------------------------------------------- /re-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/re-train.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/test.py -------------------------------------------------------------------------------- /test_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/test_sample.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/flops_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/utils/flops_benchmark.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/utils/util.py -------------------------------------------------------------------------------- /yamls/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yamls/adashare/cityscapes_2task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/cityscapes_2task.yml -------------------------------------------------------------------------------- /yamls/adashare/cityscapes_2task_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/cityscapes_2task_test.yml -------------------------------------------------------------------------------- /yamls/adashare/nyu_v2_2task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/nyu_v2_2task.yml -------------------------------------------------------------------------------- /yamls/adashare/nyu_v2_2task_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/nyu_v2_2task_test.yml -------------------------------------------------------------------------------- /yamls/adashare/nyu_v2_3task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/nyu_v2_3task.yml -------------------------------------------------------------------------------- /yamls/adashare/nyu_v2_3task_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/nyu_v2_3task_test.yml -------------------------------------------------------------------------------- /yamls/adashare/taskonomy_5task.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunxm2357/AdaShare/HEAD/yamls/adashare/taskonomy_5task.yml --------------------------------------------------------------------------------