opencv-yolo-tiny 实现车辆检测
opencv.dnn模块已经支持大部分格式的深度学习模型推理,该模块可以直接加载tensorflow、darknet、pytorch等常见深度学习框架训练出来的模型,并运行推理得到模型输出结果。opecnv.dnn模块已经作为一种模型部署方式,应用在工业落地实际场景中。
在现实中高速路口实时在线检测结果如下所示:
模型具体加载和使用流程如下:
1、加载网络,读取模型、网络结构配置等文件
2、创建输入,opencv.dnn模块对图片输入有特殊格式要求
3运行推理
4、解析输出
5、应用输出、显示输出
下面是opencv.dnn模块加载yolov3-tiny车辆检测模型并运行推理的代码:
import numpy as np
import cv2
import os
import time
from numpy import array# some variables
weights