Dynamic vid_stride with multiple RTSPs about ultralytics HOT 3 OPEN

import threading
from ultralytics import YOLO

def track_stream(source, vid_stride):
    model = YOLO('yolov8s.pt', task='detect')
    results = model.track(source, device=device, stream=True, classes=classes, verbose=False, stream_buffer=True, tracker=byte_track, vid_stride=vid_stride)
    for pred in results:
        print(pred)

# Define your streams and corresponding stride values
streams = {'rtsp1': 1, 'rtsp2': 1, 'rtsp3': 5, 'rtsp4': 5}

# Create and start a separate thread for each stream
threads = []
for stream, stride in streams.items():
    t = threading.Thread(target=track_stream, args=(stream, stride))
    t.start()
    threads.append(t)

# Wait for all threads to finish
for t in threads:
    t.join()

import cv2
from ultralytics import YOLO

def process_stream(stream_url, frame_skip, device):
    cap = cv2.VideoCapture(stream_url)
    model = YOLO('yolov8s.pt', task='detect')
    frame_count = 0

    while True:
        ret, frame = cap.read()
        if not ret:
            break
        if frame_count % (frame_skip + 1) == 0:  # frame_skip + 1 because we include the current frame in the count
            results = model.track(frame, device=device, stream=False, classes=None, verbose=False, stream_buffer=False, tracker='byte_track', vid_stride=1)
            for pred in results:
                path = pred.path
                print(f"Processed frame from {stream_url} saved to {path}")
        frame_count += 1

    cap.release()

# Device configuration
device = 'cuda'  # or 'cpu'

# Define your streams and their respective frame skips (corresponding to vid_stride - 1)
streams = {
    'rtsp1': 0,  # Process every frame
    'rtsp2': 0,  # Process every frame
    'rtsp3': 4,  # Skip 4 frames
    'rtsp4': 4,  # Skip 4 frames
}

# Loop for processing each stream one at a time
for stream_url, skip in streams.items():
    print(f"Starting processing for {stream_url}")
    process_stream(stream_url, skip, device)
    print(f"Finished processing for {stream_url}")