[IN-ISP] GAN 조사

junhjun·2023년 3월 7일
0

IN-ISP Project

목록 보기
2/16

우리 팀은 도로 이미지 증강을 위해 GAN을 채택했다.


여러 GAN 모델을 찾아본 결과,

DCGAN은 비교적 적은 이미지 데이터셋으로도 동작하지만, 고해상도 이미지 생성에는 한계가 있다고 한다.

PGGAN은 고해상도 이미지 생성에 뛰어난 성능을 보이지만, 대규모 데이터셋과 많은 컴퓨팅 자원이 필요하다고 한다.

PGGAN을 개선한 것이 StyleGAN이고, 그것의 문제를 보완한 것이 StyleGAN2이다.

현재 고화질 이미지 생성에는 StyleGAN2가 가장 적합한 것 같다.

이걸 이용해 모델을 만들면 좋을텐데, 쉽지 않다.


내게 GPU를 사용할 수 있는 환경은 Colab 밖에 없고, Colab은 이래저래 불편한 게 많다.

또, StyleGAN2 코드에 필요한 TensorFlow 1.15 버전을 Colab에서 지원하지 않는다.

연구실 서버를 사용할 수 있으면 좋겠다.

일단은 가상환경 만들어서 약식으로 돌려보는 것이 목표..

0개의 댓글