封裝yolo-windows為動態連結庫

因為yolo的原始碼是c語言寫的,所以移植到windows下還不算很難,就是要修改的地方比較多,後來在github上找到別人修改好的程式碼:

https://github.com/AlexeyAB/yolo-windows

在此基礎上做了一些修改,並將程式碼封裝成了動態庫。主要的修改為:opencv更換為2.4.10版本;標籤從txt檔案中讀取並且可以是中文(注意編碼ANSI);每次測試一個資料夾。

(1)GPU OPENCV版本

其中,CUDA版本為7.5,Opencv版本為2.4.10,配置好CUDA和Opencv後,開啟專案即可執行(VS2013)。

封裝好的程式碼下載地址:

連結:http://pan.baidu.com/s/1jIa2WWu 密碼:a8al  (檔案中帶測試模型,所以比較大)

GPU(我的顯示卡是GTX970)版本處理每張圖片約0.05s,效果如下:

(2)CPU OPENCV版本

Opencv版本為2.4.10.

封裝好的程式碼下載地址:

連結:http://pan.baidu.com/s/1o8MxVq6 密碼:gruk

CPU版本處理每張圖片約10s(可能我的CPU比較差),和GPU相比慢很多,所以不建議使用。效果如下:

如果使用GPU版本,顯示卡可能要求比較高,否則可能出現out of memory的情況。

封裝原始碼下載地址:

http://download.csdn.net/detail/sinat_30071459/9688381 (bin加入環境變數)

標籤已經修改為中文,如果測試自己的模型,修改程式碼中四個變數即可。如果你的標籤也需要修改成中文,可以使用上面程式碼中data/labels/make_labels_all.py生成標籤圖片。

每次可以測試整個資料夾的圖片,按空格下一張。

(以上程式碼均是在win7 64位,vs2013下封裝)