테크레시피

이 세상에 없는 고품질 음식 사진을?

지난 2019년 우버 엔지니어가 개발해 공개한 서비스(This Person Does Not Exist)는 AI를 이용해 이 세상에 존재하지 않는 인물 얼굴을 생성할 수 있는 툴로 전 세계에서 주목받았다. 이후 이 세상에 존재하지 않는 애니메이션 이미지를 생성하는 곳(This Anime Does Not Exist), 발 페티쉬를 위한 곳(This Foot Does Not Exist), 가상 고양이 사진을 생성할 수 있는 곳(This cat does not exist) 등 다양한 유사 도구가 등장했다. 그런데 스위스 기계학습 엔지니어(Florian Laurent)가 이 세상에 존재하지 않는 음식 사진을 생성할 수 있는 서비스(This Food Does Not Exist)를 개발했다고 발표했다.

인간 얼굴 사진을 생성하는 서비스에선 AI 알고리즘 일종이 적대적생성네트워크 GAN을 기반으로 한 스타일GAN(StyleGAN)을 이용했지만 이번 서비스는 스타일GAN 개선판인 스타일GAN2(StyleGAN2)를 기반으로 만들어져 있다.

최근 기계학습 업계에선 오픈AI가 개발한 달리(DALL・E)에 이용되고 있는 확산 모델이 유행하고 있어 사진 같은 리얼리즘에서 뛰어난 스타일GAN 모델에도 확산 모델이 따라잡고 있다는 것. 또 확산 모델은 문자 정보로부터 화상을 생성할 수 있기 때문에 제어성이 뛰어나지만 더 대규모 모델이 필요해 생성 시간이 길다는 점에서도 제한이 있다는 지적이다. 확산 모델을 이용하려면 많은 돈이 들 수 있어 이번에는 스타일GAN2를 이용해 서비스를 개발했다는 설명이다.

개발자는 구글 머신러닝 프로세서인 TPU v4-8을 사용해 스타일GAN2 모델 4개를 각각 10시간 이내에 훈련시켰다. 쿠키 이미지로 훈련한 이 서비스 내 이미지 크기는 256×256픽셀이다. 이번에 개발한 건 웹서비스로 공개되지는 않지만 TPU 훈련에 최적화된 코드와 사전 학습 모델이 깃허브에 공개되어 있다. 관련 내용은 이곳에서 확인할 수 있다.

추천기사