13.3.2 GAN 구현

    GAN 예제 역시 MNIST 데이터셋을 사용하여 파이토치로 구현하는 방법을 알아보겠습니다.

    먼저 필요한 라이브러리를 호출합니다.

    코드 13-22 라이브러리 호출

    import imageio ------ 이미지 데이터를 읽고 쓸 수 있는 쉬운 인터페이스를 제공하는 라이브러리
    from tqdm import tqdm
    
    import torch
    import torch.nn as nn
    import torch.optim as optim
    from torch.utils.data import Dataset, DataLoader
    import matplotlib.pylab as plt
    
    from torchvision.utils import make_grid, save_image
    import torchvision.datasets as datasets
    import torchvision.transforms as transforms
    matplotlib.style.use('ggplot') ------ ①
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.