把玩 YOLOv4

動機:想要玩 Darknet(YOLOv4) ,要如何安裝及實作呢?!

準備環境
1.Windows 10
2.CUDA 10.0 + CUDNN 7.6.5
3.Darknet
4.OpenCV 3.4.10
5.CMake 3.17.2
6.VS2015 (VC++)
7.Geforce RTX 2070

實作步驟
1.直接請 參攷[1.] 的說明,將 cuda、cudnn 及 OpenCV 都安裝好
2.下載 YOLOv4 源碼 ( 指令> git clone https://github.com/AlexeyAB/darknet.git ) 儲存目錄名稱為 darknet
3.下載安裝 CMake (官網 https://cmake.org/download/),並選擇前述 darknet 目錄及陸續執行 [Configure]、[Generate]、[Open Project]...即可順利開啟VS2015及產生 darknet.exe 執行檔了...
3.接著執行經典的圖片檢測範例...( 指令> darknet.exe detect cfg\yolov4.cfg yolov4.weights data\dog.jpg )
就會看到Darknet作者的自行車及狗了...如下圖

心得
YOLOv4的安裝實在是太方便及快速,想必在Deep Learning上可讓人無藉口可說了!!
另外,在 Ubuntu 上的安裝就不用再多說了...這是一定要玩的!!

參攷
1.使用Cmake-GUI在windows环境下编译yolov4, https://zhuanlan.zhihu.com/p/137485827

留言