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