Update app.py
Browse files
app.py
CHANGED
@@ -92,15 +92,6 @@ def create_video(frames, fps, type):
|
|
92 |
|
93 |
mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
|
94 |
return type + "_result.mp4"
|
95 |
-
|
96 |
-
|
97 |
-
def sharpest(f):
|
98 |
-
break_vid = get_frames(f, "vid_input_frame", "origin")
|
99 |
-
blur_s = []
|
100 |
-
for jdx, fr in enumerate(break_vid[0]):
|
101 |
-
blur_s.append(cv2.Laplacian(cv2.cvtColor(cv2.imread(fr).astype(np.uint8), cv2.COLOR_BGR2GRAY), cv2.CV_64F).var())
|
102 |
-
|
103 |
-
return break_vid[0][np.argmax(blur_s)]
|
104 |
|
105 |
|
106 |
def infer(f_in, interpolation, fps_output):
|
@@ -126,11 +117,6 @@ def infer(f_in, interpolation, fps_output):
|
|
126 |
if idx < len(frames_list) - 1:
|
127 |
next_frame = frames_list[idx+1]
|
128 |
|
129 |
-
ftype = frame.split('/')
|
130 |
-
if ftype[len(ftype)-1].split('.')[1] == 'mp4':
|
131 |
-
frame = sharpest(frame)
|
132 |
-
next_frame = sharpest(next_frame)
|
133 |
-
|
134 |
interpolated_frames = do_interpolation(frame, next_frame, interpolation, idx) # should return a list of interpolated frames
|
135 |
break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_")
|
136 |
print(break_interpolated_video[0])
|
@@ -157,6 +143,14 @@ def logscale(linear):
|
|
157 |
def linscale(linear):
|
158 |
return int(math.log2(linear))
|
159 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
160 |
def sortFiles(e):
|
161 |
e = e.split('/')
|
162 |
return e[len(e)-1]
|
@@ -164,6 +158,11 @@ def sortFiles(e):
|
|
164 |
def loadf(f):
|
165 |
if f != None:
|
166 |
f.sort(key=sortFiles)
|
|
|
|
|
|
|
|
|
|
|
167 |
return f, f
|
168 |
|
169 |
|
|
|
92 |
|
93 |
mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
|
94 |
return type + "_result.mp4"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
95 |
|
96 |
|
97 |
def infer(f_in, interpolation, fps_output):
|
|
|
117 |
if idx < len(frames_list) - 1:
|
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])
|
|
|
143 |
def linscale(linear):
|
144 |
return int(math.log2(linear))
|
145 |
|
146 |
+
def sharpest(fl, i):
|
147 |
+
break_vid = get_frames(fl, "vid_input_frame", "origin")
|
148 |
+
blur_s = []
|
149 |
+
for jdx, fr in enumerate(break_vid[0]):
|
150 |
+
blur_s.append(cv2.Laplacian(cv2.cvtColor(cv2.imread(fr).astype(np.uint8), cv2.COLOR_BGR2GRAY), cv2.CV_64F).var())
|
151 |
+
print(i +'th file, sharpest: '+np.max(blur_s))
|
152 |
+
return break_vid[0][np.argmax(blur_s)]
|
153 |
+
|
154 |
def sortFiles(e):
|
155 |
e = e.split('/')
|
156 |
return e[len(e)-1]
|
|
|
158 |
def loadf(f):
|
159 |
if f != None:
|
160 |
f.sort(key=sortFiles)
|
161 |
+
|
162 |
+
for i, fl in enumerate(f):
|
163 |
+
ftype = fl.split('/')
|
164 |
+
if ftype[len(ftype)-1].split('.')[1] == 'mp4':
|
165 |
+
fl = sharpest(fl, i)
|
166 |
return f, f
|
167 |
|
168 |
|