| # Clothes Segmentation | |
| This is my implementation of [this project](https://github.com/ternaus/cloths_segmentation)! | |
| ## Dependencies | |
| - python >= 3.6 | |
| - [pytorch](https://pytorch.org/) >= 1.2 | |
| - opencv | |
| - matplotlib | |
| - albumentations, iglovikov_helper_functions, pytorch_lightning, pytorch_toolbelt, segmentation-models-pytorch, tqdm, wandb | |
| ## Installation | |
| 1. Download & install cuda 10.2 toolkit [here](https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=debnetwork) | |
| 2. Download & install anaconda python 3.7 version | |
| 3. Install Dependencies | |
| 4. Run `main.py` | |
| ## A example | |
| <a href="url"><img src="test.jpg" align="left" style="width=50%" ></a> | |
| <a href="url"><img src="test.png" align="left" style="width=50%" ></a> | |