如何使我的数据集适应Theano?

时间:2017-03-28 23:28:00

标签: python numpy theano

我试图在我自己的数据集中使用书籍http://neuralnetworksanddeeplearning.com/中的代码。

我了解到我需要将数据放在一个numpy数组中,然后在函数中使用。为此,我使用以下代码:

import pickle
def generateDataset(filesDir):
   x = np.asarray([np.asarray(Image.open(fname)) for fname in filesDir])
   with open('dataset.pickle', 'wb') as handle:
      pickle.dump( x, handle, protocol=2) 

但是当我尝试使用cPickle加载.pkl文件并在函数中使用时,我收到了这条消息:

The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我搜索并尝试了很多东西,但没有成功。有人知道如何在这个项目或Theano中使用我自己的数据集?

谢谢。

0 个答案:

没有答案