레플리
글 수 203

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

조회 수 2714 추천 수 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
딥러닝 이론에서 텐서플로 실습까지 [2] 깊은바다 2017-11-08 786
Stable Diffusion을 사용한 사진 변환 앱 Lensa AI, 앱스토어 1위 달성 깊은바다 2022-12-09 786
사진을 텍스트로 설명 - CaptionBot file 깊은바다 2016-04-07 795
AutoML-Zero, 진화적 알고리즘을 사용하여 신경망 구조 설계 [2] 깊은바다 2020-03-13 820
당근마켓에서 딥러닝 활용하기 - 불량 게시물 검사 깊은바다 2017-12-23 821
자동회귀(autoregressive)로 이미지를 생성하는 iGPT file 깊은바다 2020-06-18 821
전신 사진을 생성하는 GAN 깊은바다 2019-05-05 832
알렉스넷과 Seq2Seq의 창시자 일리야 수츠케버(Ilya Sutskever) 깊은바다 2020-11-17 832
어떻게 해야 기계에게 글을 잘 읽고 말할 수 있게 할까? - 기계독해(MRC) 깊은바다 2020-01-14 836
앤드류 응의 YOLO 강의 영상 깊은바다 2020-09-23 838
3D 이미지를 만드는 GAN - StyleNeRF 깊은바다 2021-12-20 843
소프트웨어 2.0 깊은바다 2018-08-14 846
고해상도 GAN - A Style-Based Generator Architecture for GAN 깊은바다 2018-12-15 849
인공지능 과학자의 꿈, 범용 AI는 가능할까 깊은바다 2021-10-15 865
Recurrent Neural Network (RNN) 이해하기 - MNIST 숫자 인식 깊은바다 2018-04-11 880