From cbec3128bbc04e215e0ede2d3819809850cfd944 Mon Sep 17 00:00:00 2001 From: Junxuan Chen Date: Wed, 20 Apr 2022 15:12:53 +0800 Subject: [PATCH] add 'inv_depths_rgbd' in flip_output --- packnet_sfm/models/model_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/packnet_sfm/models/model_utils.py b/packnet_sfm/models/model_utils.py index 09d0cfbd..9260423f 100644 --- a/packnet_sfm/models/model_utils.py +++ b/packnet_sfm/models/model_utils.py @@ -144,6 +144,7 @@ def flip_output(output): 'inv_depths', 'inv_depths_context', 'inv_depths1', 'inv_depths2', 'pred_depth', 'pred_depth_context', 'pred_depth1', 'pred_depth2', 'pred_inv_depth', 'pred_inv_depth_context', 'pred_inv_depth1', 'pred_inv_depth2', + 'inv_depths_rgbd' ]): output[key] = flip(output[key], flip_lr) return output @@ -177,4 +178,4 @@ def upsample_output(output, mode='nearest', align_corners=None): ]): output[key] = [interpolate_scales( val, mode=mode, align_corners=align_corners) for val in output[key]] - return output \ No newline at end of file + return output