如何从tensorflowdatasets下载特定标签/图像

2024-03-29 13:51:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我目前正在使用coco/2017数据集,以便在克拉斯特遣部队转移学习

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

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

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


Tags: 数据代码testtruerawloadtrainlist