scriptMethod1_ncnn.py

import cv2
from picamera2 import Picamera2
from ultralytics import YOLO

# Fix for Qt plugin issue (forces X11 or Wayland compatibility)
import os
os.environ["QT_QPA_PLATFORM"] = "offscreen"

print("Initializing camera...")
picam2 = Picamera2()
picam2.preview_configuration.main.size = (1280, 720)
picam2.preview_configuration.main.format = "RGB888"
picam2.preview_configuration.align()
picam2.configure("preview")
picam2.start()
print("Camera initialized successfully!")

# Load YOLO model
model = YOLO("yolo11n_ncnn_model")
print("Model loaded!")

while True:
    frame = picam2.capture_array()

    # Run YOLO inference
    results = model(frame)

    # Visualize results
    annotated_frame = results[0].plot()

    # Show frame using OpenCV
    cv2.imshow("YOLO Detection", annotated_frame)

    # Exit when 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord("q"):
        break

# Cleanup
cv2.destroyAllWindows()