[이미지생성] 왕초보도 가능한 인공지능/딥러닝으로 자동으로 이미지 생성하기

 

 

 

디자인도 인공지능이 하는 요즘 시대

 

요즘 인공지능으로 만든 작품이 심심치 않게 전시회에 출품된다는 뉴스를 보았습니다.

인공지능이 정말 다양한 분야에서 활용되고 있는 것 같습니다.

오늘은 인공지능으로 간단하게 패션 이미지를 생성해보는 코드를 소개하고 합니다. 

 

코드는 아래의 pytorch 공식 홈페이지에서 가져왔으며,

아래 링크를 들어가면, 환경설치 없이도 간단하게 코드를 테스트해볼 수 있습니다. 

 

 

https://pytorch.org/hub/facebookresearch_pytorch-gan-zoo_dcgan/

PyTorch

 

 

 

---

 

오늘의 코드

 

코드의 설명을 주석을 참고해주세요.

짧은 코드이니 한번 쭉 보고 각 줄의 역할을 이해하는 방향으로 보면 좋을 듯 싶습니다. 

 

import torch 
import matplotlib.pyplot as plt
import torchvision

use_gpu = True if torch.cuda.is_available() else False # gpu 설치 환경 확인
model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub', 'DCGAN', pretrained=True, useGPU=use_gpu) # 딥러닝 모델 불러오기

num_images = 64 # 생성할 이미지의 수
noise, _ = model.buildNoiseData(num_images) # 이미지 생성을 입력 데이터 생성
with torch.no_grad():
    generated_images = model.test(noise) # 패션 이미지 생성
    
plt.imshow(torchvision.utils.make_grid(generated_images).permute(1, 2, 0).cpu().numpy()) # 이미지 표출 준비
plt.show() # 사용자 화면에 제공