YoLOV3初訓練

本人深度學習小白,這兩天被安排任務做目標檢測,初次探究YoLOV3,僅在此記下學習經(jīng)驗,防止以后少入坑。

本人首先是跟著官網(wǎng)走一遍的:YOLO: Real-Time Object Detection

首先,新建一個文件夾,存放:mkdir yolov3

cd yolov3

執(zhí)行:


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

cd darknet

make


下載預訓練權(quán)值


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


最后


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


當訓練跑起來的時候,發(fā)現(xiàn)一個問題,就是官網(wǎng)上的時間只有0.029329 seconds。而本人的時間卻用了0.029329 seconds?。。?!

原因:沒開GPU,CUDNN。打開darknet文件夾下的Makefile文件夾,修改里面的參數(shù)設置。本人剛開始只將GPU=1

后,重新make:


make clean

make

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


此時出現(xiàn)了一問題,訓練出錯了??!如下提示:

./darknet: error while loading shared libraries: libcudart.so.8.0: cannot open shared object file: No such file or directory

貌似是cuda8.0鏈接的問題,找到了解決方案:


sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig

sudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig

sudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig

參考自:http://blog.csdn.net/kaka20080622/article/details/40706813


最后再重新make后訓練所用的時間為0.037161 seconds,與官網(wǎng)差不多。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容