pytorch/vision

VGGFace2 dataset

开放

#1,139 创建于 2019年7月19日

 (3 条评论) (0 个反应) (0 位负责人)Python (6,858 个派生)batch import
enhancementhelp wantedmodule: datasets

仓库指标

星标
 (15,050 个星标)
PR 合并指标
 (平均合并 12天 8小时) (30 天内合并 14 个 PR)

描述

@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?

贡献者指南