Img_to_array возвращает пустое значение

Question

У меня есть скрипт увеличения изображения, который я использую для выполнения некоторых операций с изображениями. Но функция img_to_array в Keras возвращает пустое значение при вызове.

Я создал скрипт для дополнения, но он столкнулся с некоторыми проблемами.

from numpy import expand_dims
from keras.preprocessing.image import load_img
from keras.preprocessing.image import img_to_array
from keras.preprocessing.image import ImageDataGenerator
from matplotlib import pyplot
from glob import glob
import glob
import cv2
import os


def get_images(m):
    count = 0
    img_list = []
    images_path = glob.glob(m)

    out = "C:\\Python35\\augmented_images\\horizontal_shift\\"

    if not os.path.exists(out):
        os.makedirs(out, exist_ok=True)

    for folder in images_path:
        for f in glob.glob(folder+"/*.jpg"):
            img_list.append(f)
            print(img_list)

        for i in range(len(img_list)):
            img_base = os.path.basename(img_list[i])
            img_name = os.path.splitext(img_base)[0]
            img = load_img(img_list[i])
            print(img)
            data = img_to_array(img)
            samples = expand_dims(data, 0)
            datagen = ImageDataGenerator(width_shift_range=[-200, 200])
            it = datagen.flow(samples, batch_size=1)
            for i in range(9):
                batch = it.next()
                cv2.imwrite(out + img_name+"_%d.jpg" % count, batch)
                count += 1


folders = ("C:\\Python35\\augment_img_data\\*")
get_images(folders)

Почему img_to_array возвращается пустым. Что нужно изменить, чтобы можно было выполнить операцию увеличения

Answer 1

Попробуй это. Я не думаю, что вам придется расширять размеры.

img = image.load_img(file, target_size=(224, 224))
img = image.img_to_array(img)
#x = resnet50.preprocess_input(img)
x = np.array([img])
feature = model.predict(x)

или это

img = load_img(img_list[i])
print(img)
data = img_to_array(img)
samples = expand_dims(data, axis= 1)