search

open-mmlab / mmfashion

Open-source toolbox for visual fashion analysis based on PyTorch
https://open-mmlab.github.io/
Apache License 2.0
1.25k stars 284 forks source link

How to generate outfit-set in recommendation demo? #118

Open daothanhbinh20 opened 3 years ago

daothanhbinh20 commented 3 years ago

Hi, i ran test_fashion_recommender.py and obtained a compatibility score for a set of cloths. However, how can i generate an outfit given an input image? Which function should i use and Are there any papers on the recommender which you use to implement yours? Thank you

shariquep commented 1 year ago

as far as I can see there is currently no function for generating outfit. Here is what I am doing: I convert all the image into embeddings and segment them category wise. Then I take an input image and its category, pass it through the model to get its embeddings, find its compatibility with every item and select top 3 items of each category. Finally, you can create different combinations by getting one item from all the category and then find its compatibility score and then output the combination with highest score