HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: HTensor:0404f5b5b02e0b40705d0d0ba65972b42fc70fe2
Branches Tags
HTensor:main
HTensor:original-test
HTensor:asymmetric-augmentation-crestereo
HTensor:profile
HTensor:add-license-1
..
compare: HTensor:74bc06d58e5902887700a3874bba1a4463ac2b1f
Branches Tags
HTensor:original-test
HTensor:asymmetric-augmentation-crestereo
HTensor:main
HTensor:profile
HTensor:add-license-1

3 Commits
0404f5b5b0 ... 74bc06d58e

Author SHA1 Message Date
HTensor
74bc06d58e Merge branch 'original-test' 2023-05-03 04:30:45 +08:00
HTensor
9df896bb70 Update stereo_datasets.py 2023-05-01 12:25:21 +08:00
HTensor
73e65f99b8 Update evaluate_stereo.py 2023-04-30 16:03:58 +08:00
2 changed files with 1 additions and 2 deletions
Show Stats Expand all files Collapse all files

1
IGEV-Stereo/core/stereo_datasets.py
View File

@ -73,7 +73,6 @@ class StereoDataset(data.Dataset):
img2 = np.array(img2).astype(np.uint8) img2 = np.array(img2).astype(np.uint8)
disp = np.array(disp).astype(np.float32) disp = np.array(disp).astype(np.float32)
assert not (True in np.isnan(disp))
flow = np.stack([disp, np.zeros_like(disp)], axis=-1) flow = np.stack([disp, np.zeros_like(disp)], axis=-1)

2
IGEV-Stereo/evaluate_stereo.py
View File

@ -169,7 +169,7 @@ def validate_sceneflow(model, iters=32, mixed_prec=False, max_disp=192):
@torch.no_grad() @torch.no_grad()
def validate_middlebury(model, iters=32, split='F', mixed_prec=False, max_disp=192): def validate_middlebury(model, iters=32, split='H', mixed_prec=False, max_disp=192):
""" Peform validation using the Middlebury-V3 dataset """ """ Peform validation using the Middlebury-V3 dataset """
model.eval() model.eval()
aug_params = {} aug_params = {}