CycleGAN 2

(CycleGAN) pytorch로 CycleGAN의 Generator / Discriminator 구현

안녕하세요 오늘은 CycleGAN의 Generator와 Discriminator를 pytorch로 구현해보았습니다. CycleGAN의 Generator는 U-Net 구조나 ResNet 구조를 이용하는데 저는 Resnet 구조로 구현 해보았습니다. (ResNet block 9개) 참고 페이지 : https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/tree/003efc4c8819de47ff11b5a0af7ba09aee7f5fc1 GitHub - junyanz/pytorch-CycleGAN-and-pix2pix: Image-to-Image Translation in PyTorch Image-to-Image Translation in PyTorch. Contrib..

AI 2022.04.14

CycleGAN 학습을 위해 python으로 이미지 크롤링하기(wikiart, Vincent van Gogh, selenium)

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

python 2022.04.08