ICRA2020开源项目DF-VO运行记录

  • Post author:
  • Post category:其他




项目场景:

DF-VO是ICRA2020国际会议的开源项目,想着搞下来玩玩,这是项目的

传送门


两篇论文也附上:


Visual Odometry Revisited: What Should Be Learnt?



DF-VO: What Should Be Learnt for Visual Odometry?


在这里插入图片描述




运行:

This code was tested with Python 3.6, CUDA 9.0, Ubuntu 16.04, and PyTorch-1.1.

github里面提到的相关环境如上

但是我没有用官方github的方法去配置环境,因为自己的cuda版本有点不一样,所以想着自己搞环境算了。

我个人的环境是Python3.7,CUDA11.3,Ubuntu18.04,PyTorch1.8



1 其他包安装

这里就根据自己的环境安装了相关的包,少什么装什么,大概装了10个不到吧,就搞定了,没有官方说的那么多需要装的



2 下载数据和模型

官方的github给出了下载地址,KITTI可能下载有点慢,可以CSDN找一下其他百度网盘的资料,模型可能要翻墙



3 处理数据

项目中的第一个步骤是运行run.py,但是代码中读取的是jpg格式的图片,而我从KITTI下载下来的是png的,而且像素也对不上,所以要改一下。以下是我自己的代码:

import cv2 as cv
import sys
import os

dirlist = os.listdir(r'{PNG格式图片的的地址}')
j = 0
for i in dirlist:
    print(i) 
    # 我这里的地址是这样,这是PNG的地址哦
    img = cv.imread(r"/home/a/DF-VO/dataset/kitti_odom/odom_data_jpg/10/image_2_0/{}".format(i))
    res = cv.resize(img, (640, 192))
    name = i.split('.')[0]
    # 这个是保存的图片地址,就是jpg的
    cv.imwrite(r"/home/a/DF-VO/dataset/kitti_odom/odom_data_jpg/10/image_2/{}.jpg".format(name), res)
    j += 1



4 处理问题

搞定之后还有一些问题,比如说无法引入包、pytorch版本对不上、cupy版本对不上的问题。这些问题都能在CSDN找到相关的解决办法

pytorch版本能用1.3以前的最好,如果不是会有报错,解决完后还是有警告



5跑他



版权声明:本文为qq_39502099原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。