YOLOv2训练自己的数据集

今天我参考网上的方法在YOLOv2上面训练自己的数据集,我的数据集是车辆,就一个类别,其他方法我在这里就不赘述了,可以参考http://blog.csdn.net/ch_liu23/article/details/53558549和http://blog.csdn.net/ch_liu23/article/details/53558549

然后我在跑训练代码的时候遇到了无法载入图像的问题或者是cuda驱动退出的问题(darknet: ./src/cuda.c:36: check_error: Assertion `0' failed),一开始我以为是我的数据集有问题,我采用的是voc2007格式,不过我又参考了这篇博文,原来是因为我把路径搞错了,cfg/voc.data里面修改的路径应该是之前新生成的2007_train.txt的绝对路径,我还以为是voc2007里面的train.txt的绝对路径