Linux GPU DRL-RPN-TF 部署

0. 必要库安装(建议阿里云)

python  3.6

easydict  1.9

gast  0.2.0

matplotlib  3.3.1

opencv-python  4.4.0.42

pyaml  20.4.0

scipy  1.2.1

scikit-image  0.17.2

tensorflow  1.14.0

Cython  0.25.2

1. git clone

git clone https://github.com/aleksispi/drl-rpn-tf.git

2. 修改配置

cd tf-faster-rcnn/lib
# Change the GPU architecture (-arch) if necessary
vim setup.py

# GPU 2080Ti 设置为 sm_75

3. Build the Cython modules

# 路径还在lib文件夹下
make clean
make
cd ..

4. 下载预训练模型(下第一个就好)

5. 设置权重文件路径

1. 第四步下载下来的权重文件解压后可以放置在本机的一个指定地方
2. 在drl-rpn-tf/experiments/scripts/test_drl_rpn.sh 中可以配置第四步下载权重文件的路径

6. 向drl-rpn-tf/data文件夹中放入数据集

# 这里采用VOCdevkit2007格式的数据集，当然也可以覆盖之前的软链接，这样就可以不移动文件夹了

7. 一些错误的修改

   1. 去除drl-rpn-tf/experiments/scripts/test_drl_rpn.sh中第70行和第82行中的time,修改后结果如下图所示

   2. 修改drl-rpn-tf/lib/datasets/voc_eval.py中第127行修改后结果如下图所示

   3. 修改/root/PycharmProjects/drl-rpn-tf/lib/model/factory.py，修改部分如下

   4. 将/root/PycharmProjects/drl-rpn-tf/lib/utils/statcoll.py中的第37行和第41行中除法改为整除,修改结果如下

8. test结果可视化,修改/root/PycharmProjects/drl-rpn-tf/experiments/cfgs/drl-rpn-vgg16.yml中的内容

9. 代码运行

./experiments/scripts/test_drl_rpn.sh 0 pascal_voc_0712 1 1 0

10. 一些个人喜好修改

修改/root/PycharmProjects/drl-rpn-tf/lib/datasets/pascal_voc.py中第46行

如果这里的self._image_ext设置为'',那么‘./experiments/scripts/test_drl_rpn.sh 0 pascal_voc_0712 1 1 0’时的/root/PycharmProjects/drl-rpn-tf/data/VOCdevkit2007/VOC2007/ImageSets/Main/test.txt文件中的文件名就需要.jpg后缀

反之如果设置self._image_ext设置为'.jpg',那么‘./experiments/scripts/test_drl_rpn.sh 0 pascal_voc_0712 1 1 0’时的/root/PycharmProjects/drl-rpn-tf/data/VOCdevkit2007/VOC2007/ImageSets/Main/test.txt文件中的文件名就不需要.jpg后缀