1 files changed, 34 insertions, 0 deletions
diff --git a/site/content/pages/datasets/lfw/assets/fetch_lfw_people.py b/site/content/pages/datasets/lfw/assets/fetch_lfw_people.py
new file mode 100644
index 00000000..639883a6
--- /dev/null
+++ b/site/content/pages/datasets/lfw/assets/fetch_lfw_people.py
@@ -0,0 +1,34 @@
+#!/usr/bin/python
+
+# ------------------------------------------------------------
+#
+# Script to generate montage of LFW faces used in scikit-learn
+#
+# ------------------------------------------------------------
+
+import numpy as np
+from sklearn.datasets import fetch_lfw_people
+import imageio
+import imutils
+
+# download LFW dataset (first run takes a while)
+lfw_people = fetch_lfw_people(min_faces_per_person=1, resize=1, color=True, funneled=False)
+
+# introspect dataset
+n_samples, h, w, c = lfw_people.images.shape
+print(f'{n_samples:,} images at {w}x{h} pixels')
+cols, rows = (176, 76)
+n_ims = cols * rows
+
+# build montages
+im_scale = 0.5
+ims = lfw_people.images[:n_ims]
+montages = imutils.build_montages(ims, (int(w * im_scale,   int(h * im_scale)), (cols, rows))
+montage = montages[0]
+
+# save full montage image
+imageio.imwrite('lfw_montage_full.png', montage)
+
+# make a smaller version
+montage = imutils.resize(montage, width=960)
+imageio.imwrite('lfw_montage_960.jpg', montage)
+\ No newline at end of file