Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -50,8 +50,8 @@ def capture_function(vid=None):
|
|
50 |
capture.set(cv2.CAP_PROP_POS_FRAMES, i)
|
51 |
ret, frame_f = capture.read(i)
|
52 |
frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
|
53 |
-
cv2.imwrite(f'{uid}/{i}.png',frame_ff)
|
54 |
-
fbox.append(f'{uid}/{i}.png')
|
55 |
return fbox,fbox
|
56 |
|
57 |
def im_2_vid(images,fps):
|
@@ -91,19 +91,7 @@ def rem_bg(vid):
|
|
91 |
frame_num=f'Done: {i+1} of {frame_count1-1}'
|
92 |
out_vid,_file = im_2_vid(fbox2, fps)
|
93 |
yield (fbox2,frame_num,out_vid)
|
94 |
-
|
95 |
-
def rem_bg_og(inp):
|
96 |
-
#transparent background .mov
|
97 |
-
#os.system(f'backgroundremover -i "{inp}" -tv -o "{inp}.mov"')
|
98 |
-
#video video to be overlayed output
|
99 |
-
#os.system(f' backgroundremover -i "/path/to/video.mp4" -tov "/path/to/videtobeoverlayed.mp4" -o "output.mov"')
|
100 |
-
#video over image file
|
101 |
-
#os.system(f'backgroundremover -i "/path/to/video.mp4" -toi "/path/to/videtobeoverlayed.mp4" -o "output.mov"')
|
102 |
-
#output to transparent GIF
|
103 |
-
#os.system(f'backgroundremover -i "/path/to/video.mp4" -tg -o "output.gif"')
|
104 |
-
#output to matte background
|
105 |
-
os.system(f'backgroundremover -i "{inp}" -mk -o "{inp}.matte.mp4"')
|
106 |
-
return f'{inp}.matte.mp4'
|
107 |
|
108 |
def predict(text, url_params):
|
109 |
mod_url=""
|
|
|
50 |
capture.set(cv2.CAP_PROP_POS_FRAMES, i)
|
51 |
ret, frame_f = capture.read(i)
|
52 |
frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
|
53 |
+
cv2.imwrite(f'{uid}-frames/{i}.png',frame_ff)
|
54 |
+
fbox.append(f'{uid}-frames/{i}.png')
|
55 |
return fbox,fbox
|
56 |
|
57 |
def im_2_vid(images,fps):
|
|
|
91 |
frame_num=f'Done: {i+1} of {frame_count1-1}'
|
92 |
out_vid,_file = im_2_vid(fbox2, fps)
|
93 |
yield (fbox2,frame_num,out_vid)
|
94 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
95 |
|
96 |
def predict(text, url_params):
|
97 |
mod_url=""
|