Finish Homework

2018-Python-Practice-master/main.py

@@ -7,8 +7,9 @@
 '''
 1. import `Video` and `save_video` from the correct module of package "styler"
 '''
-from ... import Video
-from ... import save_video
+from styler.video import Video
+from styler.utils import save_video
+
 
 
 model_file = 'data/vg-30.pb'
@@ -41,7 +42,7 @@ def main():
         '''
         2. set the `path` to  your input
         '''
-        with Video(...) as v:
+        with Video('/Users/AndyChu/Desktop/2018-Python-Practice/2018-Python-Practice-master/input/jaguar.mp4') as v:
             frames = v.read_frames(image_h=shape[1], image_w=shape[2])
 
         logging.info("Processing image")
@@ -53,15 +54,15 @@ def main():
         '''
         processed = [
             session.run(out, feed_dict={image: [frame]})
-            ...
+            for frame in frames
         ]
 
         '''
         4. Pass the results as a argument into function
         '''
         save_video('result.mp4',
                    fps=30, h=shape[1], w=shape[2],
-                   frames=...)
+                   frames=processed)
 
         logging.info("Processing took %f" % (
             (datetime.now() - start_time).total_seconds()))

2018-Python-Practice-master/styler/video.py

@@ -1,3 +1,5 @@
+import sys
+sys.path.append('/usr/local/lib/python2.7/site-packages')
 import cv2
 
 from styler.utils import resize
@@ -28,9 +30,19 @@ def read_frames(self, image_h, image_w):
         '''
         frames = []
         # 5-1 /5-2 Read video and collect them
-
-        self.frames = ...  # 5-3 let object have the result
-        return ...  # return your results
+        success,image = self.cap.read()
+        count = 0
+        success = True
+        while success:
+            image2 = cv2.resize(image, (image_h, image_w))
+            frames.append(image2)
+            success, image = self.cap.read()
+
+            print ('Read a new frame: ', success , count)
+            count += 1
+
+        self.frames = frames  # 5-3 let object have the result
+        return frames  # return your results
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.cap.release()