├── FITE.py ├── LICENSE ├── README.md ├── assert └── gift.docx ├── captions ├── twitter2015_images.json └── twitter2017_images.json ├── face_descriptions ├── twitter15_face_description_clip16.json └── twitter17_face_description_clip16.json ├── generate_face_descriptions ├── face_attributes │ ├── tw15_dev_face_attributes.json │ ├── tw15_test_face_attributes.json │ ├── tw15_train_face_attributes.json │ ├── tw17_dev_face_attributes.json │ ├── tw17_test_face_attributes.json │ └── tw17_train_face_attributes.json └── multi_face_aspect_align.py ├── requirements.txt └── run.sh /FITE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/FITE.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/README.md -------------------------------------------------------------------------------- /assert/gift.docx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/assert/gift.docx -------------------------------------------------------------------------------- /captions/twitter2015_images.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/captions/twitter2015_images.json -------------------------------------------------------------------------------- /captions/twitter2017_images.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/captions/twitter2017_images.json -------------------------------------------------------------------------------- /face_descriptions/twitter15_face_description_clip16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/face_descriptions/twitter15_face_description_clip16.json -------------------------------------------------------------------------------- /face_descriptions/twitter17_face_description_clip16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/face_descriptions/twitter17_face_description_clip16.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw15_dev_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw15_dev_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw15_test_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw15_test_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw15_train_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw15_train_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw17_dev_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw17_dev_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw17_test_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw17_test_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/face_attributes/tw17_train_face_attributes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/face_attributes/tw17_train_face_attributes.json -------------------------------------------------------------------------------- /generate_face_descriptions/multi_face_aspect_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhit98/FITE/HEAD/generate_face_descriptions/multi_face_aspect_align.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==4.17.0 2 | torch==1.10.2 3 | numpy 4 | pandas==1.3.5 5 | sklearn 6 | -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 nohup python FITE.py >./tw17_bert_base_output_5e5.log 2>&1 & --------------------------------------------------------------------------------