Преобразование списка Python в массивный массив для Keras CNN

Question

У меня есть 200000 цветных изображений, каждое из которых (50x50x3) хранится в виде списка в Python.Поскольку все это в списке, я хочу изменить его на массив измерений (200000,50,50,3).Однако использование np.toarray(list) возвращает массив формы (200000,)

Но если я укоротю список (скажем, возьмем только 10000 элементов), а затем выполню np.toarray(shortened_list), он вернет правильный массив формы (10000, 50,50,3)

Я хочу сделать это для подачи изображений в CNN с помощью Keras.

Answer 1

Одно или несколько изображений не имеют правильной формы или не являются изображениями, поэтому их невозможно поместить в массив чисел.

Затем Numpy создает массив объектов.Найдите неправильное изображение и удалите его из списка или измените его размер правильно.

Подсказка:

for i, img in enumerate(shortened_list):

    if not hasattr(img, 'shape'):
        print("wrong image at index " + str(i))
    elif img.shape != (50,50,3):
        print("wrong image at index " + str(i))