Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def do_interpolation(frame1, frame2, interpolation, n):
|
|
37 |
mediapy.write_video(f"{n}_to_{n+1}_out.mp4", frames, fps=25)
|
38 |
return f"{n}_to_{n+1}_out.mp4"
|
39 |
|
40 |
-
def get_frames(video_in, step, name):
|
41 |
frames = []
|
42 |
cap = cv2.VideoCapture(video_in)
|
43 |
cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
@@ -73,8 +73,8 @@ def get_frames(video_in, step, name):
|
|
73 |
#resize_h = resize_w / 2.0
|
74 |
#frame = cv2.resize(frame, (int(resize_w), int(resize_h)))
|
75 |
|
76 |
-
cv2.imwrite(f"{name}_{step}{str(i)}.png",frame)
|
77 |
-
frames.append(f"{name}_{step}{str(i)}.png")
|
78 |
i+=1
|
79 |
|
80 |
cap.release()
|
@@ -144,7 +144,7 @@ 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())
|
@@ -160,7 +160,7 @@ def sortFiles(e):
|
|
160 |
return e[len(e)-1]
|
161 |
|
162 |
def loadf(f):
|
163 |
-
if f != None:
|
164 |
f.sort(key=sortFiles)
|
165 |
fnew = []
|
166 |
|
|
|
37 |
mediapy.write_video(f"{n}_to_{n+1}_out.mp4", frames, fps=25)
|
38 |
return f"{n}_to_{n+1}_out.mp4"
|
39 |
|
40 |
+
def get_frames(video_in, step, name, n):
|
41 |
frames = []
|
42 |
cap = cv2.VideoCapture(video_in)
|
43 |
cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
|
|
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 |
|
80 |
cap.release()
|
|
|
144 |
return int(math.log2(linear))
|
145 |
|
146 |
def sharpest(fl, i):
|
147 |
+
break_vid = get_frames(fl, "vid_input_frame", "origin", i)
|
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())
|
|
|
160 |
return e[len(e)-1]
|
161 |
|
162 |
def loadf(f):
|
163 |
+
if f != None and f != [None]:
|
164 |
f.sort(key=sortFiles)
|
165 |
fnew = []
|
166 |
|