OpenCV+CUDA編譯實(shí)現(xiàn)YOLOv5能加速

OpenCV開(kāi)發(fā)者基本技能之一就是要從源碼編譯OpenCV生成各種裁剪版本的OpenCV庫(kù)，同時(shí)根據(jù)需要編譯源碼生產(chǎn)支持CUDA加速版本的OpenCV庫(kù)。 但是，很多開(kāi)發(fā)者還是編譯中會(huì)遇到各種問(wèn)題，被迫放棄！可以說(shuō)還沒(méi)用CUDA加速就已經(jīng)自我放棄啦! 所以周末我又重新編譯了一遍，針對(duì)各種問(wèn)題，幫大家理清了對(duì)策，幫助大家可以完成OpenCV+CUDA編譯，實(shí)現(xiàn)性能加速！ 軟件版本信息：

OpenCV+CUDA編譯

整個(gè)編譯過(guò)程主要分為三步：

下載源碼，一定是Tag對(duì)應(yīng)的版本跟擴(kuò)展模塊CMake階段，解決無(wú)法下載的坑！去掉不必要的模塊VS工程編譯，生成lib與dll

下載源碼：

選擇tag-4.5.4，

CMake：

打開(kāi)CMake，設(shè)置源碼路徑跟目標(biāo)路徑：

上述選擇表示編譯為64位的庫(kù)！

點(diǎn)擊【Configure】,

然后再點(diǎn)擊【generate】

完成之后，搜索cuda關(guān)鍵字，如上圖，三個(gè)選項(xiàng)勾上，

然后再搜索opencv_ex, 設(shè)置擴(kuò)展模塊的代碼路徑，如上圖！設(shè)置好之后再次點(diǎn)擊【Configure】,完成之后:

搜索cuda，如上圖勾選！ 最后搜索：

默認(rèn)是勾選的，全部不要選擇！(去掉勾選！)

無(wú)法下載ffmpeg, ippicv問(wèn)題解決：

從上述兩個(gè)文件拿到下載地址，直接下載完成之后，分別放到對(duì)應(yīng)目錄：

就一切OK了。

生成好VS工程文件之后，打開(kāi)工程文件：

謝天謝地，終于編譯好啦！

OpenCV+CUDA配置與加速

基于最新編譯好的OpenCV+CUDA支持的庫(kù)，重新配置OpenCV開(kāi)發(fā)環(huán)境，對(duì)比YOLOv5部署，CPU運(yùn)行推理測(cè)試結(jié)果如下：

添加兩行代碼，啟用GPU運(yùn)行推理測(cè)試結(jié)果如下：

顯卡是3050ti！

對(duì)比一下，加速效果真得是杠杠滴！所以值得編譯OpenCV+CUDA支持，因?yàn)樗还饧铀?a href="http://srfitnesspt.com/v/tag/448/" target="_blank">深度學(xué)習(xí)模型推理，對(duì)傳統(tǒng)圖像處理均有加速！

審核編輯：劉清