如何在OpenCV中使用YOLO

點(diǎn)擊上方“小白學(xué)視覺”，選擇加"星標(biāo)"或“置頂”

重磅干貨，第一時(shí)間送達(dá)

今天，我們將研究如何在OpenCV框架中使用YOLO。YOLO于2016年問世，用于多目標(biāo)檢測(cè)，它與OpenCV框架兼容，但我們需要下載“ yolov3.weights”和“yolov3.cfg”。

現(xiàn)在讓我們來看一下代碼，它相當(dāng)簡單。第一步將是導(dǎo)入模型并讀取包含圖像標(biāo)簽的“coco.names”并獲取輸出層。

下一步是讀取輸入圖像，并創(chuàng)建Blob從輸入圖像中提取特征。圖像的輸入尺寸為416 * 416，（0,0,0）表示圖像的色彩空間。

我們將遍歷該blob并找出已檢測(cè)到的對(duì)象。但是在此之前，我們必須將blob饋給yolo算法并從輸出層提取其特征。我們可以將其與CNN模型相關(guān)聯(lián)。才外，我們還對(duì)置信度預(yù)測(cè)超過50％的對(duì)象感興趣。

挑戰(zhàn)在于分離算法檢測(cè)到的冗余對(duì)象。最后，我們可以創(chuàng)建一個(gè)邊界框并顯示圖像。

希望本文對(duì)大家理解我們?nèi)绾卧贠penCV框架中使用YOLO有所幫助。

下載1：OpenCV-Contrib擴(kuò)展模塊中文版教程

在「小白學(xué)視覺」公眾號(hào)后臺(tái)回復(fù)：擴(kuò)展模塊中文教程，即可下載全網(wǎng)第一份OpenCV擴(kuò)展模塊教程中文版，涵蓋擴(kuò)展模塊安裝、SFM算法、立體視覺、目標(biāo)跟蹤、生物視覺、超分辨率處理等二十多章內(nèi)容。

下載2：Python視覺實(shí)戰(zhàn)項(xiàng)目31講

在「小白學(xué)視覺」公眾號(hào)后臺(tái)回復(fù)：Python視覺實(shí)戰(zhàn)項(xiàng)目31講，即可下載包括圖像分割、口罩檢測(cè)、車道線檢測(cè)、車輛計(jì)數(shù)、添加眼線、車牌識(shí)別、字符識(shí)別、情緒檢測(cè)、文本內(nèi)容提取、面部識(shí)別等31個(gè)視覺實(shí)戰(zhàn)項(xiàng)目，助力快速學(xué)校計(jì)算機(jī)視覺。

下載3：OpenCV實(shí)戰(zhàn)項(xiàng)目20講

在「小白學(xué)視覺」公眾號(hào)后臺(tái)回復(fù)：OpenCV實(shí)戰(zhàn)項(xiàng)目20講，即可下載含有20個(gè)基于OpenCV實(shí)現(xiàn)20個(gè)實(shí)戰(zhàn)項(xiàng)目，實(shí)現(xiàn)OpenCV學(xué)習(xí)進(jìn)階。

交流群

歡迎加入公眾號(hào)讀者群一起和同行交流，目前有SLAM、三維視覺、傳感器、自動(dòng)駕駛、計(jì)算攝影、檢測(cè)、分割、識(shí)別、醫(yī)學(xué)影像、GAN、算法競賽等微信群（以后會(huì)逐漸細(xì)分），請(qǐng)掃描下面微信號(hào)加群，備注：”昵稱+學(xué)校/公司+研究方向“，例如：”張三?+?上海交大?+?視覺SLAM“。請(qǐng)按照格式備注，否則不予通過。添加成功后會(huì)根據(jù)研究方向邀請(qǐng)進(jìn)入相關(guān)微信群。請(qǐng)勿在群內(nèi)發(fā)送廣告，否則會(huì)請(qǐng)出群，謝謝理解~