重新调试了ROLO代码,记录一下配置过程;ROLO项目地址
ROLO运行环境:

  • python2.7
  • tensorflow0.8
  • opencv2.4(可尝试高版本opencv,低版本opencv2.4存在一个问题,步骤6也给出了该问题的解决方案)
    以下是一些注意事项,以下均是在终端翻墙情况
  1. 安装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
  1. 安装opencv2.4
conda install -c https://conda.anaconda.org/menpo opencv
主要环境配置完成,下面为代码部分
  1. 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 
  1. 此时可能会报错
    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
  1. 此时可能会报错
    AttributeError: ‘NoneType’ object has no attribute ‘shape’
    该错误是因为opencv2.4与python的接口问题,导致imread没有读入图片
    可通过该命令解决:
pip install --upgrade opencv-python
  1. 在顺利执行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 
  1. 该部分代码调通后,会生成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 

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

GitHub 加速计划 / li / linux-dash
10.39 K
1.2 K
下载
A beautiful web dashboard for Linux
最近提交(Master分支:2 个月前 )
186a802e added ecosystem file for PM2 4 年前
5def40a3 Add host customization support for the NodeJS version 4 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐