首先来看一下,tensorflow dataset都集成了哪些数据集
>>> import tensorflow_datasets as tfds
>>> print(tfds.list_builders())
['bair_robot_pushing_small', 'cats_vs_dogs', 'celeb_a', 'celeb_a_hq', 'cifar10', 'cifar100', 'coco2014', 'diabetic_retinopathy_detection', 'dummy_dataset_shared_generator', 'dummy_mnist', 'fashion_mnist', 'image_label_folder', 'imagenet2012', 'imdb_reviews', 'lm1b', 'lsun', 'mnist', 'moving_mnist', 'nsynth', 'omniglot', 'open_images_v4', 'quickdraw_bitmap', 'squad', 'starcraft_video', 'svhn_cropped', 'tf_flowers', 'wmt_translate_ende', 'wmt_translate_enfr']
一、音频类
nsynth
二、图像类
cats_vs_dogs
celeb_a:CelebFaces Attributes Dataset(CelebA)是一个大型人脸属性数据集,拥有超过200K名人图像,每个图像都有40个属性注释。https://blog.csdn.net/weixin_41036461/article/details/83109444
celeb_a_hq:高质量人脸图像
cifar10:是由AlexKrizhevsky、VinodNair与GeoffreyHinton收集的一个图像识别的数据集,有10类,飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船..https://blog.csdn.net/bryant_meng/article/details/81077196
cifar100
coco2014
colorectal_histology
colorectal_histology_large
diabetic_retinopathy_detection
fashion_mnist
image_label_folder
imagenet2012
lsun:场景分类的10个场景类别。LSUN 是一个场景理解图像数据集,主要包含了卧室、固房、客厅、教室等场景图像。
mnist:手写数字数据集
omniglot
open_images_v4
quickdraw_bitmap
svhn_cropped
tf_flowers
三、结构化数据集
titanic
四、文本类
imdb_reviews
lm1b
squad
五、翻译类
wmt_translate_ende
wmt_translate_enfr
六、视频类
bair_robot_pushing_small
moving_mnist
starcraft_video