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)