Написан код на определения машин, но 1 машина может быть сфоткана 3 - 5 раз. Нужно добавить какой то тайм аут между снимками
import cv2
import os
cap = cv2.VideoCapture("video.mp4")
car_cascade = cv2.CascadeClassifier('cars.xml')
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
while True:
# Считываем кадр из видеопотока
ret, frame = cap.read()
# Преобразуем кадр в оттенки серого
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# Определяем автомобили на кадре
cars = car_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=3, minSize=(700, 700))
# Рисуем прямоугольники вокруг автомобилей
for (x, y, w, h) in cars:
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
# Сохраняем изображениe при обнаружении автомобиля
if len(cars) > 0:
# Формируем имя файла счетчиком
photo_name = 'car_{}.jpg'.format(photo_counter)
# Увеличиваем счетчик
photo_counter += 1
# Сохраняем фотографию в папке cars_photos на рабочем столе
cv2.imwrite(os.path.join(desktop_path, 'cars_photos', photo_name), frame)
# Отображаем кадр с прямоугольниками автомобилей
cv2.imshow('frame', frame)
# Обрабатываем нажатие клавиши 'q' для выхода из программы
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release() cv2.destroyAllWindows()