face-segmentation_pytorch

🔥 Getting Started

1. Clone the code and prepare the environment

git clone https://github.com/Mrkomiljon/face-segmentation_pytorch.git
cd face-segmentation_pytorch

# create env using conda
conda create -n face-segmentation python==3.9.18
conda activate face-segmentation

🔥 Training

• 1.Prepare training data: -- download CelebAMask-HQ dataset

  -- change file path in the prepropess_data.py and run

python prepropess_data.py

If you do not wish to train the model, you can download pre-trained model and save it in res/cp.

🔥 Evaluate using GPU

python test.py

🔥 Results