Update app.py
Browse files
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(
|
|
|
|
|
|
|
| 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)
|