我已经完成了.pkl文件的编写,但是尝试加载该文件时出现错误。
f = open('train.pkl')
X, Y, X_test, Y_test = cPickle.load(f)
要么
X, Y ,X_test,Y_test=pickle.load(open("train.pkl","rb"))
两次尝试均失败。编写.pkl文件的代码如下。
在循环内处理图像和标签:
img_ndarray = numpy.asarray(img, dtype='float64')
img_raw[i] = numpy.ndarray.flatten(img_ndarray)
img_label[i]=numpy.asarray(name,dtype='float64')
i=i+1
在循环外部写入pickle文件:
write_file = open('train.pkl', 'wb')
cPickle.dump([[img_raw[0:4],img_label[0:4]], [img_raw[5:7], img_label[5:7]]],write_file, -1)
write_file.close()
感谢您的投入。
最佳答案
错误消息应该很明显,您已经腌制了一个包含2个元素的列表,因此无法将其分解为4个名称。考虑(构成一些数据):
In [35]: img_raw, img_label = range(10), 'abcdefghhijlmnop'
In [36]: [[img_raw[0:4],img_label[0:4]], [img_raw[5:7], img_label[5:7]]]
Out[36]: [[range(0, 4), 'abcd'], [range(5, 7), 'fg']]
In [37]: len([[img_raw[0:4],img_label[0:4]], [img_raw[5:7], img_label[5:7]]])
Out[37]: 2
但是,这很容易解决,因为每个子列表中也都有2项:
In [38]: (X, Y) , (X_test, Y_test) = [[img_raw[0:4],img_label[0:4]], [img_raw[5:7], img_label[5:7]]]
而且有效!
In [41]: X
Out[41]: range(0, 4)
In [42]: Y
Out[42]: 'abcd'
In [43]: X_test
Out[43]: range(5, 7)
In [44]: Y_test
Out[44]: 'fg'
通常,您解压缩的内容必须具有相同的形状:
In [45]: data = [[[1, 2], [3, 4, 5]],[6, [7, 8]]]
In [46]: (((a, b), (c, d, e)), (f, (g, h))) = data
In [47]: a,b,c,d,e,f,g,h
Out[47]: (1, 2, 3, 4, 5, 6, 7, 8)
关于python - pickle.load:需要两个以上的值才能解压缩,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/44037770/