인공지능/실습 (3) 썸네일형 리스트형 PGGAN을 이용한 강아지, 고양이 생성모델 만들기 (2) 이번 글에서는 지난 글에서 만든 데이터셋과 모델을 이용하여 훈련을 시켜보도록 하겠습니다. 먼저 데이터셋과 model state dict를 가져오기 편리하게 steps별로 path를 리스트형태로 정의해줍니다. steps는 모델의 steps와 같은 의미입니다. test_size는 train log에 찍을 이미지의 개수입니다. dataset의 directory_path는 미리 정의해둔 dataset_path의 self.steps번째에서 찾을 수 있습니다. dataset을 로드하여 trainloader와 validloader를 정의해줍니다. 생성자와 판별자도 정의해주고요. loss function과 optimizer도 정의해줍니다. test_z는 train log에 사용할 잠재벡터입니다. 마지막으로 모델의 st.. PGGAN을 이용한 강아지, 고양이 생성 모델 만들기 (1) 이번 글에서는 데이터 전처리를 진행해보도록 하겠습니다. 데이터셋은 아래 링크에서 다운로드 받으시면 됩니다. https://www.kaggle.com/andrewmvd/animal-faces Animal Faces 16,130 images belonging to 3 classes. www.kaggle.com 강아지와 고양이 생성 모델이기 때문에 dog과 cat 폴더만 사용할 것입니다. 먼저 PIL Image를 tensor로 변환하는 메소드와 폴더 안의 이미지를 모두 tensor로 변환하는 메소드를 만들어줍니다. 그리고 폴더 리스트를 모두 이미지로 변환한 후 반환하는 메소드를 만들어줍니다. 이제 알맞게 전처리를 하면 끝입니다. 추가로 몇가지를 더 해주도록 하겠습니다. make gif 메소드는 epoch마다 .. PGGAN을 이용한 강아지, 고양이 생성모델 만들기 (0) 이 챕터의 목적은 PGGAN을 이용하여 256x256 해상도의 강아지, 고양이 생성모델을 만드는 것입니다. 먼저 프로젝트를 구현해보기 전에 PGGAN에 대한 이해가 필요합니다. PGGAN에 대한 이해가 부족하신 분은 아래 링크를 참고하세요. (https://simplepro.tistory.com/44) 모델 구조는 다음 링크를 참고하였습니다. (https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/ProGAN) 이 글에서는 모델 구조만 설계합니다. 먼저 논문에서 사용한 모델 구조를 봐보도록 하겠습니다. Generator 구조를 보면 n_channels가 512, 512, 512, 512, 256, .. 이전 1 다음
