OpenVINO + OpenCV实现车辆检测与道路分割

点击上方“小白学视觉”，选择加"星标"或“置顶”

重磅干货，第一时间送达

本文转自|OpenCV学堂

模型介绍

OpenVINO支持道路分割与车辆检测，预训练模型分别为：

- road-segmentation-adas-0001
- vehicle-detection-adas-0002

其中道路分割模型的输出四个分类，格式如下：

BG, road, curb, mark， 输出格式[NCHW]=[1x4x512x896]

车辆检测模型基于SSD MobileNetv1实现，输出格式为：

NCHW = [1x1xNx7],其中N表示检测到boxes数目

代码演示

01

道路分割模型加载与推理

首先加载道路分割模型，代码如下：

 1# 道路分割
 2net = ie.read_network(model=model_xml, weights=model_bin)
 3input_blob = next(iter(net.input_info))
 4out_blob = next(iter(net.outputs))
 5
 6n, c, h, w = net.input_info[input_blob].input_data.shape
 7print(n, c, h, w)
 8
 9cap = cv.VideoCapture("D:/images/video/project_video.mp4")
10exec_net = ie.load_network(network=net, device_name="CPU")

推理与解析

 1# 推理道路分割 2image = cv.resize(frame, (w, h))
 3image = image.transpose(2, 0, 1)
 4res = exec_net.infer(inputs={input_blob: [image]})
 5
 6# 解析道路分割结果
 7res = res[out_blob]
 8res = np.squeeze(res, 0)
 9res = res.transpose(1, 2, 0)
10res = np.argmax(res, 2)
11print(res.shape)
12hh, ww = res.shape
13mask = np.zeros((hh, ww, 3), dtype=np.uint8)
14mask[np.where(res > 0)] = (0, 255, 255)
15mask[np.where(res > 1)] = (255, 0, 255)
16mask = cv.resize(mask, (frame.shape[1], frame.shape[0]))
17result = cv.addWeighted(frame, 0.5, mask, 0.5, 0)

02

车辆检测模型加载与推理解析

加载车辆检测模型，推理与解析SSD输出结果的代码如下：

 1# 车辆检测
 2vnet = ie.read_network(model=vehicel_xml, weights=vehicel_bin)
 3vehicle_input_blob = next(iter(vnet.input_info))
 4vehicle_out_blob = next(iter(vnet.outputs))
 5
 6vn, vc, vh, vw = vnet.input_info[vehicle_input_blob].input_data.shape
 7print(n, c, h, w)
 8vehicle_exec_net = ie.load_network(network=vnet, device_name="CPU")
 9
10# 车辆检测
11inf_start = time.time()
12image = cv.resize(frame, (vw, vh))
13image = image.transpose(2, 0, 1)
14vec_res = vehicle_exec_net.infer(inputs={vehicle_input_blob:[image]})
15
16# 解析车辆检测结果
17ih, iw, ic = frame.shape
18vec_res = vec_res[vehicle_out_blob]
19for obj in vec_res[0][0]:
20    if obj[2] > 0.5:
21        xmin = int(obj[3] * iw)
22        ymin = int(obj[4] * ih)
23        xmax = int(obj[5] * iw)
24        ymax = int(obj[6] * ih)
25        cv.rectangle(frame, (xmin, ymin), (xmax, ymax), (0, 0, 255), 2, 8)
26        cv.putText(frame, str(obj[2]), (xmin, ymin), cv.FONT_HERSHEY_PLAIN, 1.0, (0, 0, 255), 1)

运行结果如下：

视频文件

下载1：OpenCV-Contrib扩展模块中文版教程

在「小白学视觉」公众号后台回复：扩展模块中文教程，即可下载全网第一份OpenCV扩展模块教程中文版，涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。

下载2：Python视觉实战项目52讲

在「小白学视觉」公众号后台回复：Python视觉实战项目，即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目，助力快速学校计算机视觉。

下载3：OpenCV实战项目20讲

在「小白学视觉」公众号后台回复：OpenCV实战项目20讲，即可下载含有20个基于OpenCV实现20个实战项目，实现OpenCV学习进阶。

交流群

欢迎加入公众号读者群一起和同行交流，目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群（以后会逐渐细分），请扫描下面微信号加群，备注：”昵称+学校/公司+研究方向“，例如：”张三 + 上海交大 + 视觉SLAM“。请按照格式备注，否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告，否则会请出群，谢谢理解~