How can upper body and Lower body segmentation can be separated out?

[karndeepsingh]

Hi,
Thanks for sharing this amazing work. Would like to know, once I have generated the output results, how I can separate out the upper body and lower body clothes from the image and process these two individual object for further task? Also, is their a JSON file that is being generated as output specifying the object detected in the image?

Please help.
Thanks

[wildoctopus]

Hi @karndeepsingh

For pretrained model download with inference script for Alpha Image generation and Cloth segmentation, please check this
https://github.com/wildoctopus/huggingface-cloth-segmentation

Have cleaned, modified the scripts to run on latest pytorch and other libraries and Its very simple to use as well.
The script will automatically download the pretrained model. So no worry at all.

You guys can also check the live demo at huggingface - https://huggingface.co/spaces/wildoctopus/cloth-segmentation

If you guys like this, please drop a star to my repo.

Thanks :)