如何从张量流数据集下载特定标签/图像

时间:2019-08-15 09:17:07

标签: python python-3.x tensorflow tensorflow-datasets

我目前正在将coco / 2017数据集用于tf.keras转移学习中

我知道您可以使用以下代码下载完整的数据集:

(raw_train, raw_validation, raw_test), metadata = tfds.load(
                                        'coco/2017',
                                        split=list(splits),
                                        with_info=True,
                                        as_supervised=True)

如果我只想提取“人物”图像和标签,该怎么做。而不是完整的数据集?

0 个答案:

没有答案