Convert JPG to H5, load and Plot

Question

I'm trying to get JPG files into H5, and wrote a script for that. Images are loaded using PIL, and then written into an H5 file, and then for testing purposes I read the H5 file.

The image is loaded, and the types seem correct, do you spot any errors?

import h5py
import numpy as np
from glob import glob # import "submodule"
from PIL import Image
import matplotlib.pyplot as plt

# img source directory
src = "images/raw/cats/*"

# self explanatory
outfile = "images/clean/cat.hdf5" 


listOfImages = glob(src)

# img loaded as JpegImageFile Object
oneImage = Image.open(listOfImages[0])

# open file
f = h5py.File(outfile, "w")
data  =  np.array(oneImage) #this is an array 200x200x3

# save dataset (ndarray but with more features)
dset = f.create_dataset("images", data=data) # array 200x200x3

# Check if files can be read from the h5py file
with h5py.File(outfile, "r") as f:
    image = f['images']
    print(image.shape, image.dtype)
    plt.imshow(image)

# outout:::: (200, 200, 3) uint8, but nothing is plt.

Your code is working properly – ashraful16 Jan 11 '21 at 07:59 — ashraful16, Jan 11 '21 at 07:59
not in my PC may have to do with Wayland @ashraful16? – Minsky Jan 11 '21 at 08:00 — Minsky, Jan 11 '21 at 08:00

score 2 · Accepted Answer · answered Jan 11 '21 at 07:56

2

just do plt.show() at the end of your code if you are running in a ipython notebook, it should work

answered Jan 11 '21 at 07:56

Chandan Malla

481
5
14

I'm running locally. That was it. What a stupid question. Thanks for the tip though. – Minsky Jan 11 '21 at 07:58
@Chandan Malla, can you react on my last question please? – TangerCity Jan 11 '21 at 10:24
@TangerCity I have shared a working code buddy, please check :) – Chandan Malla Jan 11 '21 at 10:56

Convert JPG to H5, load and Plot

1 Answers1