Укладка нескольких изображений RGB в массиве для реализации CNN - PullRequest
1 голос
/ 18 мая 2019

У меня есть 1000 изображений RGB, которые я хочу прочитать из текущего каталога и сохранить их в массивном массиве в форме (1000,3,32,32) для использования в CNN.

ДляПо этой причине я прочитал пример изображения, изменил его размер до 32 * 32. Затем добавил его в массив 'a', который я создал, используя нули для формы (1000,3,32,32).Но я получаю сообщение об ошибке «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 голосов
/ 18 мая 2019

Я думаю, вы имеете в виду это:

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)
...