test faces

Author: aamini

mitdeeplearning/data/test_faces.h5py

@@ -89,6 +89,11 @@ def get_all_train_faces(self):
         return self.images[ self.pos_train_inds ]
 
 
+def get_test_faces():
+    cwd = os.path.dirname(__file__)
+    f = h5py.File(os.path.join(cwd, "data", "test_faces.h5py"), "r")
+    return f["LM"][:]/255., f["LF"][:]/255. f["DM"][:]/255., f["DF"][:]/255.
+
 
 class PPBFaceEvaluator:
     ''' Evaluate on the PPB dataset'''

mitdeeplearning/lab2.py

@@ -21,13 +21,13 @@ def get_dist(pkgname):
 setup(
   name = 'mitdeeplearning',         # How you named your package folder (MyLib)
   packages = ['mitdeeplearning'],   # Chose the same as "name"
-  version = '0.3.9',      # Start with a small number and increase it with every change you make
+  version = '0.4.0',      # Start with a small number and increase it with every change you make
   license='MIT',        # Chose a license from here: https://help.github.com/articles/licensing-a-repository
   description = 'Official software labs for MIT Introduction to Deep Learning (http://introtodeeplearning.com)',   # Give a short description about your library
   author = 'Alexander Amini',                   # Type in your name
   author_email = '[email protected]',      # Type in your E-Mail
   url = 'http://introtodeeplearning.com',   # Provide either the link to your github or to your website
-  download_url = 'https://github.com/aamini/introtodeeplearning_labs/archive/v0.3.9.tar.gz',    # I explain this later on
+  download_url = 'https://github.com/aamini/introtodeeplearning_labs/archive/v0.4.0.tar.gz',    # I explain this later on
   keywords = ['deep learning', 'neural networks', 'tensorflow', 'introduction'],   # Keywords that define your package best
   install_requires=install_deps,
   classifiers=[