본문 바로가기
반응형

AI/기술7

CycleGAN으로 성별 전환 프로그램 만들기 (2) 학교 수업의 일환으로 학교 서버를 빌렸는데, 서버가 잠깐 노는 동안 CycleGAN을 학습시켜보았습니다. 시간 부족으로 Tuning을 제대로 하지 못하여 결과가 웃기긴 하지만 어쨌든 가능성은 보이네요! CycleGAN의 코드에 대한 설명은 하지 않도록 하겠습니다. 시간이 없어 코드 정리도 제대로 못했고, 튜밍도 안되어서! 그럼 중간 결과 포스팅 해보겠습니다. 데이터 전처리 연예인의 사진에서 얼굴만 Crop 하는 전처리 과정을 거쳤다면, 이번에는 수동으로 데이터의 일관성을 맞추어주는 작업을 거쳤습니다. 제 해석으로는, 남녀의 성별적인 차이를 제외하곤 다른 차이점이 사진에 존재하지 않는 것이 학습에 유리할 것이라고 생각하였기 때문입니다. 그렇기 때문에 특이한 각도의 사진, 분장 사진과 같은 평범한 사진들과는.. 2020. 12. 11.
CycleGAN으로 성별 전환 프로그램 만들기 (1) CycleGAN 설명 유튜브 영상 - Naver D2 (링크) CycleGAN 논문 (링크) CycleGAN 페이지 (링크) CycleGAN에 대한 내용은 이번 포스팅에서는 다루지 않을 것입니다. 위의 유튜브 영상은 CycleGAN의 저자 중 한 명인 박태성 박사님이 발표하시는 영상입니다. 정말 재밌게 발표하시고 길이도 그다지 길지 않으니 꼭 한 번쯤 보시는 것을 적극 추천드립니다! 프로젝트 시작 동기 본래 CycleGAN에 관심이 많았는데, 이번에는 작게나마 사람들이 사용할 수 있는 서비스를 만들어보고 싶었습니다. 그래서 사람들이 재밌어할 만한 것이 무엇이 있을까 하다가 성별 전환 프로그램을 만들기로 결정하였습니다. 사실 성별 전환 카메라로 동일한 기능을 제공하는 앱이 이미 있지만 (Face App이 .. 2020. 12. 8.
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.
반응형