FastRCNN 训练自己数据集 (1编译配置)

http://www.cnblogs.com/louyihang-loves-baiyan/p/4885659.html

按照博客的教程配置,但自己在服务器上配置时,USE_CUDNN = 1会报错,注释掉反而能正常运行。

添加hdf5路径按照教程添加:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_x64-linux-gnu/hdf5/serial

手动下载fast_rcnn_models模型会快很多,解压后将模型文件夹放到data目录下

直接运行demo.py会报错:no display name and no $DISPLAY environment variable

 

 

 

直接在服务器上成功的实现方式:backend还是TKAgg,在demo代码里加入import matplotlib  matplotlib.use('Agg')两句。

用自己电脑连接服务器使用,backend必须改成Agg才不报错,demo里依旧加入import matplotlib  matplotlib.use('Agg')两句,但不能正常显示图片。

 

 

https://my.oschina.net/swuly302/blog/94915

http://blog.csdn.net/oldjwu/article/details/5090549

http://www.jianshu.com/p/3f4b89aaf057

 

在编译过程中会出现如下警告:

服务器安装的是CUDA版本是8.0,从CUDA 8.0开始compute capability 2.0和2.1被弃用了,所以可以将Makefile.config中-gencode arch=compute_20,code=sm_20 和-gencode arch=compute_20,code=sm_21这两行删除,去除之后就没有警告了。
 

 

至于为何需要在lib文件夹make,然后在caffe文件夹make -j8 && make pycaffe,我认为是建立caffe的python接口。

posted @ 2017-06-05 15:01  有梦就要去实现他  阅读(452)  评论(0编辑  收藏  举报