앞의 이미지를 하나씩 쪼개서 확인하고 싶다면 다음 코드와 같이 실행하면 됩니다. 생성된 열 개의 이미지를 보여 줍니다.
코드 13-34 생성된 이미지 출력
fake_images = generator(torch.randn(b_size, nz).to(device))
for i in range(10):
fake_images_img = np.reshape(fake_images.data.cpu().numpy()[i],(28, 28))
plt.imshow(fake_images_img, cmap='gray')
plt.savefig('../chap13/img/fake_images_img' + str(i) + '.png')
plt.show()
다음 그림은 생성된 열 개의 이미지가 출력된 결과입니다.
▲ 그림 13-32 첫 번째 생성된 이미지 / 그림 13-33 두 번째 생성된 이미지
… 중간 생략 …
▲ 그림 13-34 아홉 번째 생성된 이미지 / 그림 13-35 열 번째 생성된 이미지
코드 13-32와 같이 에포크를 사용하여 학습 횟수를 늘리면 더 선명한 결과를 얻을 수 있습니다.