我的电脑配置:(不同的显卡会对应不同的参数,大家看好自己的显卡型号)
1.首先安装NVIDIA驱动、CUDA9.0、cuDNN:
https://blog.csdn.net/debjiu/article/details/80824439
注意:
(1)在安装NVIDIA驱动时,使用ubuntu-drivers devices
命令,来检测推荐的驱动版本,尽量安装推荐的驱动版本。
我是按照上面链接进行的,直接自动安装的显卡驱动,安装完之后才发现可以手动下载需要的版本
sudo apt-get install nvidia-430
安装好驱动后记得重启电脑,使得新的显卡驱动生效。
(2)安装配置cuDNN我参考的下面的链接:
https://blog.csdn.net/dihuanlai9093/article/details/79253963/
2.下载YOLOv3,配置GPU修改Makefile文件 :
https://blog.csdn.net/john_bh/article/details/80625220
https://www.cnblogs.com/yumoye/p/10548714.html
GPU的compute capability列表:
https://developer.nvidia.com/cuda-gpus
3.完成之后可参照下面的链接运行demo:
https://pjreddie.com/darknet/yolo/
YOLOv3论文: https://pjreddie.com/media/files/papers/YOLOv3.pdf
我在使用YOLOv3摄像头实时检测时,FPS总是在10左右,出现视频卡顿。
解决办法: 打开cfg/yolov3.cfg,修改其中的图片大小即可。
(1)当图像大小在608*608时,FPS在10左右。
(2)当图像大小在416*416时,FPS在20左右。
(3)当图像大小在320*320时,FPS在30左右。
图像越大,准确率越高,识别速度越慢。反之。 图像大小必须为32的倍数!
yolov3.cfg参数详解: https://www.cnblogs.com/zhangAlin/p/10661768.html