├── README.md ├── config ├── config_seen_GREAT.yaml ├── config_unseen_aff_GREAT.yaml └── config_unseen_obj_GREAT.yaml ├── data_utils └── dataset_PIAD_GREAT.py ├── evalization.py ├── evalization.sh ├── image_text ├── seen │ ├── Img_test_text.txt │ ├── Img_train_text.txt │ └── Img_val_text.txt ├── unseen_aff │ ├── Img_test_text.txt │ └── Img_train_text.txt └── unseen_obj │ ├── Img_test_text.txt │ └── Img_train_text.txt ├── images ├── dataset.png └── method.png ├── model ├── GREAT.py ├── MHACoT.py └── pointnet2_utils.py ├── requirements.txt ├── train.py ├── train.sh └── utils ├── eval.py └── loss.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/README.md -------------------------------------------------------------------------------- /config/config_seen_GREAT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/config/config_seen_GREAT.yaml -------------------------------------------------------------------------------- /config/config_unseen_aff_GREAT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/config/config_unseen_aff_GREAT.yaml -------------------------------------------------------------------------------- /config/config_unseen_obj_GREAT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/config/config_unseen_obj_GREAT.yaml -------------------------------------------------------------------------------- /data_utils/dataset_PIAD_GREAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/data_utils/dataset_PIAD_GREAT.py -------------------------------------------------------------------------------- /evalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/evalization.py -------------------------------------------------------------------------------- /evalization.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=1 torchrun --nproc_per_node=1 evalization.py -------------------------------------------------------------------------------- /image_text/seen/Img_test_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/seen/Img_test_text.txt -------------------------------------------------------------------------------- /image_text/seen/Img_train_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/seen/Img_train_text.txt -------------------------------------------------------------------------------- /image_text/seen/Img_val_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/seen/Img_val_text.txt -------------------------------------------------------------------------------- /image_text/unseen_aff/Img_test_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/unseen_aff/Img_test_text.txt -------------------------------------------------------------------------------- /image_text/unseen_aff/Img_train_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/unseen_aff/Img_train_text.txt -------------------------------------------------------------------------------- /image_text/unseen_obj/Img_test_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/unseen_obj/Img_test_text.txt -------------------------------------------------------------------------------- /image_text/unseen_obj/Img_train_text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/image_text/unseen_obj/Img_train_text.txt -------------------------------------------------------------------------------- /images/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/images/dataset.png -------------------------------------------------------------------------------- /images/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/images/method.png -------------------------------------------------------------------------------- /model/GREAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/model/GREAT.py -------------------------------------------------------------------------------- /model/MHACoT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/model/MHACoT.py -------------------------------------------------------------------------------- /model/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/model/pointnet2_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0,1 torchrun --nproc_per_node=2 train.py -------------------------------------------------------------------------------- /utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/utils/eval.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yawen-shao/GREAT_code/HEAD/utils/loss.py --------------------------------------------------------------------------------