python+pqt5利用Sobel算子检测动态视频,彩色转灰色图像图像去噪,边缘检测。生成Mask掩码,直线检测绘制车道线
时间: 2024-04-01 07:31:36 浏览: 113
下面是一个基于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.cap = cv2.VideoCapture('video.mp4')
# 获取视频帧率和尺寸
self.fps = int(self.cap.get(cv2.CAP_PROP_FPS))
self.frame_size = (int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))
# 创建视频显示标签
self.label = QLabel(self)
self.label.setFixedSize(*self.frame_size)
# 设置布局
self.layout = QVBoxLayout()
self.layout.addWidget(self.label)
self.setLayout(self.layout)
# 创建定时器
self.timer = QTimer(self)
self.timer.setInterval(1000 // self.fps)
self.timer.timeout.connect(self.update_frame)
self.timer.start()
def update_frame(self):
# 读取视频帧
ret, frame = self.cap.read()
if ret:
# 将彩色图像转换为灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 对灰度图像进行高斯模糊
blur = cv2.GaussianBlur(gray, (5, 5), 0)
# 对模糊后的图像应用Sobel算子,计算每个像素在x方向和y方向的梯度
sobelx = cv2.Sobel(blur, cv2.CV_64F, 1, 0, ksize=3)
sobely = cv2.Sobel(blur, cv2.CV_64F, 0, 1, ksize=3)
# 将x方向和y方向的梯度值组合成一个梯度幅值图像
magnitude = np.sqrt(sobelx ** 2 + sobely ** 2)
magnitude = np.uint8(magnitude / np.max(magnitude) * 255)
# 对梯度幅值图像进行阈值处理,将低于阈值的像素设为0,高于阈值的像素设为255
threshold = 50
mask = np.zeros_like(magnitude)
mask[magnitude > threshold] = 255
# 对二值化图像进行形态学操作,如腐蚀和膨胀,以去除噪声和平滑边缘
kernel = np.ones((5, 5), np.uint8)
mask = cv2.erode(mask, kernel, iterations=1)
mask = cv2.dilate(mask, kernel, iterations=1)
# 使用霍夫变换检测直线,找到车道线的位置和方向
lines = cv2.HoughLinesP(mask, rho=1, theta=np.pi/180, threshold=20, minLineLength=50, maxLineGap=10)
# 绘制直线
line_image = np.zeros_like(frame)
for line in lines:
x1, y1, x2, y2 = line[0]
cv2.line(line_image, (x1, y1), (x2, y2), (0, 0, 255), 2)
# 将绘制好的直线和原始图像叠加
result = cv2.addWeighted(frame, 0.8, line_image, 1, 0)
# 显示结果
self.label.setPixmap(QPixmap.fromImage(self.convert_image(result)))
self.label.setAlignment(Qt.AlignCenter)
else:
self.timer.stop()
self.cap.release()
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_()
```
在上述代码中,我们首先打开一个视频文件,并获取其帧率和尺寸。然后,我们创建一个视频显示标签,并设置布局。接下来,我们创建一个定时器,每隔一定时间读取一帧视频,并对其进行彩色转灰色、去噪、边缘检测、生成Mask掩码和直线检测绘制车道线。最后,我们将绘制好的车道线和原始帧叠加,并在视频显示标签中显示结果。
运行代码后,我们可以看到车道线的实时检测结果:
![Lane_detection_video](https://i.postimg.cc/rwvMvJL8/lane-detection-video.gif)
阅读全文