From 58e73c3fab714c6ee7d1e7c02633f9dffa4550a5 Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 12:46:24 +0800 Subject: [PATCH] [Fix] Fix KNet IterativeDecodeHead bug in dev-1.x branch --- mmseg/models/decode_heads/knet_head.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmseg/models/decode_heads/knet_head.py b/mmseg/models/decode_heads/knet_head.py index 181ae0216..ae7afc96b 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -422,6 +422,7 @@ class IterativeDecodeHead(BaseDecodeHead): self.num_classes = self.kernel_generate_head.num_classes self.input_transform = self.kernel_generate_head.input_transform self.ignore_index = self.kernel_generate_head.ignore_index + self.out_channels = self.num_classes for head_cfg in kernel_update_head: self.kernel_update_head.append(MODELS.build(head_cfg))