在numpy数组中堆叠多个RGB图像以实现CNN

时间:2019-05-18 12:02:35

标签: python-3.x image deep-learning google-colaboratory

我要从当前目录中读取1000张RGB图像,并将其存储为(1000,3,32,32)形状的numpy数组,以便在CNN中使用。

由于这个原因,我阅读了一个示例图像,将其大小调整为32 *32。然后将其附加到我使用零(1000、3、32、32)创建的数组'a'中。但是我遇到了一个错误,即“'numpy.ndarray'对象没有属性'append'”。如何解决?如果需要任何其他方法,我也愿意接受。

import matplotlib.pyplot as plt

import numpy as np

reshapedimage =cv2.resize(cv2.imread("0 (1).png", 1), (32, 32))


a = np.zeros((1000,3,32,32))

a.append(reshapedimage)

1 个答案:

答案 0 :(得分:0)

我想你是说这个意思

import numpy as np

# Create dummy image-like thing
w, h = 32, 32
im=np.arange(h*w*3).reshape((3,h,w))                                                       

# Create empty list
stack=[]                                                                                   

# Append the image to the stack 5 times
stack.append(im)
stack.append(im)
stack.append(im)
stack.append(im)
stack.append(im)                                                                          

# Make Numpy array and check size                                                        
v = np.array(stack)                                                                        
print(v.shape)                                                                                    

输出

(5, 3, 32, 32)