typoo

Author: aamini

mitdeeplearning/lab3.py

@@ -4,10 +4,18 @@
 
 
 def play_video(filename):
-    encoded = base64.b64encode(io.open('./agent.mp4', 'r+b').read())
+    encoded = base64.b64encode(io.open(filename, 'r+b').read())
     embedded = HTML(data='''
         <video controls>
             <source src="data:video/mp4;base64,{0}" type="video/mp4" />
         </video>'''.format(encoded.decode('ascii')))
 
     return embedded
+
+def preprocess_pong(image):
+    I = image[35:195] # Crop
+    I = I[::2, ::2, 0] # Downsample width and height by a factor of 2
+    I[I == 144] = 0 # Remove background type 1
+    I[I == 109] = 0 # Remove background type 2
+    I[I != 0] = 1 # Set remaining elements (paddles, ball, etc.) to 1
+    return I.astype(np.float).reshape(80, 80, 1)

setup.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.5.0',      # Start with a small number and increase it with every change you make
+  version = '0.5.1',      # 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.5.0.tar.gz',    # I explain this later on
+  download_url = 'https://github.com/aamini/introtodeeplearning_labs/archive/v0.5.1.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=[