search

levindabhi / cloth-segmentation

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

Prorosal: Pack model for Huggingface inference #15

Open betafuzz opened 1 year ago

betafuzz commented 1 year ago

Clothes segmentation is an important task in the field of computer vision and has various applications such as fashion analysis, virtual try-on, and image editing. However, there are currently limited pre-trained models available for clothes segmentation on the Hugging Face platform.

I propose adding a pre-trained clothes segmentation model to the Hugging Face platform.

yaomilwaukee commented 1 year ago

hello the link of pre-trained model download doesn't work, Did you download the pre-trained model file? if did, could you share it ?

wildoctopus commented 1 year ago

Hi @betafuzz

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