From 72abb2f6d0205f7d022d26e9e25fd74b9360a554 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Mon, 4 Mar 2019 22:16:30 +0100 Subject: cosmetics --- .../pages/datasets/lfw/assets/fetch_lfw_people.py | 34 ++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 site/content/pages/datasets/lfw/assets/fetch_lfw_people.py (limited to 'site/content/pages/datasets/lfw/assets/fetch_lfw_people.py') 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 -- cgit v1.2.3-70-g09d2