YOLO算法有多种实现版本,论文中的作者的实现是在darknet框架下,可以参考链接:点击打开链接,darknet上已经更新到YOLO V2版本了。
这里主要讲Caffe版本的YOLO实现,主要采用yeahkun写的:点击打开链接,基本按照这个git里面的readme进行,但是因为整个流程操作起来步骤较多,所以将自己在调试过程中遇到的小问题记录如下:
大致步骤包括:1、编译。2、下载VOC数据集。3、生成list文件。4、生成LMDB文件。5、训练。6、测试
接下来详细展开
步骤1、下载项目并编译
这里说一下为什么要编译,可能你的电脑本来就已经编译好了CAFFE,其实原因是这样的,作者实现的时候可能自己定义了一些新的东西,比如新的层,所以需要重新将这些源文件编译,而不能直接用你原来电脑上编译好的CAFFE。另外编译也可以先进行,然后再去处理你的数据。
先把原来caffe-yolo-master目录下的Makefile.config.example复制并粘贴一份,并改名Makefile.config