mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Input previous results for the last cascade_decode_head (#1450) 

* [Fix] Input previous results for the latter cascade_decode_head

* minors
pull/1477/head
Miao Zheng 2022-04-13 21:16:29 +08:00 committed by GitHub
parent 3f797072d8
commit 23ae1ebab6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmseg/models/segmentors/cascade_encoder_decoder.py
View File

@ -75,8 +75,12 @@ class CascadeEncoderDecoder(EncoderDecoder):
for i in range(1, self.num_stages): for i in range(1, self.num_stages):
# forward test again, maybe unnecessary for most methods. # forward test again, maybe unnecessary for most methods.
prev_outputs = self.decode_head[i - 1].forward_test( if i == 1:
x, img_metas, self.test_cfg) prev_outputs = self.decode_head[0].forward_test(
x, img_metas, self.test_cfg)
else:
prev_outputs = self.decode_head[i - 1].forward_test(
x, prev_outputs, img_metas, self.test_cfg)
loss_decode = self.decode_head[i].forward_train( loss_decode = self.decode_head[i].forward_train(
x, prev_outputs, img_metas, gt_semantic_seg, self.train_cfg) x, prev_outputs, img_metas, gt_semantic_seg, self.train_cfg)
losses.update(add_prefix(loss_decode, f'decode_{i}')) losses.update(add_prefix(loss_decode, f'decode_{i}'))