基于Jetson TX1的R-FCN的算法搭建

Data: 2017.03.13

Author: cjh

Theme: Set up the R-FCN environment

 

由于毕设要求最近都在开发板上尝试搭建各种深度学习的算法,由于JetsonTX1对于caffe的支持还不错,同时在整个过程中也遇到了很多的问题和错误,在这里和对此刚兴趣的朋友一起交流交流。

 

具体文章:我上传到我的的资源区,有需要的朋友可以直接去下载

 

注意:在搭建该算法前,先参考本博客中有关caffe环境的搭建,否则可能出现较多的错误

      同时,该算法的内存消耗较大,由于Jetson TX1只有4GB的内存,而本来要求的硬件系统:

Requirements: hardware

Any NVIDIA GPU with 6GB or larger memoryis OK(4GB is enough for ResNet-50).

所以Jetson TX1 勉强能跑ResNet-50,有时候跑到最后会出现Killed,就不要怀疑自己哪里装错了

 

作者采用的matlab版本的代码,本文针对的是python接口的算法

 

步骤1:下载代码

git clone https://github.com/Orpine/py-R-FCN.git

 

步骤2:下载caffe代码

cd py-R-FCN

git clone https://github.com/Microsoft/caffe.git

#该步骤可选具体我也不知道有什么用

cd caffe

git reset --hard 1a2be8e 

 

步骤3:编译Cython

cd py-R-FCN/lib

make

 

步骤4:编译caffe和pycaffe

cd py-R-FCN/caffe

cp Makefile.config.example Makefile.config

vim Makefile.config  #修改配置文件

USE_CUDNN := 1

CUDA_DIR := /usr/local/cuda-8.0

WITH_PYTHON_LAYER := 1

INCLUDE_DIRS := $(PYTHON_INCLUDE)/usr/local/include /usr/include/hdf5/serial

LIBRARY_DIRS := $(PYTHON_LIB)/usr/local/lib /usr/lib /usr/lib/aarch64-linux-gnu/hdf5/serial

退出并保存

 

make -j8 && make pycaffe

 

步骤5:测试

下载resnet caffemodel

建议在windows上下载,因为文件比较大

https://1drv.ms/u/s!AoN7vygOjLIQqUWHpY67oaC7mopf

将下载好的文件解压到

py-R-FCN/data/rfcn_models/resnet50_rfcn_final.caffemodel

py-R-FCN/data/rfcn_models/resnet101_rfcn_final.caffemodel

 

最后运行demo

cd py-R-FCN/tools/

python demo_rfcn.py --net ResNet-50

python demo_rfcn.py --net ResNet-101

 

效果图我这里就不贴了了,能告诉你的是,跑完后,识别后的图片会自己打开的,如果没有打开就说明你的安装出现了问题,如果出现Killed说明是你的开发板内存不够!

你可能感兴趣的:(Jetson,TX1)