Spaces:
Sleeping
Sleeping
| import cv2 | |
| import gradio as gr | |
| # get the frames | |
| def frame_capture(video_path): | |
| # Function to extract frames | |
| vidObj = cv2.VideoCapture(video_path) | |
| # Used as counter variable | |
| count = 0 | |
| frames = [] | |
| # checks whether frames were extracted | |
| success = 1 | |
| while success: | |
| # vidObj object calls read | |
| # function to extract frames | |
| success, image = vidObj.read() | |
| # Append the frame to the list | |
| if success: | |
| frames.append(image) | |
| count += 1 | |
| return frames | |
| def extract_frames(video): | |
| frames = frame_capture(video) | |
| return frames | |
| # Define the Gradio interface | |
| video_input = gr.Video(label="Upload Video", autoplay=True) | |
| output_frames = gr.Gallery(label='Frame') | |
| # Create the Gradio app | |
| gr.Interface(fn=extract_frames, inputs=video_input, outputs=output_frames, title='Video Frame Extractor').launch() |