将列表转换为np数组

Question

我正在建立一个图像分类模型，但首先我需要读取用于训练的图像，我有105392张图像，因此我读取了我用cv2.imread_灰度下载的本地图像，并将图像大小调整为100x100

train_images = []
img_array = cv2.imread(os.path.join(path,str(train_data.filename[idx])), cv2.IMREAD_GRAYSCALE)
new_array = cv2.resize(img_array,(img_size,img_size))
train_images.append(new_array) 

然后，我将图像数据添加到一个列表中，然后将它们转换为nparray，并尝试将它们展平到一维，以便保存到文本文件中以便于处理

ndarray = np.array(train_images)
ndarray2 = np.ndarray(ndarray)
flattenarray = ndarray2.flatten(order='C')
flattenarray.tofile('train_images_bitmap.txt')

但是我得到了这个错误

Traceback (most recent call last):
File "comp2.py", line 36, in <module>
ndarray2 = np.ndarray(ndarray)
ValueError: maximum supported dimension for an ndarray is 32, found 105392

感谢您的帮助

Answer 1

您的ndarray阵列的形状为（105392100100）以展平此阵列

flatten_array = ndarray.reshape(1,105392*100*100)