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}")
|