Clothing-Segmentation

KO | EN

gitlite — search

#hacktoberfest #react #ai #golang #android #machine-learning #rust #docker #java #awesome #cli #vue #ios #open-source

Clothing-Segmentation

(★ 23)

Clothing Segmentation using FCN, DeepLabV2, U-Net in Keras

Clothing-Segmentation 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

Clothing Segmentation
===
Clothing Segmentation using deep learning.

First, after preprocessing the clothing dataset, four classes(backgrounds, skin, hiar, clothes) were segmented using deep neural network models(FCN, DeepLab, U-Net).
 
Datasets
---
Clothing Co-Parsing(CCP) dataset that i used can be found https://github.com/bearpaw/clothing-co-parsing.

Pre-Processing
---
After importing the image data , divide into four classes(Background, hair, skin, clothes) and resize 384 x 256.

Models
---
Models are found in model file. Model file includes FCN.py(Fully Covolutional Networks), DeepLabV2.py(DeepLab V2), UNet.py(U-Net).
U-Net has shown the best performance among the models in this project.

Requirement
---
* Python
* Keras
* Python packages : numpy, matplotlib, opencv, and so on...

Usage
---
In order to preprocessing the data, download the dataset from [this](https://github.com/bearpaw/clothing-co-parsing).

Then copy the annotation and photo files into the same path as pre_processing.py or pre_processing.ipynb in the download file.

After that you copy files, you can implement the codes. 
You can use Segmentation.py to get a segmented image.

### Pre-Processing
`python pre_processing.py`

### train and test
`python run_main.py --model unet`

`python segmentation.py --model_path ./save_model/UNet_model.h5 --image_path ./MyFile/example.png`

Results
---
### Implementation results by each model.

![FCN](https://raw.githubusercontent.com/IzPerfect/Clothing-Segmentation/master/result/show_result/FCN.png)
![DeepLab](https://raw.githubusercontent.com/IzPerfect/Clothing-Segmentation/master/result/show_result/DeepLab.png)
![UNet](https://raw.githubusercontent.com/IzPerfect/Clothing-Segmentation/master/result/show_result/UNet.png)

### Results by U-Net model.
![segmentation](https://raw.githubusercontent.com/IzPerfect/Clothing-Segmentation/master/result/iamge_segmentation/segmentation.png)

Reference Implementations
---
+ https://github.com/aurora95/Keras-FCN
+ https://github.com/DavideA/deeplabv2-keras
+ https://github.com/zhixuhao/unet

Clothing-Segmentation

Clothing-Segmentation 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Clothing-Segmentation

Clothing-Segmentation 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README