Tensorflow。将图像从PNG转换为JPEG

时间:2019-06-22 19:44:49

标签: python python-3.x tensorflow keras conv-neural-network

我正在使用Python和Tensorflow开发CNN。如何使用此代码将图像从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])

如果可能的话,我想通过在现有代码中添加一些代码来转换所有数据。

1 个答案:

答案 0 :(得分:0)

您可以使用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