使用mat格式导入张量到tensorflow

时间:2016-06-03 17:28:42

标签: python matlab tensorflow

我是tensorflow和Python的新手。我在Matlab中有一个图像数据集,张量大小为:96 * 96 * 5000(图像大小,图像数量),我需要将它导入Tensorflow。我查看了Tensorflow教程,该教程说我应该使用TFRecords或CSV格式,但我认为这意味着我以csv格式单独保存每个图像。有没有办法直接将张量导入张量流?

1 个答案:

答案 0 :(得分:7)

正如Olivier在his comment中所说,最简单的解决方案是将数据转换为Numpy数组,并使用TensorFlow的feeding mechanism将数据传递到TensorFlow模型。

SciPy中的scipy.io.loadmat()函数可用于将Matlab .mat文件加载到Python中,作为将Matlab矩阵名称(作为字符串)映射到Numpy数组的字典。