Conjunto de dados Flickr-Faces-HQ-Wrinkle (FFHQ-Wrinkle)

Imagem preliminar

FFHQ-Wrinkle é uma extensão do conjunto de dados FFHQ (Flickr-Faces-HQ), projetado especificamente para incluir recursos adicionais relacionados a rugas faciais. Este conjunto de dados visa apoiar a pesquisa e o desenvolvimento em reconhecimento facial, simulação de envelhecimento e outras áreas relacionadas.

Se você usar este conjunto de dados para sua pesquisa, cite nosso artigo:

Para ser atualizado

O primeiro conjunto público de dados sobre rugas faciais, 'FFHQ-Wrinkle', compreende pares de imagens faciais e suas correspondentes máscaras de rugas. Nós nos concentramos em rótulos de rugas enquanto utilizamos o conjunto de dados de imagens faciais de alta resolução FFHQ (Flickr-Faces-HQ) existente, que contém 70.000 imagens faciais de alta resolução (1024x1024) capturadas sob vários ângulos e condições de iluminação. O conjunto de dados que fornecemos consiste em um conjunto de máscaras de rugas rotuladas manualmente (N=1.000) e um conjunto de máscaras de rugas "fracas", ou mapas de textura mascarados, gerados sem trabalho humano (N=50.000). Selecionamos 50.000 imagens do conjunto de dados FFHQ, especificamente IDs de imagem 00000 a 49999. Usamos essas 50.000 imagens de rosto para criar as rugas fracamente rotuladas e amostramos aleatoriamente 1.000 imagens delas para criar as rugas verdadeiras.

Visão geral

Todos os dados estão hospedados no Google Drive:

Caminho	Tamanho	Arquivos	Formatar	Descrição
conjunto de dados ffhq-rugas	6,95 GB			Pasta principal
├ máscaras anti-rugas manuais	2,8 MB	1.000	PNG em escala de cinza de 8 bits (uint8)	Etiquetas enrugadas anotadas manualmente com resolução de 1024x1024.
├ máscaras anti-rugas fracas	6,94 GB	50.000	PNG em escala de cinza de 8 bits (uint8)	Etiquetas com rugas fracas geradas automaticamente com resolução de 1024x1024.
└ rótulos analisados por face	5,1 MB	1.000	Matriz numpy (npy)	Inclui rótulos analisados por face para extração de face.

Uso

Todos os scripts devem ser executados a partir da pasta raiz do repositório (ffhq-wrinkle-dataset).

Download de etiqueta de rugas

Para baixar os rótulos de rugas que consistem em 1.000 máscaras manuais de rugas e 50.000 máscaras de rugas fracas, execute download_ffhq_wrinkle.sh . Este script fará o download das máscaras de rugas manuais e das máscaras de rugas fracas na base_folder especificada.

bash download_ffhq_wrinkle.sh

Você pode definir a pasta de download padrão modificando a variável base_folder em download_ffhq_wrinkle.sh .

Além disso, você pode baixar os rótulos diretamente por meio do link fornecido do Google Drive.

A estrutura da pasta após seguir as instruções é a seguinte:

{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

Baixar imagem de rosto

Para baixar as imagens faciais originais do FFHQ, consulte o site do FFHQ Dataset. Usamos o subconjunto 'images1024x1024' com IDs de 00000 a 49999 do conjunto de dados FFHQ original. Após baixar as imagens, coloque-as na base_folder designada.

A estrutura da pasta após o download das imagens faciais é a seguinte:

{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

Preparação de imagem de rosto mascarado e diretrizes de mascaramento

Para seguir a estratégia de treinamento descrita em nosso artigo, você deve preparar as imagens faciais que correspondem às máscaras manuais de rugas, bem como as imagens faciais mascaradas onde outras áreas além do rosto estão mascaradas. Ao executar face_masking.sh , você pode obter as imagens de rosto e as imagens de rosto mascarado correspondentes aos rótulos de rugas manuais na pasta base.

bash face_masking.sh

Nota : Para executar esta tarefa, as imagens faciais originais do conjunto de dados FFHQ devem estar em base_folder .

Como alternativa, você pode baixar diretamente os rótulos analisados por rosto no link do Google Drive e colocá-los em base_folder . Em seguida, execute sequencialmente png_parsing.py e 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

Fornecemos rótulos de análise facial para as imagens de rosto correspondentes aos rótulos de rugas manuais como matrizes numpy 512x512, que foram obtidas usando face-parsing.PyTorch. Especificamente, mascaramos todas as áreas, exceto o rosto e o nariz.

A estrutura de pastas após as instruções é a seguinte:

{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

Licenças

O conjunto de dados FFHQ-Wrinkle é fornecido sob a mesma licença Creative Commons BY-NC-SA 4.0 que o conjunto de dados FFHQ original. Você é livre para usar, redistribuir e adaptar este conjunto de dados para fins não comerciais sob as seguintes condições:

Atribuição: Você deve dar o devido crédito citando nosso artigo , fornecer um link para a licença e indicar se foram feitas alterações. Você pode fazê-lo de qualquer maneira razoável, mas não de forma que sugira que o licenciante endossa você ou seu uso.
Não Comercial: Você não pode usar o material para fins comerciais.
Compartilhamento pela mesma licença: Se você remixar, transformar ou desenvolver o material, deverá distribuir suas contribuições sob a mesma licença do original.

Ao usar o conjunto de dados FFHQ-Wrinkle, você concorda em cumprir os termos desta licença. Para quaisquer dúvidas ou solicitações de uso comercial, entre em contato com os criadores originais do conjunto de dados FFHQ e os autores do conjunto de dados FFHQ-Wrinkle.

As imagens individuais foram publicadas no Flickr por seus respectivos autores sob licença 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. Todas essas licenças permitem uso, redistribuição e adaptação gratuitos para fins não comerciais . Porém, alguns deles exigem o devido crédito ao autor original, bem como a indicação de quaisquer alterações feitas nas imagens. A licença e o autor original de cada imagem estão indicados nos metadados.