如何在图片上检测斑马线？

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）。enter image description here您必须为可能的模板创建一个小的数据集目录，它将从中匹配。这可能是一个耗时的过程，因此您也可以选择机器学习（ObjectDetectionTensorFlow，YOLO）作为替代方法