Merge pull request yandexdataschool#63 from NickVeld/patch-17

justheuristic · web-flow · commit 5ff92e2bad41 · 2019-11-27T02:44:23.000+03:00
week9: lfw_dataset rewrited for scipy &gt;= 1.2.0
diff --git a/week09_adversarial/lfw_dataset.py b/week09_adversarial/lfw_dataset.py
@@ -1,8 +1,13 @@
 import numpy as np
 import os
-from scipy.misc import imread,imresize
+from imageio import imread # as advised https://docs.scipy.org/doc/scipy-1.2.1/reference/generated/scipy.misc.imread.html
+from PIL import Image
 import pandas as pd
 
+# as advised https://docs.scipy.org/doc/scipy-1.2.1/reference/generated/scipy.misc.imresize.html?highlight=imresize#scipy.misc.imresize
+def imresize(arr, size):
+    return np.array(Image.fromarray(arr).resize(size=size))
+
 def fetch_lfw_dataset(attrs_name = "lfw_attributes.txt",
                       images_name = "lfw-deepfunneled",
                       raw_images_name = "lfw",
@@ -68,4 +73,3 @@ def fetch_lfw_dataset(attrs_name = "lfw_attributes.txt",
     all_attrs = df.drop(["photo_path","person","imagenum"],axis=1)
     
     return all_photos,all_attrs
-    
