본문 바로가기
반응형

딥러닝2

CycleGAN으로 성별 전환 프로그램 만들기 (2) 학교 수업의 일환으로 학교 서버를 빌렸는데, 서버가 잠깐 노는 동안 CycleGAN을 학습시켜보았습니다. 시간 부족으로 Tuning을 제대로 하지 못하여 결과가 웃기긴 하지만 어쨌든 가능성은 보이네요! CycleGAN의 코드에 대한 설명은 하지 않도록 하겠습니다. 시간이 없어 코드 정리도 제대로 못했고, 튜밍도 안되어서! 그럼 중간 결과 포스팅 해보겠습니다. 데이터 전처리 연예인의 사진에서 얼굴만 Crop 하는 전처리 과정을 거쳤다면, 이번에는 수동으로 데이터의 일관성을 맞추어주는 작업을 거쳤습니다. 제 해석으로는, 남녀의 성별적인 차이를 제외하곤 다른 차이점이 사진에 존재하지 않는 것이 학습에 유리할 것이라고 생각하였기 때문입니다. 그렇기 때문에 특이한 각도의 사진, 분장 사진과 같은 평범한 사진들과는.. 2020. 12. 11.
GAN[Generative Adversarial Network] 구현하기 w/ Pytorch 2014년에 발표된 GAN은 이미 너무나 많은 갈래가 생겼습니다. 한 갈래인 CycleGAN에 대해 알게 되었는데, 정말 신기하다는 생각이 들었습니다. 그래서 이왕 궁금한 거, GAN부터 구현해보는 생각을 하게 되었습니다. 과제가 쌓여있지만... 일단... 하고 싶은 것부터 하련다.. 구현 코드는Pytorch-GAN repo를 참고하였습니다. GAN의 논문은 누구나 읽어볼 가치가 있습니다! 어떤 것을 만들어볼까? 누구나 하는 MNIST로 구현하자기엔 너무 지겨워, kaggle에서 구한 데이터들로 진행하였습니다. 총 8가지로 비행기, 고양이, 차, 강아지, 꽃, 과일, 오토바이, 사람입니다. parser.add_argument("--object", type=str, default='person', help.. 2020. 11. 30.
반응형