python - 从python中的本地目录导入训练数据

Question

我需要将一些训练数据从我的本地目录导入到 python 程序中。目前我正在学习一个教程，在本教程中，数据是在以下代码的帮助下导入的:

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/tmp/data/", one_hot = True) 

但我的问题是我的数据存在于我的本地目录中，所以我不能使用这种方法。非常感谢您的帮助。我的本地目录包含多个文件，我必须通过一个变量将它们全部导入。

Answer 1

在您放置所有数据的本地目录中创建一个文件夹data，然后您可以使用./data 引用它。然后，要访问本地 data 文件夹，应执行以下操作:

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("./data/", one_hot = True) 

也可以通过编程方式获取当前目录，然后如下构造数据目录的路径

import os 

# get the current directory 
current_directory = os.path.join(os.path.dirname(__file__))

# create the path to the data directory within the current directory
data_directory = os.path.join(current_directory, "data")

然后，按如下方式编辑您的代码:

import os 
from tensorflow.examples.tutorials.mnist import input_data

# get the current directory 
current_directory = os.path.join(os.path.dirname(__file__))

# create the path to the data directory within the current directory
data_directory = os.path.join(current_directory, "data")

mnist = input_data.read_data_sets(data_directory, one_hot = True) 

编辑:根据您的评论，您询问的是如何在 tensorflow 中加载您自己的数据:

按照文档中的建议，如果您是 TF 新手，最好从本教程开始:

• tf.contrib.learn Quickstart