princeton-vl / RAFT-3D

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

kitti image inference

troylhy1991 opened this issue · comments

Hi, thx for the code release, beautiful work ;)

I'm able to run demo with sample images in assets folder. However, the outputs of kitti images doesn't look normal to me,

image

I used raft3d.raft3d an raft3d_kitti.pth, is there any specific configurations for testing kitti images?

Also, it seems that every time i ran kitti inference, i got an SE outputs with different color (figure below), but it should be deterministic right? ;(

image

Hi, thanks for bringing this to my attention. There appears to be some issue with loading the model weights. I am looking into this today and will post a corrected model soon.

There was an issue with the model file causing the weights be loaded incorrectly. Everything should be fixed now if you re-download kitti model and the kitti_submission.py script. Thanks for pointing out this error!

Thx, Zach. now it works with KITTI images ;)
image

a quick question, do you plan to release the training code soon? i also tested it on other dataset (i.e., cityscapes), but the prediction is quite lousy, seems kitti pre-trained model not transferred well to other sets without fine-tuning ;(

@troylhy1991
Hi, I meet the same problem when try to run the demo. Could you elaborate on how to get the same result as the paper?