cs

cloths-segmentation

Code for binary segmentation of cloths

Showing:

Popularity

Downloads/wk

0

GitHub Stars

111

Maintenance

Last Commit

1yr ago

Contributors

2

Package

Dependencies

9

License

MIT

Categories

Readme

Code for binary segmentation of various cloths

Installation

pip install -U cloths_segmentation

Example inference

Jupyter notebook with the example: Open In Colab

WebApp

https://clothssegmentation.herokuapp.com/

Data Preparation

Download the dataset from https://www.kaggle.com/c/imaterialist-fashion-2019-FGVC6

Process the data using script

The script will create process the data and store images to folder images and binary masks to folder labels.

Training

Define the config.

Example at cloths_segmentation/configs

You can enable / disable datasets that are used for training and validation.

Define the environmental variable IMAGE_PATH that points to the folder with images.

Example:

export IMAGE_PATH=<path to the the folder with images>

Define the environmental variable LABEL_PATH that points to the folder with masks.

Example:

export MASK_PATH=<path to the folder with masks>

Training

python -m cloths_segmentation.train -c <path to config>

Inference

python -m torch.distributed.launch --nproc_per_node=<num_gpu> cloths_segmentation/inference.py \
                                   -i <path to images> \
                                   -c <path to config> \
                                   -w <path to weights> \
                                   -o <output-path> \
                                   --fp16

Rate & Review

Great Documentation0
Easy to Use0
Performant0
Highly Customizable0
Bleeding Edge0
Responsive Maintainers0
Poor Documentation0
Hard to Use0
Slow0
Buggy0
Abandoned0
Unwelcoming Community0
100
No reviews found
Be the first to rate

Alternatives

No alternatives found

Tutorials

No tutorials found
Add a tutorial