├── .gitignore ├── LICENSE ├── README.md ├── ViRBRunner.py ├── assets └── images │ ├── all_results.png │ └── tasks.png ├── configs ├── experiment_lists │ ├── all.yaml │ └── swav.yaml └── virb_configs │ ├── all.yaml │ ├── caltech.yaml │ ├── exemplar_grid_search.yaml │ ├── imagenet_grid_search.yaml │ └── imagenetv2.yaml ├── datasets ├── CLEVRDistEncodbleDataset.py ├── CLEVRNumObjectsEncodbleDataset.py ├── COCODetectionDataset.py ├── Caltech101EncldableDataset.py ├── Cifar100EncodbleDataset.py ├── CityscapesSemanticSegmentationDataset.py ├── EgoHandsDataset.py ├── EncodableDataloader.py ├── EncodableDataset.py ├── EurosatEncodbleDataset.py ├── ImagenetEncodbleDataset.py ├── Imagenetv2EncodbleDataset.py ├── KITTIDataset.py ├── KineticsActionPrediction.py ├── NyuDepthEncodbleDataset.py ├── NyuWalkableEncodbleDataset.py ├── OmniDataset.py ├── PetsDetectionEncodbleDataset.py ├── PetsEncodbleDataset.py ├── SUN397EncodbleDataset.py ├── TaskonomyDepthEncodbleDataset.py ├── TaskonomyEdgesEncodbleDataset.py ├── TaskonomyInpaintingEncodbleDataset.py ├── ThorActionPredictionDataset.py ├── ThorDepthEncodbleDataset.py ├── ThorNumStepsEncodbleDataset.py ├── __init__.py ├── dtdEncodbleDataset.py └── nuScenesActionPredictionDataset.py ├── main.py ├── models ├── ClassificationHead.py ├── DeepLabClassificationHead.py ├── DeepLabHead.py ├── LSTMHead.py ├── MultiEmbeddingClassificationHead.py ├── PixelWisePredictionHead.py ├── ResNet50Encoder.py ├── SiamesePixelWisePredictionHead.py ├── ViRBModel.py └── __init__.py ├── requirements.txt ├── scripts ├── exemplar_to_virb.py ├── humantorch_to_torchvision.py ├── moco_to_virb.py ├── pirl_to_virb.py ├── simsiam_to_virb.py └── swav_to_virb.py ├── tasks ├── ViRBTask.py └── __init__.py └── utils ├── __init__.py ├── error_functions.py ├── loss_functions.py ├── progress_data_packets.py └── progress_iterator.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/README.md -------------------------------------------------------------------------------- /ViRBRunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/ViRBRunner.py -------------------------------------------------------------------------------- /assets/images/all_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/assets/images/all_results.png -------------------------------------------------------------------------------- /assets/images/tasks.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/assets/images/tasks.png -------------------------------------------------------------------------------- /configs/experiment_lists/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/experiment_lists/all.yaml -------------------------------------------------------------------------------- /configs/experiment_lists/swav.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/experiment_lists/swav.yaml -------------------------------------------------------------------------------- /configs/virb_configs/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/virb_configs/all.yaml -------------------------------------------------------------------------------- /configs/virb_configs/caltech.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/virb_configs/caltech.yaml -------------------------------------------------------------------------------- /configs/virb_configs/exemplar_grid_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/virb_configs/exemplar_grid_search.yaml -------------------------------------------------------------------------------- /configs/virb_configs/imagenet_grid_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/virb_configs/imagenet_grid_search.yaml -------------------------------------------------------------------------------- /configs/virb_configs/imagenetv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/configs/virb_configs/imagenetv2.yaml -------------------------------------------------------------------------------- /datasets/CLEVRDistEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/CLEVRDistEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/CLEVRNumObjectsEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/CLEVRNumObjectsEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/COCODetectionDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/COCODetectionDataset.py -------------------------------------------------------------------------------- /datasets/Caltech101EncldableDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/Caltech101EncldableDataset.py -------------------------------------------------------------------------------- /datasets/Cifar100EncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/Cifar100EncodbleDataset.py -------------------------------------------------------------------------------- /datasets/CityscapesSemanticSegmentationDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/CityscapesSemanticSegmentationDataset.py -------------------------------------------------------------------------------- /datasets/EgoHandsDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/EgoHandsDataset.py -------------------------------------------------------------------------------- /datasets/EncodableDataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/EncodableDataloader.py -------------------------------------------------------------------------------- /datasets/EncodableDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/EncodableDataset.py -------------------------------------------------------------------------------- /datasets/EurosatEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/EurosatEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/ImagenetEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/ImagenetEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/Imagenetv2EncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/Imagenetv2EncodbleDataset.py -------------------------------------------------------------------------------- /datasets/KITTIDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/KITTIDataset.py -------------------------------------------------------------------------------- /datasets/KineticsActionPrediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/KineticsActionPrediction.py -------------------------------------------------------------------------------- /datasets/NyuDepthEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/NyuDepthEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/NyuWalkableEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/NyuWalkableEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/OmniDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/OmniDataset.py -------------------------------------------------------------------------------- /datasets/PetsDetectionEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/PetsDetectionEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/PetsEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/PetsEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/SUN397EncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/SUN397EncodbleDataset.py -------------------------------------------------------------------------------- /datasets/TaskonomyDepthEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/TaskonomyDepthEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/TaskonomyEdgesEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/TaskonomyEdgesEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/TaskonomyInpaintingEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/TaskonomyInpaintingEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/ThorActionPredictionDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/ThorActionPredictionDataset.py -------------------------------------------------------------------------------- /datasets/ThorDepthEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/ThorDepthEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/ThorNumStepsEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/ThorNumStepsEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/dtdEncodbleDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/dtdEncodbleDataset.py -------------------------------------------------------------------------------- /datasets/nuScenesActionPredictionDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/datasets/nuScenesActionPredictionDataset.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/main.py -------------------------------------------------------------------------------- /models/ClassificationHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/ClassificationHead.py -------------------------------------------------------------------------------- /models/DeepLabClassificationHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/DeepLabClassificationHead.py -------------------------------------------------------------------------------- /models/DeepLabHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/DeepLabHead.py -------------------------------------------------------------------------------- /models/LSTMHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/LSTMHead.py -------------------------------------------------------------------------------- /models/MultiEmbeddingClassificationHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/MultiEmbeddingClassificationHead.py -------------------------------------------------------------------------------- /models/PixelWisePredictionHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/PixelWisePredictionHead.py -------------------------------------------------------------------------------- /models/ResNet50Encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/ResNet50Encoder.py -------------------------------------------------------------------------------- /models/SiamesePixelWisePredictionHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/SiamesePixelWisePredictionHead.py -------------------------------------------------------------------------------- /models/ViRBModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/models/ViRBModel.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/exemplar_to_virb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/exemplar_to_virb.py -------------------------------------------------------------------------------- /scripts/humantorch_to_torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/humantorch_to_torchvision.py -------------------------------------------------------------------------------- /scripts/moco_to_virb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/moco_to_virb.py -------------------------------------------------------------------------------- /scripts/pirl_to_virb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/pirl_to_virb.py -------------------------------------------------------------------------------- /scripts/simsiam_to_virb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/simsiam_to_virb.py -------------------------------------------------------------------------------- /scripts/swav_to_virb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/scripts/swav_to_virb.py -------------------------------------------------------------------------------- /tasks/ViRBTask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/tasks/ViRBTask.py -------------------------------------------------------------------------------- /tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/error_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/utils/error_functions.py -------------------------------------------------------------------------------- /utils/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/utils/loss_functions.py -------------------------------------------------------------------------------- /utils/progress_data_packets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/utils/progress_data_packets.py -------------------------------------------------------------------------------- /utils/progress_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/ViRB/HEAD/utils/progress_iterator.py --------------------------------------------------------------------------------