Skip to content

Instantly share code, notes, and snippets.

@uchidama
Last active March 24, 2023 02:35
Show Gist options
  • Save uchidama/a75bd3b949b89f64ccc77a9a5de445a9 to your computer and use it in GitHub Desktop.
Save uchidama/a75bd3b949b89f64ccc77a9a5de445a9 to your computer and use it in GitHub Desktop.
Convert fashion_mnist data on keras datasets to jpeg images.
import keras
from keras.datasets import fashion_mnist
import numpy as np
from PIL import Image, ImageOps
import os
def save_image(filename, data_array):
im = Image.fromarray(data_array.astype('uint8'))
im_invert = ImageOps.invert(im)
im_invert.save(filename)
# Load Fashion-MNIST Data
(x_train, y_train), (x_test, y_test) = fashion_mnist.load_data()
DIR_NAME = "JPEGImages"
if os.path.exists(DIR_NAME) == False:
os.mkdir(DIR_NAME)
# Save Images
i = 0
for li in [x_train, x_test]:
print("[---------------------------------------------------------------]")
for x in li:
filename = "{0}/{1:05d}.jpg".format(DIR_NAME,i)
print(filename)
save_image(filename, x)
i += 1
@EvanMarie
Copy link

You are a god. No...seriously...YOU ARE!!!!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment