728x90
반응형
디자인도 인공지능이 하는 요즘 시대
요즘 인공지능으로 만든 작품이 심심치 않게 전시회에 출품된다는 뉴스를 보았습니다.
인공지능이 정말 다양한 분야에서 활용되고 있는 것 같습니다.
오늘은 인공지능으로 간단하게 패션 이미지를 생성해보는 코드를 소개하고 합니다.
코드는 아래의 pytorch 공식 홈페이지에서 가져왔으며,
아래 링크를 들어가면, 환경설치 없이도 간단하게 코드를 테스트해볼 수 있습니다.
https://pytorch.org/hub/facebookresearch_pytorch-gan-zoo_dcgan/
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
오늘의 코드
코드의 설명을 주석을 참고해주세요.
짧은 코드이니 한번 쭉 보고 각 줄의 역할을 이해하는 방향으로 보면 좋을 듯 싶습니다.
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() # 사용자 화면에 제공
728x90
반응형
댓글