[Fix] Fix KNet IterativeDecodeHead bug in v0.x branch

2022-11-22 12:40:19 +08:00 · 2022-11-22 12:40:19 +08:00 · 589d8ac224
parent 180ad08824
commit 589d8ac224
1 changed files with 1 additions and 0 deletions
--- 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))