search

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

Training code #4

Open hamaadtahiir opened 2 years ago

hamaadtahiir commented 2 years ago

Kindly provide the training code

ansarker commented 1 year ago

@jokerlc planning to provide training code?

MaksMaib commented 1 year ago

Hi, there. Does i understand correct, that you have the same train loop like in PF-AFN, the only difference in last step, where you use your mask guided module? Thanks for you work and answer.

jokerlc commented 1 year ago

yes, the biggest difference is the PF-Generator with our mask guided module; another difference is the postrue awareness loss

MaksMaib commented 1 year ago

Thank you, for rapid and detailed answer. You rock!