ROLO使用教程

科技前沿 • 2025-03-23 20:47 • 阅读 40

大家好，我是讯享网，很高兴认识大家。

python2.7
tensorflow0.8
opencv2.4(可尝试高版本opencv，低版本opencv2.4存在一个问题，步骤6也给出了该问题的解决方案)
以下是一些注意事项，以下均是在终端翻墙情况

安装tensorflow 0.8

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.8.0-cp34-cp34m-linux_x86_64.whl 如果上面失败，可以尝试用下面的命令，注意是在python2.7下面 python2.7 -m pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.8.0-cp34-cp34m-linux_x86_64.whl

讯享网

安装opencv2.4

讯享网conda install -c https://conda.anaconda.org/menpo opencv

主要环境配置完成，下面为代码部分

ROLO本质上就是在YOLO的基础上加上一个LSTM进行一个后处理，因此在使用ROLO代码首先要通过YOLO产生一个初步的bounding box，YOLO代码在ROLO-master/3rd party下面，该代码使用

需要下载个YOLO_small.ckpt模型，模型链接 YOLO_small.ckpt下载地址, 需翻墙
修改代码中weights_file路径；
根据你要测试的数据，修改代码中 test值，注意test下面注释不对！！！
在ROLO-master目录下运行YOLO_network.py， (一定要在该目录下运行，否则报错一个类似这样的错误
paths = [os.path.join(fold,fn) for fn in next(os.walk(fold))[2]] StopIteration)

python 3rd\ party/YOLO_network.py

此时可能会报错
import ROLO_utils as utils ImportError: No module named ROLO_utilsimport ROLO_utils as utils
解决办法为:

讯享网import sys sys.path.append("/path/to/utils") import ROLO_utils as utils

此时可能会报错
AttributeError: ‘NoneType’ object has no attribute ‘shape’
该错误是因为opencv2.4与python的接口问题，导致imread没有读入图片
可通过该命令解决：

pip install --upgrade opencv-python

在顺利执行YOLO代码后，会在benchmark/DATA/对应数据下生成yolo_out，此时即可调用ROLO_network_test_single.py，该代码需要修改的部分为：

self.rolo_weights_file / rolo_weights_file 修改为预训练好的ROLO模型位置
根据你要测试的数据，修改代码中 test值，与YOLO_network.py中一致
在import ROLO_utils as utils前面添加

讯享网import sys sys.path.append('/home/bywind/ROLO-master/utils')

在ROLO-master目录下运行ROLO_network_test_single.py

python./experiments/testing/ROLO_network_test_single.py

该部分代码调通后，会生成rolo_out/，此时可调用
ROLO_demo_test.py 进行可视化对比，该部分代码需要修改部分为:

在import ROLO_utils as utils前面添加

讯享网import sys sys.path.append('/home/bywind/ROLO-master/utils')

如果报错fourcc = cv2.cv.CV_FOURCC(*“XVID”) AttributeError: ‘module’ object has no attribute ‘cv’
则将代码中

fourcc= cv2.cv.CV_FOURCC(*'DIVX') 修改为 fourcc= cv2.VideoWriter_fourcc(*'DIVX')

运行

讯享网

讯享网python ./ROLO_demo_test.py

即可得到
在这里插入图片描述

ROLO使用教程

主要环境配置完成，下面为代码部分

相关推荐