Обнаружение и окраска полос OpenCV

Question

У меня есть дорожное изображение https://imgur.com/a/whT90Yp, и я хочу определить полосу и раскрасить обе стороны, как это https://i.stack.imgur.com/1NuD8.jpg, я сделал код в openCV, но это не работает, и я не знаю почему: (

Это код:

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt


def lanesDetection(img):
im = cv.imread("highway4.jpg")
im = cv.cvtColor(im, cv.COLOR_BGR2RGB)

# print(img.shape)
height = img.shape[0]
width = img.shape[1]

region_of_interest_vertices = [
    (200, height), (width/2, height/1.37), (width-300, height)
]
gray_img = cv.cvtColor(img, cv.COLOR_RGB2GRAY)
edge = cv.Canny(gray_img, 50, 100, apertureSize=3)
cropped_image = region_of_interest(
    edge, np.array([region_of_interest_vertices], np.int32))

lines = cv.HoughLinesP(cropped_image, rho=2, theta=np.pi/180,
                       threshold=50, lines=np.array([]), minLineLength=10, maxLineGap=30)
image_with_lines = draw_lines(img, lines)
# plt.imshow(image_with_lines)
# plt.show()
return image_with_lines


def region_of_interest(img, vertices):
mask = np.zeros_like(img)
# channel_count = img.shape[2]
match_mask_color = (255)
cv.fillPoly(mask, vertices, match_mask_color)
masked_image = cv.bitwise_and(img, mask)
return masked_image


def draw_lines(img, lines):
img = np.copy(img)
blank_image = np.zeros((img.shape[0], img.shape[1], 3), np.uint8)

for line in lines:
    for x1, y1, x2, y2 in line:
        cv.line(blank_image, (x1, y1), (x2, y2), (0, 255, 0), 2)

img = cv.addWeighted(img, 0.8, blank_image, 1, 0.0)
return img


cap.release()
cv.destroyAllWindows()

Можете ли вы помочь Ми исправить это?

Answer 1

я взял ваш код и сделал его работоспособным: некоторые вещи, которые я сделал, были:

• исправлено смешанное использование пробелов и табуляций в файле
• изменил функцию для принятия имя файла (у вас это было жестко закодировано)
• добавил 7 строк в конце (ваш код не вызывал функции или не отображал изображение результата)

import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt


def lanesDetection(imgname):
    img = cv.imread(imgname)
    img = cv.cvtColor(img, cv.COLOR_BGR2RGB)

# print(img.shape)
    height = img.shape[0]
    width = img.shape[1]

    region_of_interest_vertices = [(200, height), (width/2, height/1.37), (width-300, height)]
    gray_img = cv.cvtColor(img, cv.COLOR_RGB2GRAY)
    edge = cv.Canny(gray_img, 50, 100, apertureSize=3)
    cropped_image = region_of_interest(edge, np.array([region_of_interest_vertices], np.int32))

    lines = cv.HoughLinesP(cropped_image, rho=2, theta=np.pi/180,threshold=50, lines=np.array([]), minLineLength=10, maxLineGap=30)
    image_with_lines = draw_lines(img, lines)
# plt.imshow(image_with_lines)
# plt.show()
    return image_with_lines

def blah(imgname):
    blahb = cv.imread(imgname)
    return blahb

def region_of_interest(img, vertices):
    mask = np.zeros_like(img)
# channel_count = img.shape[2]
    match_mask_color = (255)
    cv.fillPoly(mask, vertices, match_mask_color)
    masked_image = cv.bitwise_and(img, mask)
    return masked_image


def draw_lines(img, lines):
    img = np.copy(img)
    blank_image = np.zeros((img.shape[0], img.shape[1], 3), np.uint8)

    for line in lines:
        for x1, y1, x2, y2 in line:
            cv.line(blank_image, (x1, y1), (x2, y2), (0, 255, 0), 2)

    img = cv.addWeighted(img, 0.8, blank_image, 1, 0.0)
    return img


#cap.release()
#cv.destroyAllWindows()

pimage = lanesDetection("highway4.jpg")

while 1:
    cv.imshow("Image",pimage)
    key = cv.waitKey(0) & 0xFF
    if key == ord("q"):
        break