Clothes Segmentation
Huggingface cloth segmentation using U2NET
This repo contains inference code and gradio demo script using pre-trained U2NET model for Cloths Parsing from human portrait.
Here clothes are parsed into 3 category: Upper body(red), Lower body(green) and Full body(yellow). The provided script also generates alpha images for each class.
Original Source Code
https://github.com/wildoctopus/huggingface-cloth-segmentation.gitInference
clone the repo
git clone git@github.com:asadrizvi64/Cloth-segmentation.git.Install dependencies
pip install -r requirements.txtRun
python process.py --image 'input/03615_00.jpg' . Script will automatically download the pretrained model.Outputs will be saved in
output folder.output/alpha/.. contains alpha images corresponding to each class.output/cloth_seg contains final segmentation.Gradio Demo
Run
python app.pyNavigate to local or public url provided by app on successfull execution.
OR
Inference in colab from here
Huggingface Demo
Check gradio demo on Huggingface space from here huggingface-cloth-segmentation.
Output samples
This model works well with any background and almost all poses.
Acknowledgements
U2net model is from original u2net repo. Thanks to Xuebin Qin for amazing repo.
Most of the code is taken and modified from levindabhi/cloth-segmentation
0 likes · 32 views · Sep 9, 2024
Used influence from the git repo . Contribute to asadrizvi64/Cloth-segmentation development by creating an account on GitHub.
