search

minar09 / cp-vton-plus

Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020
https://minar09.github.io/cpvtonplus/
MIT License
343 stars 120 forks source link

GMM for bottom clothes #92

Open omaraga12 opened 2 years ago

omaraga12 commented 2 years ago

Hello. Thank you for this great job. Is there a way to train the GMM network for bottom clothes deformation like pants? I suppose I have to change in cp dataset file. Please can you guide me how I should start

thaithanhtuan commented 2 years ago

You can change the cp_dataset for pants by replace the label = 5 (top clothing) to the label of pants or skirt in your dataset. Some place that you have to change: +for calculate the loss extracted top clothing (label =5) should change to the label for bottom clothing +inshop clothing top and its mask -->> inshop clothing pants and its mask. +Clothing agnostic for top clothing --> change to clothing agnostic for pants.