Fatal编程技术网
  • python/
  • Python 如何在图片上检测斑马线？

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

python machine-learning computer-vision

Python 如何在图片上检测斑马线？,python,machine-learning,computer-vision,Python,Machine Learning,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

早上好，我刚刚开始学习计算机视觉。我正在尝试建立一个斑马线检测系统。我做的和看电影时一样多。接下来我该怎么办

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()

首先尝试从完整图像中仅获取ROI部分，然后可以使用模板匹配检查类似交叉（）。您必须为可能的模板创建一个小的数据集目录，它将从中匹配。这可能是一个耗时的过程，因此您也可以选择机器学习（ObjectDetectionTensorFlow，YOLO）作为替代方法