天天看點

Linux環境下Darknet的安裝與測試

1.下載下傳Darknet源碼

sudo git clone https://github.com/pjreddie/darknet
           

2.編譯Darknet源碼

cd darknet

make
           

3.下載下傳權重檔案yolov3.weights,放入到darknet檔案夾下

下載下傳位址:

https://pjreddie.com/media/files/yolov3.weights
           

4.修改cfg檔案夾下的yolov3.cfg檔案,原檔案是訓練模式,需要改成測試模式。

具體修改方法是,打開Testing下的兩個參數,登出掉Training下的兩個參數

Linux環境下Darknet的安裝與測試

5.運作網絡,檢視檢測效果

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
           

可以看出網絡運作時間非常慢,要12s多。這主要是因為目前網絡是在CPU上運作,沒有用到GPU。

Linux環境下Darknet的安裝與測試

6.修改Makefile檔案,打開GPU的開關,将GPU=0改為GPU=1,并重新編譯。

Linux環境下Darknet的安裝與測試

7.再次執行

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
           

這下速度快多了

Linux環境下Darknet的安裝與測試

同時可以看出,GPU跑起來了。

Linux環境下Darknet的安裝與測試

繼續閱讀