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 |
|