第一步去github找到Faster-RCNN_TF的文件,进行下载。下载到本地后,解压完毕
1.cd 到根目录下的lib中 然后 make一下。
2.运行github中给的运行指令出现如下问题:
a. No module named easydict?
运行命令:pip install easydict
b. tensorflow.python.framework.errors_impl.NotFoundError:/home/hqd/Faster-RCNN_TF-car/tools/../li/roi_pooling_layer/roi_pooling.so: undefined symbol: _ZN10tensorflow7strings6StrCatB5cxx11ERKNS0_8AlphaNumE?
解决办法:修改make.sh文件
TF_LIB=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')
g++ -std=c++11 -shared -o roi_pooling.so roi_pooling_op.cc -D_GLIBCXX_USE_CXX11_ABI=0 \
roi_pooling_op.cu.o -I $TF_INC -L $TF_LIB -ltensorflow_framework -D GOOGLE_CUDA=1 \
-fPIC $CXXFLAGS -lcudart -L $CUDA_PATH/lib64
下面是我配置好的make.sh文件。你可以直接copy过去即可
TF_INC=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_include())