freealise commited on
Commit
53a7ccb
·
verified ·
1 Parent(s): 9f3c7d5

Update app.py

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