jokerlc / RMGN-VITON

Official PyTorch implementation of "RMGN: A Regional Mask Guided Network for Parser-free Virtual Try-on" (IJCAI-ECAI 2022)
43 stars 5 forks source link

How to generate for custom cloth and person? #2

Closed gaurav-95 closed 2 years ago

gaurav-95 commented 2 years ago

Hello, I want to run the model on a custom set of cloth and people. Do I generate masks for the custom data? If so, how do I get the masks that you have generated in the test data?

Or am I missing something else here?

jokerlc commented 2 years ago

Yes, the masks of the clothes are necessary. The masks in the test data are provided by others. But i think you can find any opensouce model for background removing, which can extract the masks of your custom cloth