├── LICENSE ├── README.md ├── dataloader ├── cirr.py ├── dataset.py ├── fashion200k_patch.py ├── fashioniq.py └── shoes.py ├── environment.yml ├── imgs └── approach.png ├── losses ├── loss.py └── others │ ├── ce_loss.py │ ├── cyclip_loss.py │ ├── generation_loss.py │ ├── rank_loss.py │ └── text_relevance_loss.py ├── models ├── clip_model.py ├── fusion_model.py ├── model.py └── others │ ├── AMC_Model.py │ ├── ARTEMIS_Model.py │ ├── CaLa_Model.py │ ├── Combiner_Model.py │ ├── CompoDiff_Model.py │ ├── DQU_CIR_Model.py │ ├── VTF_Model.py │ ├── attention_fusion_model.py │ └── modeling_clip.py ├── run ├── test │ ├── test_200k.py │ ├── test_cirr.py │ ├── test_fiq.py │ ├── test_shoes.py │ └── test_val.py ├── train │ ├── base_trainer.py │ ├── train_200k.py │ ├── train_cirr.py │ ├── train_fiq.py │ └── train_shoes.py └── valid │ ├── validate_cirr.py │ ├── validate_fiq.py │ └── validate_shoes.py └── utils ├── extract_fashioniq_patch.py ├── extract_patch_features_200k.py ├── utils.py └── visualize.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/README.md -------------------------------------------------------------------------------- /dataloader/cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/dataloader/cirr.py -------------------------------------------------------------------------------- /dataloader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/dataloader/dataset.py -------------------------------------------------------------------------------- /dataloader/fashion200k_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/dataloader/fashion200k_patch.py -------------------------------------------------------------------------------- /dataloader/fashioniq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/dataloader/fashioniq.py -------------------------------------------------------------------------------- /dataloader/shoes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/dataloader/shoes.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/environment.yml -------------------------------------------------------------------------------- /imgs/approach.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/imgs/approach.png -------------------------------------------------------------------------------- /losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/loss.py -------------------------------------------------------------------------------- /losses/others/ce_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/others/ce_loss.py -------------------------------------------------------------------------------- /losses/others/cyclip_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/others/cyclip_loss.py -------------------------------------------------------------------------------- /losses/others/generation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/others/generation_loss.py -------------------------------------------------------------------------------- /losses/others/rank_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/others/rank_loss.py -------------------------------------------------------------------------------- /losses/others/text_relevance_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/losses/others/text_relevance_loss.py -------------------------------------------------------------------------------- /models/clip_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/clip_model.py -------------------------------------------------------------------------------- /models/fusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/fusion_model.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/model.py -------------------------------------------------------------------------------- /models/others/AMC_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/AMC_Model.py -------------------------------------------------------------------------------- /models/others/ARTEMIS_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/ARTEMIS_Model.py -------------------------------------------------------------------------------- /models/others/CaLa_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/CaLa_Model.py -------------------------------------------------------------------------------- /models/others/Combiner_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/Combiner_Model.py -------------------------------------------------------------------------------- /models/others/CompoDiff_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/CompoDiff_Model.py -------------------------------------------------------------------------------- /models/others/DQU_CIR_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/DQU_CIR_Model.py -------------------------------------------------------------------------------- /models/others/VTF_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/VTF_Model.py -------------------------------------------------------------------------------- /models/others/attention_fusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/attention_fusion_model.py -------------------------------------------------------------------------------- /models/others/modeling_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/models/others/modeling_clip.py -------------------------------------------------------------------------------- /run/test/test_200k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/test/test_200k.py -------------------------------------------------------------------------------- /run/test/test_cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/test/test_cirr.py -------------------------------------------------------------------------------- /run/test/test_fiq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/test/test_fiq.py -------------------------------------------------------------------------------- /run/test/test_shoes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/test/test_shoes.py -------------------------------------------------------------------------------- /run/test/test_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/test/test_val.py -------------------------------------------------------------------------------- /run/train/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/train/base_trainer.py -------------------------------------------------------------------------------- /run/train/train_200k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/train/train_200k.py -------------------------------------------------------------------------------- /run/train/train_cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/train/train_cirr.py -------------------------------------------------------------------------------- /run/train/train_fiq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/train/train_fiq.py -------------------------------------------------------------------------------- /run/train/train_shoes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/train/train_shoes.py -------------------------------------------------------------------------------- /run/valid/validate_cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/valid/validate_cirr.py -------------------------------------------------------------------------------- /run/valid/validate_fiq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/valid/validate_fiq.py -------------------------------------------------------------------------------- /run/valid/validate_shoes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/run/valid/validate_shoes.py -------------------------------------------------------------------------------- /utils/extract_fashioniq_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/utils/extract_fashioniq_patch.py -------------------------------------------------------------------------------- /utils/extract_patch_features_200k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/utils/extract_patch_features_200k.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenAnno/FashionERN_AAAI2024/HEAD/utils/visualize.py --------------------------------------------------------------------------------