|
|
import cv2 as cv
|
|
|
import numpy as np
|
|
|
import gradio as gr
|
|
|
import os
|
|
|
|
|
|
def process_video(input_video):
|
|
|
output_path = "car_output.mp4"
|
|
|
|
|
|
cap = cv.VideoCapture(input_video)
|
|
|
fourcc = cv.VideoWriter_fourcc(*"mp4v")
|
|
|
fps = int(cap.get(cv.CAP_PROP_FPS))
|
|
|
width = int(cap.get(cv.CAP_PROP_FRAME_WIDTH))
|
|
|
height = int(cap.get(cv.CAP_PROP_FRAME_HEIGHT))
|
|
|
out = cv.VideoWriter(output_path, fourcc, fps, (width, height))
|
|
|
|
|
|
while cap.isOpened():
|
|
|
ret, frame = cap.read()
|
|
|
if not ret:
|
|
|
break
|
|
|
|
|
|
b, g, r = cv.split(frame)
|
|
|
rgb_frame = cv.merge((r, g, b))
|
|
|
out.write(rgb_frame)
|
|
|
|
|
|
cap.release()
|
|
|
out.release()
|
|
|
|
|
|
return output_path
|
|
|
|
|
|
gr.Interface(
|
|
|
fn=process_video,
|
|
|
inputs=gr.Video(label="Upload a video"),
|
|
|
outputs=gr.Video(label="Processed Video Output"),
|
|
|
title="RGB Channel Swap Video Processor",
|
|
|
description="This app reads a video, swaps the BGR to RGB channel, and shows the processed video.",
|
|
|
allow_flagging="never"
|
|
|
).launch()
|
|
|
|
