From fcd9fc0da04c59a420d0bc0fdf819fe78e69aaba Mon Sep 17 00:00:00 2001 From: Hojin Kim <127716636+HojinKimSIA@users.noreply.github.com> Date: Tue, 9 Apr 2024 14:59:15 +0900 Subject: [PATCH] [fix] test mode in depth estimator --- mmseg/models/segmentors/depth_estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/segmentors/depth_estimator.py b/mmseg/models/segmentors/depth_estimator.py index 1020637e73..b9ebc05c0f 100644 --- a/mmseg/models/segmentors/depth_estimator.py +++ b/mmseg/models/segmentors/depth_estimator.py @@ -323,7 +323,7 @@ def inference(self, inputs: Tensor, batch_img_metas: List[dict]) -> Tensor: level=logging.WARN) if self.test_cfg.mode == 'slide': depth_map = self.slide_inference(inputs, batch_img_metas) - if self.test_cfg.mode == 'slide_flip': + elif self.test_cfg.mode == 'slide_flip': depth_map = self.slide_flip_inference(inputs, batch_img_metas) else: depth_map = self.whole_inference(inputs, batch_img_metas)