view_images.py

# view_images.py
# Written By Connor Cozad
#
# Purpose of this file:
# This file is not a necessary part of preparing data or running the neural network. It gives the user a glimpse of
# what the images of hurricanes look like. These are the images that the neural network is trained on in model.py
#
# Outline of this file:
# - Reads numpy files containing hurricane images and the wind speed they are associated with
# - Shows 10 random satellite images and their wind speeds
# - When running the script, closing the current matplotlib window will cause the next one to open


import numpy as np
import matplotlib.pyplot as plt
import random

images = np.load('images.npy')
labels = np.load('labels.npy')

for x in range(10):
    i = random.randint(0, images.shape[0])
    image = np.reshape(images[i], (images[i].shape[0], images[i].shape[1]))
    plt.imshow(image, cmap='binary')
    plt.title('Image #' + str(i) + '   ' + str(labels[i]) + ' knots')
    plt.show()