From 589d8ac224f43f57403c0ebf99326d325731244f Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Tue, 22 Nov 2022 12:40:19 +0800 Subject: [PATCH] [Fix] Fix KNet IterativeDecodeHead bug in v0.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 f73daccb6..bb5dec590 100644 --- a/mmseg/models/decode_heads/knet_head.py +++ b/mmseg/models/decode_heads/knet_head.py @@ -420,6 +420,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(build_head(head_cfg))