这与我的问题有关, here 。
我现在有如下更新的代码:
import numpy as np
import _pickle as cPickle
from PIL import Image
import sys,os
pixels = []
labels = []
traindata = []
i = 0
directory = 'C:\\Users\\abc\\Desktop\\Testing\\images'
for root, dirs, files in os.walk(directory):
for file in files:
floc = file
im = Image.open(str(directory) + '\\' + floc)
pix = np.array(im.getdata())
pixels.append(pix)
labels.append(1)
pixels = np.array(pixels)
labels = np.array(labels)
traindata.append(pixels)
traindata.append(labels)
traindata = np.array([traindata[i][i],traindata[1]], dtype=object)
i = i + 1
# do the same for validation and test data
# put all data and labels into 'data' array
cPickle.dump(traindata,open('data.pickle','wb'))
FILE = open("data.pickle", 'rb')
content = cPickle.load(FILE)
print (content)
当只有一张图片时,代码运行良好。但是,当我添加另一个或更多图像时,我得到以下信息:
Traceback (most recent call last):
File "pickle_data.py", line 17, in <module>
pixels.append((pix))
AttributeError: 'numpy.ndarray' object has no attribute 'append'
我该如何解决这个问题?
谢谢。
原文由 Simplicity 发布,翻译遵循 CC BY-SA 4.0 许可协议
到目前为止还可以。但是您想将
pixels
保留为列表,直到您完成迭代。你在另一个问题中有这个缩进。发生了什么? 以前的
迭代,在列表中收集值,然后最后将事物连接到更大的数组中是正确的方法。为了清楚起见,我经常喜欢使用如下符号:
如果名称表明对象的性质,我不太可能得到像你这样的错误。
我不明白你想用
traindata
做什么。我怀疑您是否需要在循环期间构建它。pixels
和labels
有基本信息。那
来自上一个问题。我不确定你是否理解这个答案。
如果在循环外完成只是
labels
是一个一维数组,一堆 1(或者 [0,1,2,3…],如果我的猜测是正确的)。pixels
是一个高维数组。它的形状是什么?停在那儿。将该列表转换为数组是没有意义的。您可以使用
pickle
保存列表。您正在从较早的问题中复制代码,并且格式错误。 cPickle 大量数据