From 05677ac64d61b615048eeba1789fb89c4d7d4cab Mon Sep 17 00:00:00 2001 From: Pang Date: Tue, 17 Oct 2023 00:40:26 +0800 Subject: [PATCH] Update adapter.py Solve the problem of mismatch in the number of return values in special cases. --- open_vocab_seg/modeling/clip_adapter/adapter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/open_vocab_seg/modeling/clip_adapter/adapter.py b/open_vocab_seg/modeling/clip_adapter/adapter.py index 864d20b..c90a1b1 100644 --- a/open_vocab_seg/modeling/clip_adapter/adapter.py +++ b/open_vocab_seg/modeling/clip_adapter/adapter.py @@ -125,7 +125,7 @@ def forward( ): (regions, unnorm_regions), region_masks, valid_flag = self._preprocess_image(image, mask, normalize=normalize) if regions is None: - return None, valid_flag + return None, None, valid_flag if isinstance(regions, list): assert NotImplementedError image_features = torch.cat( @@ -177,7 +177,7 @@ def _preprocess_image( regions.append(region.unsqueeze(0)) region_masks.append(region_mask.unsqueeze(0)) if len(regions) == 0: - return None, valid + return (None, None), None, valid unnorm_regions = regions if normalize: regions = [(r - self.pixel_mean) / self.pixel_std for r in regions]