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 dev-1.x branch

Browse Source
This commit is contained in:
MengzhangLI 2022-11-22 12:46:24 +08:00
parent 6d7af0707d
commit 58e73c3fab
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

@ -422,6 +422,7 @@ class IterativeDecodeHead(BaseDecodeHead):
self.num_classes = self.kernel_generate_head.num_classes self.num_classes = self.kernel_generate_head.num_classes
self.input_transform = self.kernel_generate_head.input_transform self.input_transform = self.kernel_generate_head.input_transform
self.ignore_index = self.kernel_generate_head.ignore_index self.ignore_index = self.kernel_generate_head.ignore_index
self.out_channels = self.num_classes
for head_cfg in kernel_update_head: for head_cfg in kernel_update_head:
self.kernel_update_head.append(MODELS.build(head_cfg)) self.kernel_update_head.append(MODELS.build(head_cfg))