Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ def get_frames(video_in, step, name, n):
|
|
73 |
#resize_h = resize_w / 2.0
|
74 |
#frame = cv2.resize(frame, (int(resize_w), int(resize_h)))
|
75 |
|
76 |
-
cv2.imwrite(f"{str(n)}_{name}_{step}{str(i)}.png",frame)
|
77 |
frames.append(f"{str(n)}_{name}_{step}{str(i)}.png")
|
78 |
i+=1
|
79 |
|
@@ -118,7 +118,7 @@ def infer(f_in, interpolation, fps_output):
|
|
118 |
next_frame = frames_list[idx+1]
|
119 |
|
120 |
interpolated_frames = do_interpolation(frame, next_frame, interpolation, idx) # should return a list of interpolated frames
|
121 |
-
break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_")
|
122 |
print(break_interpolated_video[0])
|
123 |
for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
|
124 |
print(f"IMG:{img}")
|
|
|
73 |
#resize_h = resize_w / 2.0
|
74 |
#frame = cv2.resize(frame, (int(resize_w), int(resize_h)))
|
75 |
|
76 |
+
cv2.imwrite(f"{str(n)}_{name}_{step}{str(i)}.png", frame)
|
77 |
frames.append(f"{str(n)}_{name}_{step}{str(i)}.png")
|
78 |
i+=1
|
79 |
|
|
|
118 |
next_frame = frames_list[idx+1]
|
119 |
|
120 |
interpolated_frames = do_interpolation(frame, next_frame, interpolation, idx) # should return a list of interpolated frames
|
121 |
+
break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_", -1)
|
122 |
print(break_interpolated_video[0])
|
123 |
for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
|
124 |
print(f"IMG:{img}")
|