Chapter 6: Decoding Images

Activity 17: Predict if an Image Is of a Cat or a Dog

Solution:

1. If you look at the name of the images in the dataset, you will find that the images of dogs start with dog followed by '.' and then a number, for example – "dog.123.jpg". Similarly, the images of cats start with cat. So, let's create a function to get the label from the name of the file:

   def get_label(file):

       class_label = file.split('.')[0]

       if class_label == 'dog': label_vector = [1,0]

       elif class_label == 'cat': label_vector = [0,1]

       return label_vector

   Then, create a function to read, resize, and preprocess the images:

   import os

   import numpy as np

   from PIL import Image

   from tqdm import tqdm

   from random import shuffle

    

   SIZE = 50

    

   def get_data():

       data = []

       files = os.listdir(PATH)

    

     &...