pytorch/vision

VGGFace2 dataset

Open

#1,139 opened on Jul 19, 2019

View on GitHub
 (3 comments) (0 reactions) (0 assignees)Python (6,858 forks)batch import
enhancementhelp wantedmodule: datasets

Repository metrics

Stars
 (15,050 stars)
PR merge metrics
 (Avg merge 12d 8h) (14 merged PRs in 30d)

Description

@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

Research direction
Implement a VGGFace2 dataset class in torchvision.datasets following the pattern of existing datasets like ImageFolder. Allow users to specify targets (ID, gender, age, bounding box) and optionally crop the face using provided bounding boxes.
Tech stack
python
Domain
computer vision
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-2 days
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Mostly clear
Prerequisites
PythonPyTorch
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
50