У меня есть код который фиксирует и сохраняет фото проезжающих машин на дорого (Камера фоткает машины). Нужно что бы машина полностью попадала в кадр

Фиксация происходит как машина только попадает в кадр, из за этого фоткается только лобовая часть машины. Моё мнение, нужно что бы прямоугольник который отслеживает машины находился снизу экрана и только тогда фото машины сохранялось. Как это реализовать не знаю(

import cv2
import os
import time

cap = cv2.VideoCapture("video.mp4")
car_cascade = cv2.CascadeClassifier('cars.xml')

# Получаем путь к рабочему столу и создаем в нем папку cars_photos
desktop_path = os.path.join(os.path.expanduser("~"), "Desktop")
if not os.path.exists(os.path.join(desktop_path, 'cars_photos')):
    os.mkdir(os.path.join(desktop_path, 'cars_photos'))

photo_counter = 0
save_counter = 0
while True:
    # Считываем кадр из видеопотока
    ret, frame = cap.read()

    # Преобразуем кадр в оттенки серого
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # Определяем автомобили на кадре
    cars = car_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=3, minSize=(600, 600))

    # Рисуем прямоугольники вокруг автомобилей
    for (x, y, w, h) in cars:
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

        # Сохраняем изображение машины
        if save_counter % 3 == 0:
            photo_name = 'car_{}.jpg'.format(photo_counter)
            photo_counter += 1
            cv2.imwrite(os.path.join(desktop_path, 'cars_photos', photo_name), frame)
        save_counter += 1

        # Отображаем кадр с прямоугольником вокруг машины
        cv2.imshow('frame', frame[y:y + h, x:x + w])
    else:
        cv2.imshow('frame', frame)

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

cap.release()
cv2.destroyAllWindows()

Ответы (0 шт):