Tensorflow实战05-猫狗分类器

淡淡的烟草味﹌ 2023-03-02 05:10 41阅读 0赞

> ##### 注：学习完深度学习的基本知识后，本练习可帮助大家逐步从理论转向实战，如果对深度学习还不太了解，前先学习深度学习的基本知识以及原理 #####

###### 在这个练习中，你将训练一个CNN关于猫和狗的完整数据集 ######

###### 这将需要您做大量的数据预处理，因为 ######

###### 数据集没有为您划分为培训和验证 ######

###### 此代码块具有所有必需的输入 ######

#### 在下方有空的地方填写代码并检验： ####

import os
    import zipfile
    import random
    import tensorflow as tf
    from tensorflow.keras.optimizers import RMSprop
    from tensorflow.keras.preprocessing.image import ImageDataGenerator
    from shutil import copyfile

# This code block downloads the full Cats-v-Dogs dataset and stores it as 
    # cats-and-dogs.zip. It then unzips it to /tmp
    # which will create a tmp/PetImages directory containing subdirectories
    # called 'Cat' and 'Dog' (that's how the original researchers structured it)
    # If the URL doesn't work, 
    # .   visit https://www.microsoft.com/en-us/download/confirmation.aspx?id=54765
    # And right click on the 'Download Manually' link to get a new URL
    
    !wget --no-check-certificate \
        "https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_3367a.zip" \
        -O "/tmp/cats-and-dogs.zip"
    
    local_zip = '/tmp/cats-and-dogs.zip'
    zip_ref = zipfile.ZipFile(local_zip, 'r')
    zip_ref.extractall('/tmp')
    zip_ref.close()

print(len(os.listdir('/tmp/PetImages/Cat/')))
    print(len(os.listdir('/tmp/PetImages/Dog/')))
    
    # Expected Output:
    # 12501
    # 12501

# Use os.mkdir to create your directories
    # You will need a directory for cats-v-dogs, and subdirectories for training
    # and testing. These in turn will need subdirectories for 'cats' and 'dogs'
    try:
        #YOUR CODE GOES HERE
    except OSError:
        pass

# Write a python function called split_data which takes
    # a SOURCE directory containing the files
    # a TRAINING directory that a portion of the files will be copied to
    # a TESTING directory that a portion of the files will be copie to
    # a SPLIT SIZE to determine the portion
    # The files should also be randomized, so that the training set is a random
    # X% of the files, and the test set is the remaining files
    # SO, for example, if SOURCE is PetImages/Cat, and SPLIT SIZE is .9
    # Then 90% of the images in PetImages/Cat will be copied to the TRAINING dir
    # and 10% of the images will be copied to the TESTING dir
    # Also -- All images should be checked, and if they have a zero file length,
    # they will not be copied over
    #
    # os.listdir(DIRECTORY) gives you a listing of the contents of that directory
    # os.path.getsize(PATH) gives you the size of the file
    # copyfile(source, destination) copies a file from source to destination
    # random.sample(list, len(list)) shuffles a list
    def split_data(SOURCE, TRAINING, TESTING, SPLIT_SIZE):
    # YOUR CODE STARTS HERE
    # YOUR CODE ENDS HERE
    
    
    CAT_SOURCE_DIR = "/tmp/PetImages/Cat/"
    TRAINING_CATS_DIR = "/tmp/cats-v-dogs/training/cats/"
    TESTING_CATS_DIR = "/tmp/cats-v-dogs/testing/cats/"
    DOG_SOURCE_DIR = "/tmp/PetImages/Dog/"
    TRAINING_DOGS_DIR = "/tmp/cats-v-dogs/training/dogs/"
    TESTING_DOGS_DIR = "/tmp/cats-v-dogs/testing/dogs/"
    
    split_size = .9
    split_data(CAT_SOURCE_DIR, TRAINING_CATS_DIR, TESTING_CATS_DIR, split_size)
    split_data(DOG_SOURCE_DIR, TRAINING_DOGS_DIR, TESTING_DOGS_DIR, split_size)
    
    # Expected output
    # 666.jpg is zero length, so ignoring
    # 11702.jpg is zero length, so ignoring

print(len(os.listdir('/tmp/cats-v-dogs/training/cats/')))
    print(len(os.listdir('/tmp/cats-v-dogs/training/dogs/')))
    print(len(os.listdir('/tmp/cats-v-dogs/testing/cats/')))
    print(len(os.listdir('/tmp/cats-v-dogs/testing/dogs/')))
    
    # Expected output:
    # 11250
    # 11250
    # 1250
    # 1250

# DEFINE A KERAS MODEL TO CLASSIFY CATS V DOGS
    # USE AT LEAST 3 CONVOLUTION LAYERS
    model = tf.keras.models.Sequential([
    # YOUR CODE HERE
    ])
    
    model.compile(optimizer=RMSprop(lr=0.001), loss='binary_crossentropy', metrics=['acc'])

TRAINING_DIR = #YOUR CODE HERE
    train_datagen = #YOUR CODE HERE
    train_generator = #YOUR CODE HERE
    
    VALIDATION_DIR = #YOUR CODE HERE
    validation_datagen = #YOUR CODE HERE
    validation_generator = #YOUR CODE HERE
    
    
    
    # Expected Output:
    # Found 22498 images belonging to 2 classes.
    # Found 2500 images belonging to 2 classes.

history = model.fit_generator(train_generator,
                                  epochs=15,
                                  verbose=1,
                                  validation_data=validation_generator)
    
    # The expectation here is that the model will train, and that accuracy will be > 95% on both training and validation
    # i.e. acc:A1 and val_acc:A2 will be visible, and both A1 and A2 will be > .9

# PLOT LOSS AND ACCURACY
    %matplotlib inline
    
    import matplotlib.image  as mpimg
    import matplotlib.pyplot as plt
    
    #-----------------------------------------------------------
    # Retrieve a list of list results on training and test data
    # sets for each training epoch
    #-----------------------------------------------------------
    acc=history.history['acc']
    val_acc=history.history['val_acc']
    loss=history.history['loss']
    val_loss=history.history['val_loss']
    
    epochs=range(len(acc)) # Get number of epochs
    
    #------------------------------------------------
    # Plot training and validation accuracy per epoch
    #------------------------------------------------
    plt.plot(epochs, acc, 'r', "Training Accuracy")
    plt.plot(epochs, val_acc, 'b', "Validation Accuracy")
    plt.title('Training and validation accuracy')
    plt.figure()
    
    #------------------------------------------------
    # Plot training and validation loss per epoch
    #------------------------------------------------
    plt.plot(epochs, loss, 'r', "Training Loss")
    plt.plot(epochs, val_loss, 'b', "Validation Loss")
    
    
    plt.title('Training and validation loss')
    
    # Desired output. Charts with training and validation metrics. No crash :)

# Here's a codeblock just for fun. You should be able to upload an image here 
    # and have it classified without crashing
    
    import numpy as np
    from google.colab import files
    from keras.preprocessing import image
    
    uploaded = files.upload()
    
    for fn in uploaded.keys():
     
      # predicting images
      path = '/content/' + fn
      img = image.load_img(path, target_size=(# YOUR CODE HERE))
      x = image.img_to_array(img)
      x = np.expand_dims(x, axis=0)
    
      images = np.vstack([x])
      classes = model.predict(images, batch_size=10)
      print(classes[0])
      if classes[0]>0.5:
        print(fn + " is a dog")
      else:
        print(fn + " is a cat")

#### 参考答案： ####

# If the URL doesn't work, visit https://www.microsoft.com/en-us/download/confirmation.aspx?id=54765
    # And right click on the 'Download Manually' link to get a new URL to the dataset
    
    # Note: This is a very large dataset and will take time to download
    
    !wget --no-check-certificate \
        "https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_3367a.zip" \
        -O "/tmp/cats-and-dogs.zip"
    
    local_zip = '/tmp/cats-and-dogs.zip'
    zip_ref   = zipfile.ZipFile(local_zip, 'r')
    zip_ref.extractall('/tmp')
    zip_ref.close()

print(len(os.listdir('/tmp/PetImages/Cat/')))
    print(len(os.listdir('/tmp/PetImages/Dog/')))
    
    # Expected Output:
    # 12501
    # 12501

try:
        os.mkdir('/tmp/cats-v-dogs')
        os.mkdir('/tmp/cats-v-dogs/training')
        os.mkdir('/tmp/cats-v-dogs/testing')
        os.mkdir('/tmp/cats-v-dogs/training/cats')
        os.mkdir('/tmp/cats-v-dogs/training/dogs')
        os.mkdir('/tmp/cats-v-dogs/testing/cats')
        os.mkdir('/tmp/cats-v-dogs/testing/dogs')
    except OSError:
        pass

def split_data(SOURCE, TRAINING, TESTING, SPLIT_SIZE):
        files = []
        for filename in os.listdir(SOURCE):
            file = SOURCE + filename
            if os.path.getsize(file) > 0:
                files.append(filename)
            else:
                print(filename + " is zero length, so ignoring.")
    
        training_length = int(len(files) * SPLIT_SIZE)
        testing_length = int(len(files) - training_length)
        shuffled_set = random.sample(files, len(files))
        training_set = shuffled_set[0:training_length]
        testing_set = shuffled_set[-testing_length:]
    
        for filename in training_set:
            this_file = SOURCE + filename
            destination = TRAINING + filename
            copyfile(this_file, destination)
    
        for filename in testing_set:
            this_file = SOURCE + filename
            destination = TESTING + filename
            copyfile(this_file, destination)
    
    
    CAT_SOURCE_DIR = "/tmp/PetImages/Cat/"
    TRAINING_CATS_DIR = "/tmp/cats-v-dogs/training/cats/"
    TESTING_CATS_DIR = "/tmp/cats-v-dogs/testing/cats/"
    DOG_SOURCE_DIR = "/tmp/PetImages/Dog/"
    TRAINING_DOGS_DIR = "/tmp/cats-v-dogs/training/dogs/"
    TESTING_DOGS_DIR = "/tmp/cats-v-dogs/testing/dogs/"
    
    split_size = .9
    split_data(CAT_SOURCE_DIR, TRAINING_CATS_DIR, TESTING_CATS_DIR, split_size)
    split_data(DOG_SOURCE_DIR, TRAINING_DOGS_DIR, TESTING_DOGS_DIR, split_size)
    
    # Expected output
    # 666.jpg is zero length, so ignoring
    # 11702.jpg is zero length, so ignoring

model = tf.keras.models.Sequential([
        tf.keras.layers.Conv2D(16, (3, 3), activation='relu', input_shape=(150, 150, 3)),
        tf.keras.layers.MaxPooling2D(2, 2),
        tf.keras.layers.Conv2D(32, (3, 3), activation='relu'),
        tf.keras.layers.MaxPooling2D(2, 2),
        tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
        tf.keras.layers.MaxPooling2D(2, 2),
        tf.keras.layers.Flatten(),
        tf.keras.layers.Dense(512, activation='relu'),
        tf.keras.layers.Dense(1, activation='sigmoid')
    ])
    
    model.compile(optimizer=RMSprop(lr=0.001), loss='binary_crossentropy', metrics=['acc'])

TRAINING_DIR = "/tmp/cats-v-dogs/training/"
    train_datagen = ImageDataGenerator(rescale=1.0/255.)
    train_generator = train_datagen.flow_from_directory(TRAINING_DIR,
                                                        batch_size=100,
                                                        class_mode='binary',
                                                        target_size=(150, 150))
    
    VALIDATION_DIR = "/tmp/cats-v-dogs/testing/"
    validation_datagen = ImageDataGenerator(rescale=1.0/255.)
    validation_generator = validation_datagen.flow_from_directory(VALIDATION_DIR,
                                                                  batch_size=100,
                                                                  class_mode='binary',
                                                                  target_size=(150, 150))
    
    # Expected Output:
    # Found 22498 images belonging to 2 classes.
    # Found 2500 images belonging to 2 classes.

# Note that this may take some time.
    history = model.fit_generator(train_generator,
                                  epochs=50,
                                  verbose=1,
                                  validation_data=validation_generator)

%matplotlib inline
    
    import matplotlib.image  as mpimg
    import matplotlib.pyplot as plt
    
    #-----------------------------------------------------------
    # Retrieve a list of list results on training and test data
    # sets for each training epoch
    #-----------------------------------------------------------
    acc=history.history['acc']
    val_acc=history.history['val_acc']
    loss=history.history['loss']
    val_loss=history.history['val_loss']
    
    epochs=range(len(acc)) # Get number of epochs
    
    #------------------------------------------------
    # Plot training and validation accuracy per epoch
    #------------------------------------------------
    plt.plot(epochs, acc, 'r', "Training Accuracy")
    plt.plot(epochs, val_acc, 'b', "Validation Accuracy")
    plt.title('Training and validation accuracy')
    plt.figure()
    
    #------------------------------------------------
    # Plot training and validation loss per epoch
    #------------------------------------------------
    plt.plot(epochs, loss, 'r', "Training Loss")
    plt.plot(epochs, val_loss, 'b', "Validation Loss")
    plt.figure()
    
    
    # Desired output. Charts with training and validation metrics. No crash :)

# Here's a codeblock just for fun. You should be able to upload an image here 
    # and have it classified without crashing
    import numpy as np
    from google.colab import files
    from keras.preprocessing import image
    
    uploaded = files.upload()
    
    for fn in uploaded.keys():
     
      # predicting images
      path = '/content/' + fn
      img = image.load_img(path, target_size=(150, 150))
      x = image.img_to_array(img)
      x = np.expand_dims(x, axis=0)
    
      images = np.vstack([x])
      classes = model.predict(images, batch_size=10)
      print(classes[0])
      if classes[0]>0.5:
        print(fn + " is a dog")
      else:
        print(fn + " is a cat")