mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
MengzhangLI 2022-11-22 12:40:19 +08:00
parent 180ad08824
commit 589d8ac224
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmseg/models/decode_heads/knet_head.py
View File

@ -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))