Как обнаружить зебру на картинке?

Question

Доброе утро, я только начал с Computer Vision. Я пытаюсь построить обнаружение пересечения зебры. Я сделал так же, как на картинке. Что я должен делать дальше?

import cv2

test_image = cv2.imread('dataset/crossway4.jpg')
gray_image = cv2.cvtColor(test_image, cv2.COLOR_BGR2GRAY)

# Extract white color pixels 
white_mask = cv2.inRange(gray_image, 200, 255)
gaussian_blur = cv2.GaussianBlur(white_mask, (5, 5), cv2.BORDER_DEFAULT)

cv2.imshow('Zebra Crossing Detection', gaussian_blur)

cv2.waitKey(0)
cv2.destroyAllWindows()

Answer 1

Сначала попытайтесь получить только часть ROI из полного изображения, затем вы можете использовать сопоставление с шаблоном, чтобы проверить на подобное пересечение (https://docs.opencv.org/master/d4/dc6/tutorial_py_template_matching.html). введите описание изображения здесь Вы должны сделать небольшая директория набора данных возможного шаблона, из которой он будет соответствовать. эта обработка может занять много времени, поэтому вы также можете выбрать машинное обучение (тензор потока обнаружения объекта, YOLO) в качестве альтернативного метода.