pytorch/vision

VGGFace2 dataset

Open

#1.139 geöffnet am 19. Juli 2019

Auf GitHub ansehen
 (3 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (6.858 Forks)batch import
enhancementhelp wantedmodule: datasets

Repository-Metriken

Stars
 (15.050 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 12T 8h) (14 gemergte PRs in 30 T)

Beschreibung

@fmassa how would you like to go about this?

Currently, I preprocess VGGFace2 by cropping out the face from each image using the provided bounding box csv. After doing that, I load it as an ImageFolder dataset.

VGGFace2 also has a bunch of other CSVs that annotate gender, age group, and facial landmarks.

I propose we allow users to specify the targets they want (such as ID, gender, age, bounding box etc.) and have a crop_bb=False if they only want the face.

What do you think?

Contributor Guide