Ensemble de données Flickr-Faces-HQ-Wrinkle (FFHQ-Wrinkle)

Image d'accroche

FFHQ-Wrinkle est une extension de l'ensemble de données FFHQ (Flickr-Faces-HQ), spécialement conçu pour inclure des fonctionnalités supplémentaires liées aux rides du visage. Cet ensemble de données vise à soutenir la recherche et le développement dans le domaine de la reconnaissance faciale, de la simulation du vieillissement et d'autres domaines connexes.

Si vous utilisez cet ensemble de données pour votre recherche, veuillez citer notre article :

À mettre à jour

Le premier ensemble de données public sur les rides du visage, « FFHQ-Wrinkle », comprend des paires d'images de visage et leurs masques anti-rides correspondants. Nous nous sommes concentrés sur les étiquettes de rides tout en utilisant l'ensemble de données d'images de visage haute résolution existant FFHQ (Flickr-Faces-HQ), qui contient 70 000 images de visage haute résolution (1 024 x 1 024) capturées sous différents angles et conditions d'éclairage. L'ensemble de données que nous fournissons se compose d'un ensemble de masques anti-rides étiquetés manuellement (N = 1 000) et d'un ensemble de masques anti-rides « faibles », ou de cartes de texture masquées, générés sans travail humain (N = 50 000). Nous avons sélectionné 50 000 images de l'ensemble de données FFHQ, en particulier les ID d'image 00000 à 49999. Nous avons utilisé ces 50 000 images de visage pour créer les rides faiblement étiquetées et avons échantillonné au hasard 1 000 images à partir de celles-ci pour créer les rides de vérité terrain.

Aperçu

Toutes les données sont hébergées sur Google Drive :

Chemin	Taille	Fichiers	Format	Description
ensemble de données ffhq-rides	6,95 Go			Dossier principal
├ masques-rides-manuels	2,8 Mo	1 000	PNG en niveaux de gris 8 bits (uint8)	Étiquettes anti-rides annotées manuellement avec une résolution de 1024 x 1024.
├ masques anti-rides	6,94 Go	50 000	PNG en niveaux de gris 8 bits (uint8)	Étiquettes de rides faibles générées automatiquement avec une résolution de 1024 x 1024.
└ étiquettes analysées par visage	5,1 Mo	1 000	Tableau Numpy (npy)	Comprend des étiquettes analysées par visage pour l’extraction de visage.

Usage

Tous les scripts doivent être exécutés à partir du dossier racine du référentiel (ffhq-rides-dataset).

Télécharger l'étiquette des rides

Pour télécharger les étiquettes anti-rides composées de 1 000 masques anti-rides manuels et de 50 000 masques anti-rides faibles, veuillez exécuter download_ffhq_wrinkle.sh . Ce script téléchargera à la fois les masques anti-rides manuels et les masques anti-rides faibles dans le base_folder spécifié.

bash download_ffhq_wrinkle.sh

Vous pouvez définir le dossier de téléchargement par défaut en modifiant la variable base_folder dans le download_ffhq_wrinkle.sh .

De plus, vous pouvez télécharger directement les étiquettes via le lien Google Drive fourni.

La structure des dossiers après avoir suivi les instructions est la suivante :

{base_folder}/
├── manual_wrinkle_masks/
│   ├── 00001.png
│   ├── 00011.png
│   ├── ...
│   └── 21035.png
└── weak_wrinkle_masks/
    ├── 00000/
    │   ├── 00000.png
    │   ├── 00001.png
    │   ├── ...
    │   └── 00999.png
    ├── 01000/
    │   ├── 01000.png
    │   ├── 01001.png
    │   ├── ...
    │   └── 01999.png
    ├── ...
    └── 49000/
        ├── 49000.png
        ├── 49001.png
        ├── ...
        └── 49999.png

Téléchargement d'images de visage

Pour télécharger les images originales du visage FFHQ, veuillez vous référer au site Web FFHQ Dataset. Nous utilisons le sous-ensemble « images1024x1024 » avec les ID de 00000 à 49999 de l'ensemble de données FFHQ d'origine. Après avoir téléchargé les images, placez-les dans le base_folder désigné.

La structure des dossiers après le téléchargement des images de visage est la suivante :

{base_folder}/
├── images1024x1024/
│   ├── 00000/
│   │   ├── 00000.png
│   │   ├── 00001.png
│   │   ├── ...
│   │   └── 00999.png
│   ├── ...
│   └── 49000/
│       ├── 49000.png
│       ├── 49001.png
│       ├── ...
│       └── 49999.png
├── manual_wrinkle_masks/
│   ├── 00001.png
│   ├── 00011.png
│   ├── ...
│   └── 21035.png
└── weak_wrinkle_masks/
    ├── 00000/
    │   ├── 00000.png
    │   ├── 00001.png
    │   ├── ...
    │   └── 00999.png
    ├── ...
    └── 49000/
        ├── 49000.png
        ├── 49001.png
        ├── ...
        └── 49999.png

Préparation des images de visage masqué et directives de masquage

Pour suivre la stratégie de formation décrite dans notre article, vous devez préparer les images du visage correspondant aux masques anti-rides manuels, ainsi que les images du visage masqué où des zones autres que le visage sont masquées. En exécutant face_masking.sh , vous pouvez obtenir les images de visage et les images de visage masqué correspondant aux étiquettes de rides manuelles dans le dossier de base.

bash face_masking.sh

Remarque : Pour effectuer cette tâche, les images de visage originales de l'ensemble de données FFHQ doivent se trouver dans le base_folder .

Vous pouvez également télécharger directement les étiquettes analysées par visage à partir du lien Google Drive et les placer dans le base_folder . Exécutez ensuite séquentiellement png_parsing.py et face_masking.py .

python png_parsing.py $base_folder /images1024x1024 $base_folder /manual_wrinkle_masks $base_folder /face_images
python face_masking.py $base_folder /face_parsed_labels $base_folder /face_images $base_folder /masked_face_images

Nous fournissons des étiquettes analysées par visage pour les images de visage correspondant aux étiquettes de rides manuelles sous forme de tableaux numpy 512x512, qui ont été obtenus à l'aide de l'analyse de visage.PyTorch. Concrètement, nous masquons toutes les zones sauf le visage et le nez.

La structure des dossiers après les instructions est la suivante :

{base_folder}/
├── etcs/
│   └── face_parsed_labels/
│       ├── 00000.npy
│       ├── 00011.npy
│       ├── ...
│       └── 21035.npy
├── face_images/
│   ├── 00001.png
│   ├── 00011.png
│   ├── ...
│   └── 21035.png
├── images1024x1024/
│   ├── 00000/
│   │   ├── 00000.png
│   │   ├── 00001.png
│   │   ├── ...
│   │   └── 00999.png
│   ├── ...
│   └── 49000/
│       ├── 49000.png
│       ├── 49001.png
│       ├── ...
│       └── 49999.png
├── manual_wrinkle_masks/
│   ├── 00001.png
│   ├── 00011.png
│   ├── ...
│   └── 21035.png
├── masked_face_images/
│   ├── 00001.png
│   ├── 00011.png
│   ├── ...
│   └── 21035.png
└── weak_wrinkle_masks/
    ├── 00000/
    │   ├── 00000.png
    │   ├── 00001.png
    │   ├── ...
    │   └── 00999.png
    ├── ...
    └── 49000/
        ├── 49000.png
        ├── 49001.png
        ├── ...
        └── 49999.png

Licences

L'ensemble de données FFHQ-Wrinkle est fourni sous la même licence Creative Commons BY-NC-SA 4.0 que l'ensemble de données FFHQ d'origine. Vous êtes libre d'utiliser, de redistribuer et d'adapter cet ensemble de données à des fins non commerciales dans les conditions suivantes :

Attribution : vous devez attribuer le crédit approprié en citant notre article , fournir un lien vers la licence et indiquer si des modifications ont été apportées. Vous pouvez le faire de toute manière raisonnable, mais en aucun cas qui suggère que le concédant de licence vous approuve ou approuve votre utilisation.
Non commercial : vous ne pouvez pas utiliser le matériel à des fins commerciales.
ShareAlike : si vous remixez, transformez ou développez le matériel, vous devez distribuer vos contributions sous la même licence que l'original.

En utilisant l'ensemble de données FFHQ-Wrinkle, vous acceptez de respecter les termes de cette licence. Pour toute demande complémentaire ou demande d'utilisation commerciale, veuillez contacter les créateurs originaux de l'ensemble de données FFHQ et les auteurs de l'ensemble de données FFHQ-Wrinkle.

Les images individuelles ont été publiées sur Flickr par leurs auteurs respectifs sous licence Creative Commons BY 2.0, Creative Commons BY-NC 2.0, Public Domain Mark 1.0, Public Domain CC0 1.0 ou US Government Works. Toutes ces licences permettent une utilisation, une redistribution et une adaptation gratuites à des fins non commerciales . Cependant, certains d’entre eux nécessitent de mentionner de manière appropriée l’auteur original, ainsi que d’indiquer les modifications apportées aux images. La licence et l'auteur original de chaque image sont indiqués dans les métadonnées.