提交 ebe3e885 编写于 作者: 文幕地方's avatar 文幕地方

Support variable length input

上级 444080a0
......@@ -304,9 +304,8 @@ class CSPPAN(nn.Layer):
for idx in range(len(self.in_channels) - 1, 0, -1):
feat_heigh = inner_outs[0]
feat_low = inputs[idx - 1]
upsample_feat = F.upsample(
feat_heigh, size=feat_low.shape[2:4], mode="nearest")
feat_heigh, size=paddle.shape(feat_low)[2:4], mode="nearest")
inner_out = self.top_down_blocks[len(self.in_channels) - 1 - idx](
paddle.concat([upsample_feat, feat_low], 1))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册