Skip to content

[CVPR(W) 2022] UIGR: Unified Interactive Garment Retrieval

License

Notifications You must be signed in to change notification settings

BrandonHanx/CompFashion

Repository files navigation

CompFashion

A code base for conducting image retrieval with text feedback experiments.

Available methods:

Data Preparation

  1. Download FashionIQ dataset according to this. Then move data into train_data/fashioniq/.

  2. Build vocabulary.

    python -m nltk.downloader 'punkt'
    python build_vocab.py
  3. (Optional) Generate GloVe embeddings.

    python -m spacy download en_vectors_web_lg
    python gen_glove.py

Running

python train_net.py --config-file configs/fashioniq/tirg_bigru_init.yaml --use-tensorboard

Results

coming soon...

Different Settings in Original Methods

  1. VAL uses smaller val split and GloVe.
  2. CoSMo constructs captions with both orders for concatenating.
  3. ...

About

[CVPR(W) 2022] UIGR: Unified Interactive Garment Retrieval

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages