[tensorflow] celeb_a 데이터 다운로드하기

skyepodium·2021년 5월 23일
0
post-thumbnail
post-custom-banner

1. celeb_a

CelebFaces Attributes Dataset
는 유명인의 이미지가 포함된 데이터 세트입니다.

웃음여부 등등 여러 속성들이 가공되어 있고, 딥러닝 이미지 인식 모델에 사용합니다.

2. tfds

tensorflow_datasets는 데이터 세트 모음입니다. tensorflow에는 들어있지 않고, 다운받아서 사용합니다.

3. 코드

# 1. tfds 임포트
import tensorflow_datasets as tfds

# 2. 사용가능한 데이터 세트를 확인합니다.
tfds.list_builders()
"""
 'abstract_reasoning',
 'accentdb',
 'aeslc',
 'aflw2k3d',
 ... 생략
"""

# 3. celeba 데이터 다운로드
# 약 8분 정도 소요됩니다.
celeb_a = tfds.load('celeb_a')
profile
callmeskye
post-custom-banner

0개의 댓글