听说拍人家违章月入十万？那我用python写个自动检测车辆是否违章不是发财了。。。

可口可乐没有乐

已于 2024-01-10 15:30:27 修改

阅读量230

点赞数

分类专栏： python 文章标签： python opencv 计算机视觉开发语言爬虫

于 2023-02-11 17:16:00 首次发布

本文链接：https://blog.csdn.net/m0_59236602/article/details/128985900

版权

python 专栏收录该内容

395 篇文章 6 订阅

订阅专栏

导语

SO 由于对考试，车辆有了执着，所以学习以及今天教大家的也是关于基于opencv的车辆检测系统！！！

正文

想想看，如果你能在红绿灯摄像头中集成车辆检测系统，你可以轻松地同时跟踪许多有用的东西：

白天交通路口有多少辆车？
什么时候交通堵塞？
什么样的车辆（重型车辆、汽车等）正在通过交叉路口？
有没有办法优化交通，并通过不同的街道进行分配？
还有很多例子就不一一列举。应用程序是无止境的~

首先环境安装：

我们先导入所需的库和模块—— opencv安装：pip install opencv-python

import os
import re
import cv2 # opencv library
import numpy as np
from os.path import isfile, join
import matplotlib.pyplot as plt

将框架保存在工作目录中的文件夹以及导入帧并保存：

# get file names of the frames
col_frames = os.listdir('frames/')
 
# sort file names
col_frames.sort(key=lambda f: int(re.sub('\D', '', f)))
 
# empty list to store the frames
col_images=[]
 
for i in col_frames:
    # read the frames
    img = cv2.imread('frames/'+i)
    # append the frames to the list
    col_images.append(img)
让我们显示两个连续的帧：

# plot 13th frame
i = 13
 
for frame in [i, i+1]:
    plt.imshow(cv2.cvtColor(col_images[frame], cv2.COLOR_BGR2RGB))
    plt.title("frame: "+str(frame))
    plt.show()

获取两个连续帧的像素值的差值将有助于我们观察移动目标。那么，让我们在上面两个帧上使用该技术：

# convert the frames to grayscale
grayA = cv2.cvtColor(col_images[i], cv2.COLOR_BGR2GRAY)
grayB = cv2.cvtColor(col_images[i+1], cv2.COLOR_BGR2GRAY)
 
# plot the image after frame differencing
plt.imshow(cv2.absdiff(grayB, grayA), cmap = 'gray')
plt.show()

现在我们可以清楚地看到第13帧和第14帧中的移动目标。其他没有移动的东西都被减去了。

图像预处理——为所有帧中的所有移动车辆添加了轮廓：

# specify video name
pathOut = 'vehicle_detection_v3.mp4'
 
# specify frames per second
fps = 14.0
接下来阅读列表中的最后一帧：

frame_array = []
files = [f for f in os.listdir(pathIn) if isfile(join(pathIn, f))]
files.sort(key=lambda f: int(re.sub('\D', '', f)))
 
for i in range(len(files)):
    filename=pathIn + files[i]
 
    #read frames
    img = cv2.imread(filename)
    height, width, layers = img.shape
    size = (width,height)
 
    #inserting the frames into an image array
    frame_array.append(img)

最后使用以下代码制作目标检测视频：

out = cv2.VideoWriter(pathOut,cv2.VideoWriter_fourcc(*'DIVX'), fps, size)
 
for i in range(len(frame_array)):
    # writing to a image array
    out.write(frame_array[i])
 
out.release()

好啦！你学会了嘛？