车牌检测业务介绍
车牌检测应用业务完成通过以下功能单元编排完成:
- 通过RTSP地址输入一段视频,经过视频解码功能单元后变成为图片;
- 将图片经过预处理,包括缩放(Resize)、颜色转置(ColorTranspose)和归一化(Normalize);
- 使用Yolo模型进行推理,得到图片中所有的车牌的框数据和置信度,再通过Yolo后处理可以得到车牌的检测框;
- 使用绘框功能单元,将框绘制到相应的图片上;
- 使用视频编码功能单元,将绘框后图片编码成视频流并输出。
功能单元编排流程图如图1所示。
用户这里可能已有代码实现,其大致的伪代码如下所示:
in = VideoReader('rtsp:://in')
out = VideoWriter('rtsp:://out')
while(in.isOpened()):
pic = cap.read()
frame = Resize(pic)
frame = ColorTranspose(frame)
frame = Normlize(frame)
plate_model = loadModel(plate.pb)
bbox,pred = plate_model.run(frame)
bbox_reuslts = yolo(bbox,pred)
draw_pic = pic
for(bbox_reuslt in bbox_reuslts)
draw_pic = draw_bbox(draw_pic,bbox_reuslt)
out.write(draw_pic)