search

sergeywong / cp-vton

Reimplemented code for "Toward Characteristic-Preserving Image-based Virtual Try-On Network"
MIT License
474 stars 182 forks source link

about mask loss is different between paper and code in stage II #22

Open helen0461 opened 5 years ago

helen0461 commented 5 years ago

Thx for your job and sharing. It's really helpful to my research. Thanks again. I have trained and tested your model successfully, and have a litte question about loss. When I train in satge II, the mask loss almost have not change from the beigining to the end. So, I check the calculation abouot mask loss, and find that the "loss_mask = criterionMask(m_composite, cm)" in line 133 of train.py is differernt from the equation (4) in paper. Is there any detail about this I can konw? Thank you a lot. @sergeywong

Hanv2 commented 4 years ago

i think the loss in code can achieve expectation that makes model concentrated more on the cloth region while preserving the background of the person in clothing‘s image.