From b8b6ee6017e8503b8cadebc17d986ad2dc751c5a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= Date: Mon, 19 Jun 2023 18:34:52 +0800 Subject: [PATCH] fix assert (#3119) --- mmseg/models/segmentors/encoder_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/segmentors/encoder_decoder.py b/mmseg/models/segmentors/encoder_decoder.py index ca37ed59a..b68dab1e9 100644 --- a/mmseg/models/segmentors/encoder_decoder.py +++ b/mmseg/models/segmentors/encoder_decoder.py @@ -328,7 +328,7 @@ class EncoderDecoder(BaseSegmentor): """ assert self.test_cfg.get('mode', 'whole') in ['slide', 'whole'], \ f'Only "slide" or "whole" test mode are supported, but got ' \ - f'{self.test_cfg.mode}' + f'{self.test_cfg["mode"]}.' ori_shape = batch_img_metas[0]['ori_shape'] assert all(_['ori_shape'] == ori_shape for _ in batch_img_metas) if self.test_cfg.mode == 'slide':