StyleGAN : un générateur de visages hyper-réalistes open source proposé par NVIDIA
mer, 27/03/2019 - 15:53
StyleGAN est basé sur l'outil de machine learning de Google TensorFlow. Il génère des visages à partir de vraies photos. Des visages si réalistes que l'on ne peut pas dire s'ils sont réels ou artificiels d'après NVIDIA. StyleGAN tourne sous Windows ou Linux.
Pour générer des visages, les développeurs peuvent utiliser leur propre modèle ou le modèle pré-entraîné. Les modèles pré-formés sont stockés en tant que fichiers pickle (.pkl) dans Google Drive.
StyleGAN demande une puissance de calcul très importante et selon NVIDIA le temps de formation pour des images de résolution 1024x1024 à l'aide du GPU Tesla V100 est le suivant :
GPUs | Temps | |
1 | 1 | 5 semaines |
2 | 2 | 3 semaines |
3 | 4 | 2 semaines |
4 | 8 | 1 semaine |
Hormis le temps requis, NIDIA souligne que le nombre de GPU est très important pour obtenir des résultats réalistes et recommande l'utilisation de 8 GPUs.
StyleGAN est un logiciel libre sous licence NVIDIA Creative Commons BY-NC 4.0, disponible sur GitHub.