search

ywz978020607 / HESIC

Official Code of "Deep Homography for Efficient Stereo Image Compression"[cvpr21oral]
Apache License 2.0
54 stars 12 forks source link

using pre-trained models #5

Open amarzullo24 opened 2 years ago

amarzullo24 commented 2 years ago

Hi! I am trying to use your scripts to reproduce the results in the paper but I am not sure I am doing it right. In particular, when running the following command (as suggested in the README.md):

python test3real.py -d "/home/ywz/database/aftercut512" --seed 0 --patch-size 512 512 --batch-size 1 --test-batch-size 1

I get an import error due to the missing newnet9 module.

By using the script test3_real.py, instead I am able to load all the models and complete the evaluation.

I wrote a small Colab Notebook so you can check what I have done (I think it may also be useful for others). https://colab.research.google.com/drive/1kWtOCHsQRd8Ae_tCgVn5XfPyhM-OxppB?usp=sharing

I have some questions:

  1. Is the one described above the right procedure for reproducing your best results?
  2. The numbers on InStereo2K seems similar to the ones reported in the paper. However I am observing a drop when testing on KITTI. Is this expected?
  3. On which resolution should I test KITTI? I can't find it on the paper

I apologize for these multiple questions. Hope you can help!

amarzullo24 commented 2 years ago

Hi, would you help me with this issue?