잡다한 포스팅

최근 GAN(Generative Adversarial Network)에 대해 공부를 하면서 MNIST 손글씨 이미지를 생성하는 기초적인 모델 학습을 해보았습니다. 예전에 인터넷을 보다가 사진을 입력하면 아래와 같이 입력한 사진을 명화 풍 으로 변경해주는 딥러닝 기술이 있다는 것을 알았는데 GAN 과 관련된 정보를 찾아보다가 'CycleGAN'을 이용하여 해당 기능을 수행할 수 있다는것같아서 직접 만들어보기로 했습니다. CycleGAN 학습을 위한 데이터셋을 구축하려고 'WikiArt (https://www.wikiart.org/)' 에서 'Vincent van Gogh' 의 그림들을 크롤링하고있습니다. 'WikiArt'에는 빈센트 반 고흐의 그림은 총 1931개 입니다. 지난 3년간 C++만 사용하다 P..

Pytorch를 이용하여 Model을 저장하는 방법은 아래와 같습니다. import torch import torch.nn as nn device = 'cuda' if torch.cuda.is_available() else 'cpu' # CNN_model 예시 class CNN_model(nn.Module): #tistory 코드 블럭의 문제인지 indent가 맞질 않습니다... def __init__(self): ... ... def forward(self, x): ... ... model = CNN_model() # torch.save(model, path_dict_file_name) # model 전체 저장 torch.save(model, 'model.pt') # state_dict = 학습 가능한..