CODEWITHSUNDEEP
neural-networkartificial-intelligenceclassificationimage-recognitionmnist
Riley Fitzpatrick
Riley Fitzpatrick

Reputation: 919

How do you "import" image data for MNIST?

So I've been using Tensorflow's tutorials for neural networks. I completed the "basic classification" that is essentially just MNIST and have been working on making my own custom variation as a little thought experiment. Everything is pretty self explanatory except putting the datasets into a usable form as the tutorial uses a premade dataset and looks like it cuts some corners. All I would like to know is how to put a colored photo into a usable piece of data. I assume that will just be a 1D array. As a side question, will a neural network lose any effectiveness if a 2d photo is stored in a 1d array if its not a CNN.

Upvotes: 1

Views: 1607

Answers (1)

Shubham Panchal
Shubham Panchal

Reputation: 4289

Datasets included in Keras are premade and usually preprocessed so that beginner could easily try a hand on them. For using your own images, like for a cat-dog image classification problem, you can place the images in two separate directories, for example, in images/cats and images/dogs. Now, we parse each and every image in these directories,

import os
from PIL import Image

master_dir = 'images'
img_dirs = os.listdir( master_dir ) 
for img_dir in img_dirs:
    img_names = os.listdir( os.path.join( master_dir , img_dir ) ) 
    for name in img_names:
        img_path = os.path.join( master_dir , img_dir , name ) 
        image = Image.open( img_path ).resize( ( 64 , 64 ) ).convert( 'L' ) 
        # Store this image in an array with its corresponding label

Here. the image will be an array of shape (64, 64 ) which indicates that the image is grayscale. Besides .convert( 'L' ) in the code, we can use .convert( 'RGB' ) to have an image of shape (64,64,3) RGB image.

Now,

  • Collect all the images and labels in a Python list.
  • Convert the lists to NumPy arrays.
  • Store the NumPy arrays in a .npy file using the np.save() method.
  • In the file which trains the model, load these files using np.load() method and feed them to the model.

Upvotes: 5

Related Questions