使用YOLO网络对图像进行目标检测时，CUDA Error: out of memory

环境：Ubuntu14，CUDA8.0、Cudnn5.1、opencv3.1.0、

显卡：GTX1050 运行内存：2G

在darknet编译好后（GPU=1，CUDNN=1，OPENCV=1），下载好weights文件，试运行

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

报错：

CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

原因：1050运行内存太小，YOLOv3模型加载到GPU中导致内存溢出。

解决方法

gedit path_to/darknet/cfg/yolov3.cfg    #也可以先备份一下

修改配置文件

batch=1
subdivisions=1
width=416   
height=416     #参数可以根据自己的显卡性能修改，目的是减小占用内存

如果仍然不够，长短再各减32，以此类推

参考：https://github.com/pjreddie/darknet/issues/791