首页 > 解决方案 > 如何修改此 Google Colab 以接受我自己的压缩文件,并在我自己的图像上运行模型?

问题描述

这是 google colab 笔记本的链接: https ://colab.research.google.com/drive/1CwknSSXVtb-UBAw7Lm9slu_DgCO5M5wW

我无法将我的数据集链接到代码并在其上运行模型。我正在尝试调整此笔记本https://colab.research.google.com/github/tensorflow/hub/blob/master/examples/colab/tf2_image_retraining.ipynb以在我自己的数据集上运行它,但我收到此错误:

<ipython-input-14-542fb7cdb61e> in <module>()
      6     **datagen_kwargs)
      7 valid_generator = valid_datagen.flow_from_directory(
----> 8     data_dir, subset="validation", shuffle=False, **dataflow_kwargs)
      9 
     10 do_data_augmentation = False #@param {type:"boolean"}

1 frames
/tensorflow-2.1.0/python3.6/keras_preprocessing/image/directory_iterator.py in __init__(self, directory, image_data_generator, target_size, color_mode, classes, class_mode, batch_size, shuffle, seed, data_format, save_to_dir, save_prefix, save_format, follow_links, subset, interpolation, dtype)
    104         if not classes:
    105             classes = []
--> 106             for subdir in sorted(os.listdir(directory)):
    107                 if os.path.isdir(os.path.join(directory, subdir)):
    108                     classes.append(subdir)

NotADirectoryError: [Errno 20] Not a directory: '/root/.keras/datasets/training_photos'

如果有人能给我一些关于如何调整原始谷歌 colab 笔记本以在我自己的数据集(存储在我的计算机文件中)上运行并获得预测的建议,我将非常感激,因为我是新手。

标签: pythonfiletensorflowkerasgoogle-colaboratory

解决方案


推荐阅读