Linux GPU DRL-RPN-TF 部署
程序员文章站
2022-04-30 22:42:38
...
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后缀
推荐阅读
-
Linux部署msmtp+mutt发送邮件功能
-
详解在LINUX上部署带有JAR包的JAVA项目
-
Linux工具之Nethogs按进程监控网络带宽的安装部署
-
linux系统中轻量级内核虚拟化技术LXC安装部署实践
-
在虚拟机Linux上部署DB2pureScale实践过程
-
Linux下实现SSH免密码登录和实现秘钥的管理、分发、部署SHELL脚本分享
-
.net core3.0部署Linux服务器 使用Docker容器和Nginx反代理教程
-
CentOS 7搭建Linux GPU服务器的教程
-
Linux服务器部署.Net Core笔记:六、安装MySQL
-
Mysql 5.7在Linux上部署及远程访问