python+pqt5利用Sobel算子彩色转灰色图像图像去噪,边缘检测。生成Mask掩码,直线检测绘制车道线
时间: 2024-04-01 17:31:42 浏览: 104
下面是一个基于Python和PyQt5的实现,可以实现利用Sobel算子彩色转灰色图像、去噪、边缘检测、生成Mask掩码和直线检测绘制车道线的功能:
```python
import cv2
import numpy as np
from PyQt5.QtWidgets import QWidget, QLabel, QVBoxLayout, QApplication
from PyQt5.QtGui import QImage, QPixmap
class LaneDetectionWidget(QWidget):
def __init__(self):
super().__init__()
# 读取图像
self.image = cv2.imread('lane.jpg')
# 将彩色图像转换为灰度图像
self.gray = cv2.cvtColor(self.image, cv2.COLOR_BGR2GRAY)
# 对灰度图像进行高斯模糊
self.blur = cv2.GaussianBlur(self.gray, (5, 5), 0)
# 对模糊后的图像应用Sobel算子,计算每个像素在x方向和y方向的梯度
self.sobelx = cv2.Sobel(self.blur, cv2.CV_64F, 1, 0, ksize=3)
self.sobely = cv2.Sobel(self.blur, cv2.CV_64F, 0, 1, ksize=3)
# 将x方向和y方向的梯度值组合成一个梯度幅值图像
self.magnitude = np.sqrt(self.sobelx ** 2 + self.sobely ** 2)
self.magnitude = np.uint8(self.magnitude / np.max(self.magnitude) * 255)
# 对梯度幅值图像进行阈值处理,将低于阈值的像素设为0,高于阈值的像素设为255
self.threshold = 50
self.mask = np.zeros_like(self.magnitude)
self.mask[self.magnitude > self.threshold] = 255
# 对二值化图像进行形态学操作,如腐蚀和膨胀,以去除噪声和平滑边缘
self.kernel = np.ones((5, 5), np.uint8)
self.mask = cv2.erode(self.mask, self.kernel, iterations=1)
self.mask = cv2.dilate(self.mask, self.kernel, iterations=1)
# 使用霍夫变换检测直线,找到车道线的位置和方向
self.lines = cv2.HoughLinesP(self.mask, rho=1, theta=np.pi/180, threshold=20, minLineLength=50, maxLineGap=10)
# 绘制直线
self.line_image = np.zeros_like(self.image)
for line in self.lines:
x1, y1, x2, y2 = line[0]
cv2.line(self.line_image, (x1, y1), (x2, y2), (0, 0, 255), 2)
# 将绘制好的直线和原始图像叠加
self.result = cv2.addWeighted(self.image, 0.8, self.line_image, 1, 0)
# 显示结果
self.label = QLabel(self)
self.label.setPixmap(QPixmap.fromImage(self.convert_image(self.result)))
self.label.setAlignment(Qt.AlignCenter)
self.layout = QVBoxLayout()
self.layout.addWidget(self.label)
self.setLayout(self.layout)
def convert_image(self, image):
h, w, c = image.shape
bytesPerLine = c * w
return QImage(image.data, w, h, bytesPerLine, QImage.Format_BGR888)
if __name__ == '__main__':
app = QApplication([])
widget = LaneDetectionWidget()
widget.show()
app.exec_()
```
在上述代码中,我们首先读取一张车道线的图像,然后将其转换为灰度图像,并对其进行高斯模糊。接下来,我们使用Sobel算子计算每个像素在x方向和y方向的梯度,将其组合成一个梯度幅值图像,并对其进行阈值处理,生成Mask掩码。然后,我们对Mask掩码进行形态学操作,如腐蚀和膨胀,以去除噪声和平滑边缘。最后,我们使用霍夫变换检测直线,找到车道线的位置和方向,并绘制直线。最后,我们将绘制好的直线和原始图像叠加,显示结果。
运行代码后,我们可以看到绘制好的车道线:
![Lane_detection_result](https://i.postimg.cc/pLj3ZfW2/lane-detection-result.png)
阅读全文