PGGAN을 이용한 강아지, 고양이 생성 모델 만들기 (1)

이번 글에서는 데이터 전처리를 진행해보도록 하겠습니다.

 

데이터셋은 아래 링크에서 다운로드 받으시면 됩니다.

https://www.kaggle.com/andrewmvd/animal-faces

Animal Faces

강아지와 고양이 생성 모델이기 때문에 dog과 cat 폴더만 사용할 것입니다.

 

먼저 PIL Image를 tensor로 변환하는 메소드와 폴더 안의 이미지를 모두 tensor로 변환하는 메소드를 만들어줍니다.

 

그리고 폴더 리스트를 모두 이미지로 변환한 후 반환하는 메소드를 만들어줍니다.

 

이제 알맞게 전처리를 하면 끝입니다.

 

추가로 몇가지를 더 해주도록 하겠습니다.

make gif 메소드는 epoch마다 train log를 찍어둘 것인데, 이를 한번에 연결해주기 위한 메소드입니다.

merge_test_pred 메소드는 epoch마다 찍어주는 train log에 사용되는데, 모델이 예측한 값을 1024x1024 해상도의 이미지에 알맞게 이어 붙혀서 반환합니다.

 

이렇게 데이터 전처리는 끝이납니다.

다음 글에서는 최종적으로 훈련을 시켜보도록 하고, 결과를 확인해보겠습니다.

 

utils.py 코드 - https://github.com/SimplePro/PGGAN/blob/main/utils.py