1. 程式人生 > >FastRCNN 訓練自己數據集 (1編譯配置)

FastRCNN 訓練自己數據集 (1編譯配置)

backend key article tail back art model plot osc

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編譯配置)