如何在colab中读取多个图像文件(ppm文件)(未定义skimage)

时间:2020-01-24 12:10:36

标签: python tensorflow google-colaboratory scikit-image

我是Tensorflow和colab的完全新手。我目前正在研究初学者教程。我在谷歌的合作实验室工作。

我正在尝试运行以下代码,以便从包含多个子目录的google驱动器目录中读取许多图像文件。

def load_data(data_directory):
    directories = [d for d in os.listdir(data_directory) 
                   if os.path.isdir(os.path.join(data_directory, d))]
    labels = []
    images = []
    for d in directories:
        label_directory = os.path.join(data_directory, d)
        file_names = [os.path.join(label_directory, f) 
                      for f in os.listdir(label_directory) 
                      if f.endswith(".ppm")]
        for f in file_names:
            images.append(skimage.data.imread(f))
            labels.append(int(d))
    return images, labels

但是出现以下错误:

NameError: name 'skimage' is not defined

我不确定是否是因为未加载正确的库。 任何帮助将不胜感激。

0 个答案:

没有答案