Vision Sensors & Cameras

import cv2
import numpy as np

def main():
    # Initialize basic USB camera (Index 0)
    cap = cv2.VideoCapture(0)

    if not cap.isOpened():
        print("Error: Could not open camera.")
        return

    print("Press 'q' to quit.")

    while True:
        # Capture frame-by-frame
        ret, frame = cap.read()
        if not ret:
            break

        # Convert to grayscale (simulating basic processing)
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

        # Apply Canny edge detection
        edges = cv2.Canny(frame, 100, 200)

        # Display the resulting frames
        cv2.imshow('RGB Feed', frame)
        cv2.imshow('Edge Detection', edges)

        # Wait for 'q' key
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

    # Release the capture
    cap.release()
    cv2.destroyAllWindows()

if __name__ == "__main__":
    main()