gallenszl / CFNet

CFNet: Cascade and Fused Cost Volume for Robust Stereo Matching(CVPR2021)
MIT License
155 stars 23 forks source link

Error when robust_test #31

Closed rebecca0011 closed 2 years ago

rebecca0011 commented 2 years ago

I want to evaluate the accuracy of the self-trained checkpoints on Kitti, running Robust_Test.py, but get the following error ading model /home/rc/20220410StereoMatching/CFNet/checkpoints/sceneflow/pretrained/checkpoint_000009.ckpt start at epoch 0 downscale epochs: [300], downscale rate: 10.0 setting learning rate to 0.001 Traceback (most recent call last): File "robust_test.py", line 335, in <module> train() File "robust_test.py", line 163, in train loss, scalar_outputs, image_outputs = test_sample(sample, compute_metrics=do_summary) File "/home/rc/20220410StereoMatching/CFNet/utils/experiment.py", line 30, in wrapper ret = func(*f_args, **f_kwargs) File "robust_test.py", line 280, in test_sample imgL, imgR, disp_gt = sample['left'], sample['right'], sample['disparity'] KeyError: 'disparity'