This project does not contain any training dataset. The model trained model is based on
DeepFashion. If you want to fine-tune
the model, please configure train_dataset_config.json
with the proper dataset path in value field.
Then execute
python train_for_shopee.py
The file test_for_shopee.py
is written for shopee data. Please excute
python test_for_shopee.py -p <path-to-your-test-image-folder>
The final model is over 100mb, so github does not allowed to push, so the model is put in my google drive
- You can download it in the url and put it in
model_output
folder then execute the above python script - You can either directly run the script and then
chrome
will pop out which will need your authentication for further downloading.
Slides are prepared for the explanation of my solution. It is int the report
folder. For more details, you may refer the slides instead.