Tôi đã phát hiện ra một đường ranh giới làn đường không thẳng bằng cách sử dụng biến đổi Hough và sau đó trích xuất dòng đó một cách riêng biệt.Sau đó pha trộn với một hình ảnh khác có một đường thẳng.Bây giờ tôi cần tính toán góc giữa hai dòng đó, nhưng tôi không biết tọa độ của các dòng đó.Vì vậy, tôi đã thử với mã cung cấp tọa độ của các đường thẳng đứng, nhưng nó không thể xác định cụ thể các tọa độ đó.Có cách nào để đo góc giữa những dòng đó không?Đây là mã tính toán tọa độ của tôi và hình ảnh pha trộn với hai dòng
import cv2 as cv import numpy as np src = cv.imread("blended2.png", cv.IMREAD_COLOR) if len(src.shape) != 2: gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY) else: gray = src gray = cv.bitwise_not(gray) bw = cv.adaptiveThreshold(gray, 255, cv.ADAPTIVE_THRESH_MEAN_C, cv.THRESH_BINARY, 15, -2) horizontal = np.copy(bw) vertical = np.copy(bw) cols = horizontal.shape[1] horizontal_size = int(cols / 30) horizontalStructure = cv.getStructuringElement(cv.MORPH_RECT, (horizontal_size, 1)) horizontal = cv.erode(horizontal, horizontalStructure) horizontal = cv.dilate(horizontal, horizontalStructure) cv.imwrite("img_horizontal8.png", horizontal) h_transpose = np.transpose(np.nonzero(horizontal)) print("h_transpose") print(h_transpose[:100]) rows = vertical.shape[0] verticalsize = int(rows / 30) verticalStructure = cv.getStructuringElement(cv.MORPH_RECT, (1, verticalsize)) vertical = cv.erode(vertical, verticalStructure) vertical = cv.dilate(vertical, verticalStructure) cv.imwrite("img_vertical8.png", vertical) v_transpose = np.transpose(np.nonzero(vertical)) print("v_transpose") print(v_transpose[:100]) img = src.copy() # edges = cv.Canny(vertical,50,150,apertureSize = 3) minLineLength = 100 maxLineGap = 200 lines = cv.HoughLinesP(vertical,1,np.pi/180,100,minLineLength,maxLineGap) for line in lines: for x1,y1,x2,y2 in line: cv.line(img,(x1,y1),(x2,y2),(0,255,0),2) cv.imshow('houghlinesP_vert', img) cv.waitKey(0)Tôi có hai dòng sẽ được vẽ qua các sự kiện chuột và để tìm góc giữa chúng như nếu tôi vẽ hai đường song song, góc cho tôi 0
Những gì tôi đã thử:
import numpy as np import cv2 import math btn_down = False def get_points(im): data = {} data['im'] = im.copy() data['lines'] = [] cv2.imshow("Image", im) cv2.setMouseCallback("Image", mouse_handler,data) cv2.waitKey(0) points = np.uint16(data['lines']) return points, data['im'] def mouse_handler(event, x, y, flags, data): global btn_down if event == cv2.EVENT_LBUTTONUP and btn_down: btn_down = False data['lines'][0].append((x, y)) cv2.circle(data['im'], (x, y), 3, (0, 0, 255),5) cv2.line(data['im'], data['lines'][0][0], data['lines'][0][1], (0,0,255), 2) cv2.imshow("Image", data['im']) elif event == cv2.EVENT_MOUSEMOVE and btn_down: image = data['im'].copy() cv2.line(image, data['lines'][0][0], (x, y), (0,0,0), 1) cv2.imshow("Image", image) elif event == cv2.EVENT_LBUTTONDOWN: btn_down = True data['lines'].insert(0,[(x, y)]) cv2.circle(data['im'], (x, y), 3, (0, 0, 255), 5, 16) cv2.imshow("Image", data['im']) def gradient(pt1, pt2): return (pt2[1] - pt1[1]) / (pt2[0] - pt1[0]) def getangle(pt1,pt2,pt3,pt4): pt1,pt2,pt3,pt4 = data['im'][-4:] print(f"point1: {pt1} \n point2: {pt2} \n point3 : {pt3}") m1 = gradient(pt1, pt2) m2 = gradient(pt1, pt3) angR = math.atan((m2 - m1) / (1 + (m2 * m1))) angD = math.degrees(angR) cv2.putText(final_image, str(angD), (pt1[0] - 20, pt1[1] - 20), cv2.FONT_HERSHEY_COMPLEX,1, (0, 0, 255), 2) print(f"angle: {angD} \n angleR: {angR}") img = cv2.imread('2.jpg', 1) pts, final_image = get_points(img) while True: if len(pts) % 4 == 0 and len(pts) != 0: getangle( data['im']) cv2.imshow('Image', final_image) print(pts[-2:]) cv2.waitKey(0)