stereo-transformer icon indicating copy to clipboard operation
stereo-transformer copied to clipboard

Reproduce pre-trained results

Open VitorGuizilini-TRI opened this issue 2 years ago • 2 comments

Hi, how can I evaluate the pre-trained models to reproduce the KITTI 2015 numbers? Thank you!

VitorGuizilini-TRI avatar Aug 25 '21 18:08 VitorGuizilini-TRI

Hi @VitorGuizilini-TRI The training and evaluation scripts are provided in the scripts folder.

mli0603 avatar Aug 26 '21 13:08 mli0603

I'm using the kitti_toy_eval.sh script, but pointing to the KITTI 2015 dataset, with the 200 training images, and the kitti_finetuned_model.pth.tar. Out of the box it only evaluates on 10 images, giving these results:

Epoch 0, epe 0.6943, iou 0.9984, px error 0.0204

If I modify the dataloader to evaluate on all 200 images, I get these results:

Epoch 0, epe 0.4727, iou 0.9987, px error 0.0116

I'm assuming these are the numbers I should be getting, right?

Dataset 3px Error EPE
KITTI 2015 training 0.79 0.41

VitorGuizilini-TRI avatar Aug 26 '21 15:08 VitorGuizilini-TRI