Clothes Segmentation
This is my implementation of this project!
Dependencies
- python >= 3.6
- pytorch >= 1.2
- opencv
- matplotlib
- albumentations, iglovikov_helper_functions, pytorch_lightning, pytorch_toolbelt, segmentation-models-pytorch, tqdm, wandb
Installation
- Download & install cuda 10.2 toolkit here
- Download & install anaconda python 3.7 version
- Install Dependencies
- Run
main.py
A example
