레플리
글 수 206

StyleGAN을 이말년 웹툰으로 파인튜닝한 모델

조회 수 2731 추천 수 0 2020.08.07 02:23:50


 

https://github.com/bryandlee/malnyun_faces

 

 

 

요즘 딥러닝의 화두는 전이학습(Transfer Learning)입니다. 지금은 과거처럼 바닥부터 새로 학습하지 않습니다. 사람과 비슷하게 이전의 지식을 기반으로 다른 일을 배워나갑니다. 예를 들어, ImageNet과 같은 대량의 사진으로 사전훈련(Pre-training)을 합니다. 그다음 나만의 데이터로 파인튜닝(Fine-tuning)을 수행합니다. 이렇게 하면 적은 데이터로도 더 높은 성능을 낼 수 있습니다. 2018년부터는 자연어처리에서도 전이학습이 대세가 되었습니다. BERT나 GPT2가 대표적입니다.

 

이제는 GAN과 같은 생성 기법에서도 사전훈련 모델을 활용하기 시작했습니다. NVIDIA의 StyleGAN은 7만장의 얼굴사진으로 훈련을 했습니다. 이 모델을 기반으로 파인튜닝을 했는데, 불과 500장의 웹툰 사진만으로도 학습이 가능하다고 합니다.

 

엮인글 :
List of Articles
제목 글쓴이 날짜 조회 수sort
Transformer와 동급의 성능에 속도는 훨씬 빨라진 MLP-Mixer file 깊은바다 2021-06-11 1465
R-CNN 에서 부터 Mask R-CNN 까지 깊은바다 2019-09-03 1409
딥페이크로 일론 머스크인 척 친구들을 속이는 영상 깊은바다 2020-04-19 1394
목소리와 입모양을 변경할 수 있는 GAN 모델 깊은바다 2019-06-13 1391
텐서플로우에서 소프트맥스로 숫자 인식 구현 file 깊은바다 2017-03-09 1378
오픈소스로 공개된 Text-to-Image 모델 - Stable Diffusion 깊은바다 2022-09-07 1368
텐서플로우로 선형 회귀 구현 file 깊은바다 2017-03-09 1361
사진을 디즈니 만화로 바뀌주는 딥러닝 모델 - Toonify 깊은바다 2020-09-21 1358
아이폰에서 실행하는 Stable Diffusion 앱 - Draw Things 깊은바다 2022-11-11 1330
딥러닝 범용 모델 - Foundation Models 깊은바다 2021-09-06 1323
RNN(Recurrent Neural Network) 튜토리얼 깊은바다 2017-07-22 1303
하나의 얼굴 이미지로 다양하게 변형된 얼굴을 만드는 StarGAN [1] NeuroWhAI 2017-12-30 1303
딥러닝 GAN으로 생성한 고해상도 사진 [2] 깊은바다 2017-06-08 1292
전문가 수준의 사진을 생성하는 NVIDIA의 PoE-GAN 깊은바다 2022-02-16 1285
최신 딥러닝 모델들의 학습 비용 깊은바다 2019-12-12 1279