ubuntu18.04安装yoloV

3

1. 安装CUDA和cudnn

首先需要安装CUDA和cudnn，这里不再赘述，可以参考官方文档进行安装。

2. 安装OpenCV

sudo apt-get update
sudo apt-get install libopencv-dev

3. 下载yoloV3

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

4. 编译

cd darknet
mkdir build_release
cd build_release
cmake -DOpenCV_DIR=/usr/local/share/OpenCV -DGPU=ON -DCMAKE_INSTALL_PREFIX=../install_release ..
make -j8
make install

5. 测试

下载权重文件

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

测试图片

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

6. 可能出现的问题

- 缺少libopencv_highgui.so.3.3

解决方法：

sudo apt-get install libopencv-highgui-dev

- 缺少libopencv_videoio.so.3.3

解决方法：

sudo apt-get install libopencv-videoio-dev

- 缺少libopencv_imgcodecs.so.3.3

解决方法：

sudo apt-get install libopencv-imgcodecs-dev

- 缺少libcudart.so.10.0

解决方法：

sudo ln -s /usr/local/cuda-10.0/lib64/libcudart.so.10.0 /usr/lib/x86_64-linux-gnu/libcudart.so.10.0

- 缺少libcudnn.so.7

解决方法：

sudo ln -s /usr/local/cuda-10.0/lib64/libcudnn.so.7.0.5 /usr/lib/x86_64-linux-gnu/libcudnn.so.7

7. 其他注意事项

- 如果想要使用CPU版本，需要将-DGPU=ON改为-DGPU=OFF。
- 如果想要使用其他版本的yolo，可以将下载的yoloV3替换为其他版本的yolo。