1.安装tensorflow
我的PC基本配置是:只有cpu,没有gpu(AMD显卡,对于深度学习相当于没有);系统是 ubuntu 14.04,自带的python2.7,6G内存。我这个PC机配置基本上接近底线了吧,呵呵。
源码安装tensorflow,我主要是参考:
tensorflow源码安装过程记录(ubuntu16、基于CPU) - CSDN博客作者写的图文并茂,非常详细,良心作者,非常感谢。
安装tensorflow遇到的错误:
(1)运行import tensorflow as tf,出现ImportError: No module named protobuf:
先卸载 pip uninstall protobuf
然后重新安装protobuf即可
(2)当运行 python test.py ,报错:
File "/home/thm/yolo_tensorflow/yolo/yolo_net.py", line 5, in <module>
slim = tf.contrib.slim
AttributeError: 'module' object has no attribute 'contrib'
原因是我的tensorflow之前是二进制安装,没有contrib模块,将tensorflow卸载,按照以上步骤源码安装tensorflow。
2.下载源代码及数据
(1)从github上下载源码:
hizhangp/yolo_tensorflow(2)下在训练数据以及模型参数:
在下载的源码文件夹中新建文件夹:data
在data中分别新建文件夹pascal_VOC、weights。目录结构如下图所示:
<img src="https://pic3.zhimg.com/v2-6b8867a11a2b2fd97ee1d9841df9a0c4_b.jpg" data-caption="" data-size="normal" data-rawwidth="372" data-rawheight="170" class="content_image" width="372">![](https://i-blog.csdnimg.cn/blog_migrate/f697525dc2720ec30e557bccf37faba2.png)
在pascal_voc放入训练数据集,下载链接:
链接: https://pan.baidu.com/s/1kWshVhl 密码: 89iu
在 weights 放入模型参数:
链接: https://pan.baidu.com/s/1htt9YBE 密码: ehw2
3. 测试
cd yolo_tensorflow
python test.py
运行无误时应该出现如下图片:
<img src="https://pic4.zhimg.com/v2-086ec07faed13c10384025fb357ec512_b.jpg" data-caption="" data-size="normal" data-rawwidth="638" data-rawheight="441" class="origin_image zh-lightbox-thumb" width="638" data-original="https://pic4.zhimg.com/v2-086ec07faed13c10384025fb357ec512_r.jpg">![](https://i-blog.csdnimg.cn/blog_migrate/c1f8cb8169dad7ae639f37592afd213b.png)
如遇到如下报错:
'module' object has no attribute 'CV_AA'
我将对应的 cv2.CV_AA改成了 cv2.LINE_AA即好了。
这是因为opencv版本的问题。