search

levindabhi / cloth-segmentation

This repo contains code and a pre-trained model for clothes segmentation.
MIT License
565 stars 102 forks source link

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

Open karndeepsingh opened 2 years ago

karndeepsingh commented 2 years ago

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 commented 1 year ago

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 :)