freealise commited on
Commit
a3c8099
·
verified ·
1 Parent(s): cb6c3ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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