added starter code for tinkering with simple images

ben519 · Jul 21, 2021 · 5800b3a · 5800b3a
1 parent 506c950
commit 5800b3a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lectures/3.1 simple_images.py b/lectures/3.1 simple_images.py
@@ -0,0 +1,13 @@
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+
+# Load the simple images data
+train = pd.read_csv('https://raw.githubusercontent.com/ben519/nnets-for-your-dog/master/data/simple_images_train.csv')
+train.head()
+
+# Inspect
+train.shape
+
+# Plot the first simple image
+plt.imshow(train.drop(columns='label').to_numpy()[0].reshape(2,2), cmap='gray', vmin=0, vmax=255)