Tensorflow.Преобразование изображений из PNG в JPEG

Question

Я работаю над CNN, используя Python и Tensorflow. Как я могу конвертировать изображения из PNG в JPEG в этом коде?

У меня есть некоторые идеи по поводу использования цикла for и, возможно, модуля изображения PIL, но у меня нет опыта, поэтому я не хочу t know how to make it. I делать это преобразование любым возможным способом.

ВНИМАНИЕ: Слои не являются большим блоком кода t included because it, но в моем ноутбуке Jupyter они есть.

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.optimizers import RMSprop
import numpy as np
import os

train_dir = os.path.join('D:/nnfp')
train_tank_dir = os.path.join('D:/nnfp/Tank')
train_plane_dir = os.path.join('D:/nnfp/Planes')

train_datagen = ImageDataGenerator(rescale = 1.0/255)
train_generator = train_datagen.flow_from_directory(train_dir,
                                               target_size = (300,300),
                                               batch_size = 128,
                                               class_mode = 'binary')
model.compile(optimizer = RMSprop(lr = 0.001),
         loss = 'binary_crossentropy',
         metrics = ['accuracy'])

model.fit_generator(train_generator,
               steps_per_epoch = 15,
               epochs = 15,
               #validation_data = validation_generator,
               #validation_steps = 2,
               verbose = 1,
               callbacks = [callbacks])

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

Answer 1

Вы можете использовать библиотеку PIL для преобразования изображений из png в jpeg, используя метод convert(), определенный в библиотеке PIL.

Например:

from PIL import Image

image = Image.open("<Image-Name>.png")
image_rgb = im.convert('RGB')
image_rgb.save('<Image-Name-Converted>.jpg')

Надеюсь, это поможет!

Вы можете прочитать документы здесь для деталей: https://pillow.readthedocs.io/en/latest/reference/Image.html#PIL.Image.Image.convert