tensorflow / models

Models and examples built with TensorFlow
Other
77.16k stars 45.76k forks source link

Ego-motion inference problem #6583

Open yoavnaveh5 opened 5 years ago

yoavnaveh5 commented 5 years ago

Hello @rezama, thanks for sharing the code.

System information

Describe the problem

I am trying to test the ego-motion predictions by your trained model on the KITTI dataset. currently inference.py produces only depth estimations. can you instruct me on the needed modifications to be able to predict egomotion? thanks!!

tensorflowbutler commented 4 years ago

Hi There, We are checking to see if you still need help on this, as this seems to be an old issue. Please update this issue with the latest information, code snippet to reproduce your issue and error you are seeing. If we don't hear from you in the next 7 days, this issue will be closed automatically. If you don't need help on this issue any more, please consider closing this.

rezama commented 4 years ago

You can get egomotion predictions by passing a mode argument to the inference method:

inference_model.inference(inputs, sess, mode='egomotion')

Note that here inputs should be an image containing 3 frames, as defined here:

https://github.com/tensorflow/models/blob/20c78a9186ff3d35612f15764262f7e3e33eeae9/research/vid2depth/model.py#L345