freealise commited on
Commit
978f0af
·
verified ·
1 Parent(s): d8d1ea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -137,7 +137,9 @@ def infer(f_in, interpolation, fps_output):
137
  return final_vid, files
138
 
139
 
140
- def remove_bg(frame):
 
 
141
  b = 5
142
  #subtract background (get scene with shadow) and correct hue against light
143
  bg = cv2.medianBlur(frame, 255)
 
137
  return final_vid, files
138
 
139
 
140
+ def remove_bg(fr):
141
+ frame = cv2.imread(fr).astype(np.uint8)
142
+
143
  b = 5
144
  #subtract background (get scene with shadow) and correct hue against light
145
  bg = cv2.medianBlur(frame, 255)