From 53f2242570da12a7a9aacac99db690b62882987a Mon Sep 17 00:00:00 2001 From: RoyYang0714 Date: Tue, 15 Oct 2024 13:55:52 +0200 Subject: [PATCH] fix: Fix function args. --- vis4d/eval/common/depth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vis4d/eval/common/depth.py b/vis4d/eval/common/depth.py index 02d7db4b..cee434ef 100644 --- a/vis4d/eval/common/depth.py +++ b/vis4d/eval/common/depth.py @@ -100,8 +100,8 @@ def process_batch( # type: ignore # pylint: disable=arguments-differ """Process a batch of data. Args: - prediction (np.array): Prediction optical flow, in shape (H, W, 2). - groundtruth (np.array): Target optical flow, in shape (H, W, 2). + prediction (np.array): Prediction optical flow, in shape (B, H, W). + groundtruth (np.array): Target optical flow, in shape (B, H, W). """ preds = ( array_to_numpy(prediction, n_dims=None, dtype=np.float32)