首页 > 解决方案 > 谷歌协作

问题描述

我正在谷歌 colab 上开发一个 CNN 程序。
我正在尝试运行,ImageDataGenerator.flow_from_directory。

我想在这个函数中包含数据集目录。我创建了一个谷歌驱动器文件夹“App”,并在该文件夹中创建了“数据集”文件夹,其中包含了用于训练和测试的数据集。

问题是我无法在 colab 中为此命令包含此驱动器文件夹。

任何人都有在 colab 上运行 ImageDataGenerator.flow_from_directory 的经验?如何从谷歌驱动器添加数据集?

标签: machine-learninggoogle-drive-api

解决方案


根据API,这应该允许您将文件上传到 colaboratory venv:

from google.colab import files
uploaded = files.upload()   
for fn in uploaded.keys():
  print('User uploaded file "{name}" with length {length} bytes'.format(
      name=fn, length=len(uploaded[fn])))

推荐阅读