TorchvisionのCelebAデータセットについて

概要

218*178のカラーの顔画像202599枚

引数

torchvision.datasets.CelebA(root: str, split: str = 'train', target_type: Union[List[str], str] = 'attr', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)

• root:ダウンロードデータを入れるディレクトリ
• split:[train:162770 test:19962 valid:19867 all:202599]から選択
• target_type:[attr, identity, bbox, landmarks]
  attrは40種類ある特徴(0:5_o_Clock_Shadow(髭の剃り残し),1:Arched_Eyebrows(三日月眉),...39:Young(若さ))に該当しているか -1 or 1
  identityは人のラベル(同じ人物は同じ数字)
  bboxはbounding box(x,y,width,height) 顔の位置
  landmarksは　顔のパーツ(目や鼻)の位置(lefteye_x, lefteye_y, righteye_x, righteye_y, nose_x, nose_y, leftmouth_x, leftmouth_y, rightmouth_x, rightmouth_y)
• transform:画像データに対する加工
• target_transform:target_typeに対する加工
• download:データをrootにダウンロードするか

参考
http://cedro3.com/ai/celeba-dataset-attribute/