时间戳

首页 > 解决方案 > 类型对象“Food101”没有属性“load_data”

python - 类型对象“Food101”没有属性“load_data”

问题描述

我写了这段代码：

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import matplotlib.pyplot as plt
import numpy as np
import tensorflow as tf

import tensorflow_datasets as tfds

food = tfds.image.food101.Food101
(train_images, train_labels), (test_images, test_labels) = food.load_data()

我收到以下错误：

AttributeError
Traceback (most recent call last)
<ipython-input-10-92f141508c7c> in <module>()
            9 import tensorflow_datasets as tfds
         1  0 food = tfds.image.food101.Food101
      ---> 11 (train_images, train_labels), (test_images, test_labels) = 
      food.load_data()
AttributeError: type object 'Food101' has no attribute 'load_data'

我应该怎么做才能克服这个错误？

标签： pythontensorflow

解决方案

从 Tensorflow 页面，您加载数据集的正确方法是：

data, info = tfds.load("desired_dataset", with_info=True)

因此，在您的代码中，您应该加载它：

food, info = tfds.load("food101", with_info=True)

如果这不起作用，请加载它的 builder：

builder = tfds.builder("tfds.image.food101.Food101")
builder.download_and_prepare()
datasets = builder.as_dataset()

参考数据 集

推荐阅读