인물 사진 Segmentation 실습

Privacy-Preserving Portrait Matting를 실습해 보았다.

Pycharm 가상환경 만들기

참고

requirements에 맞춰서 설치해준다.

pip install numpy
pip install opencv-python
pip install pillow
pip install scikit-image
pip install scipy
pip install tqdm

다시 Visual Studio Code에서 시작

가상환경 잘 되었고, pip install도 완료.

python core/test.py --cuda --dataset_choice=SAMPLES --model_path='models/pretrained/p3mnet_pretrained_on_p3m10k.pth' --test_choice=HYBRID

작동이 되지 않았다.

CUDA와 cuDNN 설치를 완료한 후에 코드를 다시 실행해보자.
pip install tensorflow
pip install tensorflow-gpu==2.4.0
안되네...

CUDA버전 확인 nvcc --version

흠..
참고코드를 사용해서 torch.cuda.is_available() 도 확인해보았을 때, False가 뜬다..

PyTorch 사이트에서 다운받아보자.

이것도 완료되고,
CUDA 드라이버도 다시 받았는데 둘 중 하나 덕분에 코드를 다시 쳐서 돌리니깐 실행이 되었다!!!

cuda의 available도 True가 되었다.

실행

python core/test.py --cuda --dataset_choice=SAMPLES --model_path=models/pretrained/p3mnet_pretrained_on_p3m10k.pth --test_choice=HYBRID
라고 입력하면 코드가 실행되면서 original 사진의 배경이 제거된 사진을 얻을 수 있다.