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

Creating image-mask #74

Closed echoi3 closed 2 years ago

echoi3 commented 3 years ago

Could anyone help me with generating image-mask of person images?

I was able to generate cloth-mask by looking into your Cloth-Matching repo, but I figured that generating image-mask is a bit different from generating cloth-mask.

Any help would be appreciated.

Thanks in advance.

minar09 commented 3 years ago

Hi @echoi3 , I hope you solved your issue by now. Sorry for my late response. You can convert the image-parse into binary masks by changing all the labels value >0 to 255, or you can use the masking code from that repo for the person image. You can utilize any suitable binary masking network pretrained model as well, e.g. deepcut etc.

minar09 commented 3 years ago

You can also use the code from old commits: https://github.com/minar09/cp-vton-plus/blob/fcf29a512388881041ba84a97a43edfd1531cc38/body_binary_masking.py