├── .gitignore ├── README.md ├── __pycache__ └── utils.cpython-38.pyc ├── cmds ├── execute_hist │ └── cmds_add_data.sh └── shell_scrips │ ├── imagenet-100-add-data.sh │ ├── imagenet-100-inference.sh │ └── sampling.sh ├── data ├── LT_Dataset.py ├── augmentation.py ├── imagenet_extract.sh └── utils.py ├── exp_analyse.py ├── imgs └── pipeline.png ├── inference └── inference_save.py ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── resnet.cpython-36.pyc │ └── resnet.cpython-37.pyc ├── resnet.py └── utils.py ├── sampling.py ├── split ├── imagenet-100 │ ├── ImageNet_100_test.txt │ ├── ImageNet_100_train.txt │ ├── ImageNet_100_val.txt │ ├── imageNet_100_LT_train.txt │ └── imageNet_100_sub_balance_train_0.01.txt ├── imagenet-900 │ └── ImageNet_900_train.txt └── imagenet-places-mix │ └── imageNet_200_places_200_mix.txt ├── train_imagenet.py ├── train_simCLR.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | *.slurm 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /cmds/execute_hist/cmds_add_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/cmds/execute_hist/cmds_add_data.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/imagenet-100-add-data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/cmds/shell_scrips/imagenet-100-add-data.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/imagenet-100-inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/cmds/shell_scrips/imagenet-100-inference.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/sampling.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/cmds/shell_scrips/sampling.sh -------------------------------------------------------------------------------- /data/LT_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/data/LT_Dataset.py -------------------------------------------------------------------------------- /data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/data/augmentation.py -------------------------------------------------------------------------------- /data/imagenet_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/data/imagenet_extract.sh -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/data/utils.py -------------------------------------------------------------------------------- /exp_analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/exp_analyse.py -------------------------------------------------------------------------------- /imgs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/imgs/pipeline.png -------------------------------------------------------------------------------- /inference/inference_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/inference/inference_save.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/models/utils.py -------------------------------------------------------------------------------- /sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/sampling.py -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-100/ImageNet_100_test.txt -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-100/ImageNet_100_train.txt -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-100/ImageNet_100_val.txt -------------------------------------------------------------------------------- /split/imagenet-100/imageNet_100_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-100/imageNet_100_LT_train.txt -------------------------------------------------------------------------------- /split/imagenet-100/imageNet_100_sub_balance_train_0.01.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-100/imageNet_100_sub_balance_train_0.01.txt -------------------------------------------------------------------------------- /split/imagenet-900/ImageNet_900_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-900/ImageNet_900_train.txt -------------------------------------------------------------------------------- /split/imagenet-places-mix/imageNet_200_places_200_mix.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/split/imagenet-places-mix/imageNet_200_places_200_mix.txt -------------------------------------------------------------------------------- /train_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/train_imagenet.py -------------------------------------------------------------------------------- /train_simCLR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/train_simCLR.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/MAK/HEAD/utils.py --------------------------------------------------------------------------------