HOW TO PLOT IMAGES FORM DATALOADER

 # get some random training images

dataiter = iter(train_loader)

images, labels = dataiter.next()

arthopod_types = {0: 'Coleoptera', 1: 'Diptera', 2: 'Hymenoptera', 3: 'Lepidoptera'}

# Viewing data examples used for training

fig, axis = plt.subplots(3, 5, figsize=(15, 10))

for i, ax in enumerate(axis.flat):

    with torch.no_grad():

        image, label = images[i], labels[i]

        ax.imshow(img_display(image)) # add image

        ax.set(title = f"{arthopod_types[label.item()]}") # add label