general_detection.py

import cv2
from ultralytics import YOLO

# Load the YOLOv8 model
model = YOLO("yolo11n_p.pt")  # Use correct model name

# Open webcam (0 for default camera)
cap = cv2.VideoCapture(0)  # Use 0 for webcam or video path for video file

# Set camera resolution (optional)
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)

# Loop through the video frames
while cap.isOpened():
    # Read a frame from the video
    success, frame = cap.read()
    
    if success:
        # Run YOLOv8 tracking on the frame, persisting tracks between frames
        results = model.track(frame, persist=True)
        
        # Visualize the results on the frame
        if results and len(results) > 0:
            annotated_frame = results[0].plot()
            
            # Display the annotated frame
            cv2.imshow("Live feed of something yay", annotated_frame)
        
        # Break the loop if 'q' is pressed
        if cv2.waitKey(1) & 0xFF == ord("q"):
            break
    else:
        # Break the loop if the end of the video is reached
        break

# Release the video capture object and close the display window
cap.release()
cv2.destroyAllWindows()