本人深度學習小白,這兩天被安排任務做目標檢測,初次探究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)差不多。