File size: 1,754 Bytes
077d557 de2aa9b 3cd966d d34396b e50e349 d34396b ba4d1a9 e6ecfd9 671a10f 172a03a ba4d1a9 172a03a 63b8881 172a03a 63b8881 172a03a e50e349 172a03a d4ae210 172a03a 63b8881 172a03a d4ae210 172a03a 63b8881 172a03a b25a6ce 172a03a 3cd966d 5be44ed ce6434e 9444948 63b8881 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import spaces
import gradio as gr
import cv2
#import numpy as np
#from PIL import Image
#from transparent_background import Remover
from backgroundremover from bg import remove
remover = remove()
#remover = Remover()
#@spaces.GPU
#def doo(video):
#cap = cv2.VideoCapture(video)
#fps = cap.get(cv2.CAP_PROP_FPS)
#writer = None
#processed_frames = 0
#while cap.isOpened():
# ret, frame = cap.read()
#
# if ret is False:
# break
# frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
# img = Image.fromarray(frame).convert('RGB')
# if writer is None:
# writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
# processed_frames += 1
# print(f"Processing: {processed_frames}")
# out = remover.process(img, type='green')
# writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
#cap.release()
#writer.release()
#return 'output.mp4'
@spaces.GPU
def doo(video):
video = cv2.VideoCapture(video)
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter('output.mp4', fourcc, 20.0, (640, 480))
while(video.isOpened()):
ret, frame = video.read()
if ret == True:
# Remove the background from the frame
no_bg_frame = remover.remove(frame)
# Write the frame into the file 'output.mp4'
out.write(no_bg_frame)
else:
break
# Release everything when the job is finished
video.release()
out.release()
cv2.destroyAllWindows()
return 'output.mp4'
iface = gr.Interface(fn=doo, inputs="video", outputs="video")
iface.launch() |