From d1a395cc78ddfdb195c2b1765dcfeb21ce81249c Mon Sep 17 00:00:00 2001 From: Gangwei Xu Date: Sun, 12 Mar 2023 20:22:40 +0800 Subject: [PATCH] Initial Commit --- IGEV-Stereo/core/utils/augmentor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/IGEV-Stereo/core/utils/augmentor.py b/IGEV-Stereo/core/utils/augmentor.py index 256f8ec..fa81ba9 100644 --- a/IGEV-Stereo/core/utils/augmentor.py +++ b/IGEV-Stereo/core/utils/augmentor.py @@ -302,6 +302,7 @@ class SparseFlowAugmentor: img2 = img2[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]] flow = flow[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]] valid = valid[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]] + return img1, img2, flow, valid