Spaces:
Running
Running
File size: 4,100 Bytes
ae1809b 6820b4e 83907f9 c5a4e4e ae1809b 6820b4e c5a4e4e ae1809b |
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 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 |
import av
import cv2
import numpy as np
import streamlit as st
from streamlit_webrtc import webrtc_streamer, WebRtcMode
from src.opencv_utils import OpenCVUtils
from utils.turn import get_ice_servers
st.set_page_config(page_title="OpenCV Explorer", page_icon="🎨", layout="wide")
@st.cache_resource
def get_app():
return OpenCVUtils()
app = get_app()
# --- HIDE STREAMLIT STYLE ---
hide_st_style = """
<style>
#MainMenu {visibility: hidden;}
footer {visibility: hidden;}
header {visibility: hidden;}
</style>
"""
st.markdown(hide_st_style, unsafe_allow_html=True)
# ---------------------------
st.markdown("# 🎨 OpenCV Explorer")
st.markdown("Explore filters and transformations in real-time using your webcam.")
# Sidebar Controls
FUNCTIONS = [
"Color Filter",
"Canny",
"Blur",
"Rotation",
"Resize",
"Contour",
"Histogram Equalization",
"Adaptive Threshold",
"Morphology",
"Sharpen",
"Hough Lines",
"Optical Flow",
"Pencil Sketch",
"Color Quantization",
"Hand Tracker",
"Face Tracker",
]
selected_functions = st.sidebar.multiselect(
"Select and order functions:", FUNCTIONS, default=[]
)
# Parameters
with st.sidebar.expander("Color Filter"):
lh = st.slider("Lower Hue", 0, 180, 0)
uh = st.slider("Upper Hue", 0, 180, 180)
ls = st.slider("Lower Sat", 0, 255, 0)
us = st.slider("Upper Sat", 0, 255, 255)
lv = st.slider("Lower Val", 0, 255, 0)
uv = st.slider("Upper Val", 0, 255, 255)
with st.sidebar.expander("Canny Edge"):
lc = st.slider("Lower Canny", 0, 255, 100)
uc = st.slider("Upper Canny", 0, 255, 200)
with st.sidebar.expander("Blur"):
bk = st.slider("Kernel Size (odd)", 1, 15, 5, step=2)
with st.sidebar.expander("Rotation"):
ang = st.slider("Angle", 0, 360, 0)
with st.sidebar.expander("Resize"):
w = st.slider("Width", 100, 1920, 640)
h = st.slider("Height", 100, 1080, 480)
with st.sidebar.expander("Morphology"):
morph_op = st.selectbox("Operation", ["erode", "dilate", "open", "close"])
morph_ks = st.slider("Kernel Size", 1, 31, 5, step=2)
prev_gray = None
def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
global prev_gray
img = frame.to_ndarray(format="bgr24")
curr_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
for fn in selected_functions:
if fn == "Color Filter":
img = app.apply_color_filter(img, (lh, ls, lv), (uh, us, uv))
elif fn == "Canny":
img = app.apply_edge_detection(img, lc, uc)
elif fn == "Blur":
img = app.blur_image(img, bk)
elif fn == "Rotation":
img = app.rotate_image(img, ang)
elif fn == "Resize":
img = app.resize_image(img, w, h)
elif fn == "Contour":
img = app.apply_contour_detection(img)
elif fn == "Histogram Equalization":
img = app.equalize_histogram(img)
elif fn == "Adaptive Threshold":
img = app.adaptive_threshold(img)
elif fn == "Morphology":
img = app.morphology(img, morph_op, morph_ks)
elif fn == "Sharpen":
img = app.sharpen(img)
elif fn == "Hough Lines":
img = app.hough_lines(img)
elif fn == "Optical Flow" and prev_gray is not None:
img = app.optical_flow(prev_gray, curr_gray, img)
elif fn == "Pencil Sketch":
img = app.pencil_sketch(img)
elif fn == "Color Quantization":
img = app.color_quantization(img)
elif fn == "Hand Tracker":
img = app.detect_hands(img)
elif fn == "Face Tracker":
img = app.detect_faces(img)
prev_gray = curr_gray
return av.VideoFrame.from_ndarray(img, format="bgr24")
webrtc_streamer(
key="opencv-explorer",
mode=WebRtcMode.SENDRECV,
rrtc_configuration={"iceServers": get_ice_servers()},
video_frame_callback=video_frame_callback,
media_stream_constraints={"video": True, "audio": False},
async_processing=True,
)
|