Merge pull request #47 from atgambardella/master

changed reshape to moveaxis

Author: ikostrikov2

envs.py

@@ -19,10 +19,10 @@ def _process_frame42(frame):
     # aren't close enough to the pixel boundary.
     frame = cv2.resize(frame, (80, 80))
     frame = cv2.resize(frame, (42, 42))
-    frame = frame.mean(2)
+    frame = frame.mean(2, keepdims=True)
     frame = frame.astype(np.float32)
     frame *= (1.0 / 255.0)
-    frame = np.reshape(frame, [1, 42, 42])
+    frame = np.moveaxis(frame, -1, 0)
     return frame