首页 > 代码库 > FastRCNN 训练自己数据集 (1编译配置)
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
FastRCNN 训练自己数据集 (1编译配置)