본문 바로가기
카테고리 없음

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

by 코드퀸 2022. 12. 10.
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
반응형

댓글