dilation_filter.py

import cv2
import numpy as np


def main():
    """Apply grayscale dilation to webcam stream."""
    cap = cv2.VideoCapture(0)
    if not cap.isOpened():
        print("Error: Could not open webcam")
        return

    print("Webcam opened successfully!")
    print("Press 'q' to quit")

    # 5x5 square kernel for dilation
    kernel = np.ones((5, 5), np.uint8)

    while True:
        ret, frame = cap.read()
        if not ret:
            print("Error: Could not read frame")
            break

        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        dilated = cv2.dilate(gray, kernel, iterations=1)

        # Convert to BGR for stacking
        gray_bgr = cv2.cvtColor(gray, cv2.COLOR_GRAY2BGR)
        dilated_bgr = cv2.cvtColor(dilated, cv2.COLOR_GRAY2BGR)

        combined = np.hstack((frame, gray_bgr, dilated_bgr))

        cv2.putText(combined, "Original", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)
        cv2.putText(
            combined,
            "Grayscale",
            (frame.shape[1] + 10, 30),
            cv2.FONT_HERSHEY_SIMPLEX,
            0.7,
            (0, 255, 0),
            2,
        )
        cv2.putText(
            combined,
            "Dilated Gray",
            (frame.shape[1] * 2 + 10, 30),
            cv2.FONT_HERSHEY_SIMPLEX,
            0.7,
            (0, 255, 0),
            2,
        )

        cv2.imshow("Grayscale Dilation Filter", combined)

        if cv2.waitKey(1) & 0xFF == ord("q"):
            break

    cap.release()
    cv2.destroyAllWindows()


if __name__ == "__main__":
    main()