我试图在我自己的数据集中使用书籍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中使用我自己的数据集?
谢谢。