Skip to content
Snippets Groups Projects
Commit cc3a0528 authored by Miao Zheng's avatar Miao Zheng Committed by GitHub
Browse files

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

* [Fix] Input previous results for the latter cascade_decode_head

* minors
parent f50bfe3e
No related branches found
No related tags found
No related merge requests found
......@@ -75,8 +75,12 @@ class CascadeEncoderDecoder(EncoderDecoder):
for i in range(1, self.num_stages):
# forward test again, maybe unnecessary for most methods.
prev_outputs = self.decode_head[i - 1].forward_test(
x, img_metas, self.test_cfg)
if i == 1:
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(
x, prev_outputs, img_metas, gt_semantic_seg, self.train_cfg)
losses.update(add_prefix(loss_decode, f'decode_{i}'))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment