Я хочу написать список изображений в папку после прогнозирования. Как я могу решить это?

Question

predict=[]
for img in glob.glob("test1/*.jpg"):
    n= cv2.imread(img)
    predicted = model.prediction(n)
    predict.append(predicted) 

Как сохранить этот прогнозируемый список изображений в виде изображений в папке? Модель возвращает 224x224 фотографии в оттенках серого.

Answer 1

Используйте imwrite для каждого изображения в списке:

cv2.imwrite(filename, predicted)

Это сохранит изображения с именем filename в текущем рабочем каталоге. Если вы хотите сохранить в другую папку, вместо имени файла укажите путь к файлу:

cv2.imwrite(path_to_file, predicted)