├── .gitignore ├── ConcatplusSimilarityNetworkTrain.py ├── FullModelTest.py ├── README.md ├── SimilarityNetworkTrain.py ├── VR_Encoder ├── configs │ ├── data_config.yaml │ ├── model_config.yaml │ └── train_config.yaml ├── dataloader │ ├── vg150_dataset.py │ └── vrr_vg_dataset.py └── model │ ├── concat.py │ └── vtranse.py ├── VR_SimilarityNetwork ├── configs │ ├── data_config_test.yaml │ ├── data_config_train.yaml │ ├── model_config.yaml │ ├── test_config.yaml │ └── train_config.yaml ├── dataloader │ ├── VG150Dataset.py │ ├── VrRVGDatasetTest.py │ └── VrRVGDatasetTrain.py └── model │ ├── SimilarityNetworkConcat.py │ └── SimilarityNetworkVREncoder.py ├── data ├── vrrvg_predicates_test.json └── vrrvg_predicates_train.json ├── data_preparation ├── README.md └── vrc_extract_frcnn_feats.py ├── requirements.txt ├── train_vr_encoder.py └── utils ├── logger.py ├── sampling_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/.gitignore -------------------------------------------------------------------------------- /ConcatplusSimilarityNetworkTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/ConcatplusSimilarityNetworkTrain.py -------------------------------------------------------------------------------- /FullModelTest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/FullModelTest.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/README.md -------------------------------------------------------------------------------- /SimilarityNetworkTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/SimilarityNetworkTrain.py -------------------------------------------------------------------------------- /VR_Encoder/configs/data_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/configs/data_config.yaml -------------------------------------------------------------------------------- /VR_Encoder/configs/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/configs/model_config.yaml -------------------------------------------------------------------------------- /VR_Encoder/configs/train_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/configs/train_config.yaml -------------------------------------------------------------------------------- /VR_Encoder/dataloader/vg150_dataset.py: -------------------------------------------------------------------------------- 1 | # TODO: include vg150 dataset here -------------------------------------------------------------------------------- /VR_Encoder/dataloader/vrr_vg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/dataloader/vrr_vg_dataset.py -------------------------------------------------------------------------------- /VR_Encoder/model/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/model/concat.py -------------------------------------------------------------------------------- /VR_Encoder/model/vtranse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_Encoder/model/vtranse.py -------------------------------------------------------------------------------- /VR_SimilarityNetwork/configs/data_config_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/configs/data_config_test.yaml -------------------------------------------------------------------------------- /VR_SimilarityNetwork/configs/data_config_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/configs/data_config_train.yaml -------------------------------------------------------------------------------- /VR_SimilarityNetwork/configs/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/configs/model_config.yaml -------------------------------------------------------------------------------- /VR_SimilarityNetwork/configs/test_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/configs/test_config.yaml -------------------------------------------------------------------------------- /VR_SimilarityNetwork/configs/train_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/configs/train_config.yaml -------------------------------------------------------------------------------- /VR_SimilarityNetwork/dataloader/VG150Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/dataloader/VG150Dataset.py -------------------------------------------------------------------------------- /VR_SimilarityNetwork/dataloader/VrRVGDatasetTest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/dataloader/VrRVGDatasetTest.py -------------------------------------------------------------------------------- /VR_SimilarityNetwork/dataloader/VrRVGDatasetTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/dataloader/VrRVGDatasetTrain.py -------------------------------------------------------------------------------- /VR_SimilarityNetwork/model/SimilarityNetworkConcat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/model/SimilarityNetworkConcat.py -------------------------------------------------------------------------------- /VR_SimilarityNetwork/model/SimilarityNetworkVREncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/VR_SimilarityNetwork/model/SimilarityNetworkVREncoder.py -------------------------------------------------------------------------------- /data/vrrvg_predicates_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/data/vrrvg_predicates_test.json -------------------------------------------------------------------------------- /data/vrrvg_predicates_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/data/vrrvg_predicates_train.json -------------------------------------------------------------------------------- /data_preparation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/data_preparation/README.md -------------------------------------------------------------------------------- /data_preparation/vrc_extract_frcnn_feats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/data_preparation/vrc_extract_frcnn_feats.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_vr_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/train_vr_encoder.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/utils/sampling_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vl2g/VRC/HEAD/utils/utils.py --------------------------------------------------------------------------------