owj0421 / deepfashion Goto Github PK

Deep Fashion is a Easy-to-use, Modular and Extendible package of deep-learning based fashion recommendation models with PyTorch.

Behind the fact that none of the numerous papers released since 2018 have been implemented, we implement and distribute the model ourselves. We aimed to implement the paper as much as possible, but since it is a personal project, there may be some other aspects. Therefore, if there is a better way, please contribute.

What is included

• Data proprocessor that can easily configure set of outfits as Dataset
• Fashion compatibility models
• Metric learning loss that can be applied immediately to Batch configured with outfit-wise dataset

Notes

• Implementation is based on the above papers, but there may be differents.
• In the test, for fairness, the embedding size was fixed at 32, and only images were used.
• Only the models studied for the purpose of retrieval were developed, so the prediction-based models(SCE-Net, Outfit-Transformer etc) were not implemented.

This project recommends Python 3.7 or higher.

python -m pip install -r requirements.txt

1. Download the Polyvore dataset from here.

2. $MODEL is same as above mentioned sheet.

   !python train.py --model $MODEL --embedding_dim $NUM --dataset_type outfit --train_batch 64 --valid_batch 64 --fitb_batch 32 --n_epochs 5 --save_dir $DIR --data_dir $DIR --num_workers 4 --scheduler_step_size 500 --learning_rate 5e-5
   

Preparing for demos...

• This is NON-OFFICIAL implementation.
• The part that uses the HGLMM Fisher vector is replaced by SBERT Embedding. If you want to use Fisher vector, you can change txt_type to 'hglmm'. but it requires to revise model codes.