KiseKloset / DM-VTON

👗 DM-VTON: Distilled Mobile Real-time Virtual Try-On
https://sites.google.com/view/ltnghia/research/DMVTON
Other
107 stars 23 forks source link

Can't warp the lower garments. #22

Closed arifshahriarnsl closed 1 week ago

arifshahriarnsl commented 4 months ago

I used the DressCode dataset for training the model. It can warp the upper and full body garments but lower garments can't. What is the problem? Please help. 43000

zero-nnkn commented 3 months ago

I haven't reviewed your training code, but I guess your issue is the garment mask. In this code, I set the garment mask to upper body garments. So, in the training process, the model will try to fit the lower garments to the upper mask. You can modify your Dresscode Dataset class to get the lower-body garment mask.

image