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

8 篇文章 1 订阅
6 篇文章 0 订阅

项目场景:

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跑他
  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值