diff --git a/2018-Python-Practice-master/main.py b/2018-Python-Practice-master/main.py index 9d8e7df..0681b54 100644 --- a/2018-Python-Practice-master/main.py +++ b/2018-Python-Practice-master/main.py @@ -7,8 +7,8 @@ ''' 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 +41,7 @@ def main(): ''' 2. set the `path` to your input ''' - with Video(...) as v: + with Video("./input/jaguar.mp4") as v: frames = v.read_frames(image_h=shape[1], image_w=shape[2]) logging.info("Processing image") @@ -53,7 +53,7 @@ def main(): ''' processed = [ session.run(out, feed_dict={image: [frame]}) - ... + for frame in frames ] ''' @@ -61,7 +61,7 @@ def main(): ''' 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())) diff --git a/2018-Python-Practice-master/result.mp4 b/2018-Python-Practice-master/result.mp4 new file mode 100644 index 0000000..5a154a3 Binary files /dev/null and b/2018-Python-Practice-master/result.mp4 differ diff --git a/2018-Python-Practice-master/styler/__pycache__/utils.cpython-35.pyc b/2018-Python-Practice-master/styler/__pycache__/utils.cpython-35.pyc new file mode 100644 index 0000000..04d17af Binary files /dev/null and b/2018-Python-Practice-master/styler/__pycache__/utils.cpython-35.pyc differ diff --git a/2018-Python-Practice-master/styler/__pycache__/video.cpython-35.pyc b/2018-Python-Practice-master/styler/__pycache__/video.cpython-35.pyc new file mode 100644 index 0000000..4eb3d17 Binary files /dev/null and b/2018-Python-Practice-master/styler/__pycache__/video.cpython-35.pyc differ diff --git a/2018-Python-Practice-master/styler/video.py b/2018-Python-Practice-master/styler/video.py index deb57be..035bac7 100644 --- a/2018-Python-Practice-master/styler/video.py +++ b/2018-Python-Practice-master/styler/video.py @@ -27,10 +27,15 @@ def read_frames(self, image_h, image_w): - Return your results ''' frames = [] + while(self.cap.isOpened): + ret, frame = self.cap.read() + if ret == False: + break + frames.append(cv2.resize(frame, (512, 512))) # 5-1 /5-2 Read video and collect them - self.frames = ... # 5-3 let object have the result - return ... # return your results + 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()