adapt to paddle about 0D-Tensor (#3508)

PaddlePaddle · Oct 11, 2023 · 0f1fd70 · 0f1fd70
1 parent bf93244
commit 0f1fd70
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Matting/ppmatting/models/human_matting.py b/Matting/ppmatting/models/human_matting.py
@@ -423,12 +423,12 @@ def forward(self, src, pha, err, hid, tri):
         x = paddle.concat([hid, pha, tri], axis=1)
         x = F.interpolate(
             x,
-            paddle.concat((h_half, w_half)),
+            paddle.stack((h_half, w_half)).squeeze(),
             mode='bilinear',
             align_corners=False)
         y = F.interpolate(
             src,
-            paddle.concat((h_half, w_half)),
+            paddle.stack((h_half, w_half)).squeeze(),
             mode='bilinear',
             align_corners=False)
 
@@ -440,11 +440,11 @@ def forward(self, src, pha, err, hid, tri):
         x = self.conv2(x)
 
         if self.kernel_size == 3:
-            x = F.interpolate(x, paddle.concat((h_full + 4, w_full + 4)))
+            x = F.interpolate(x, paddle.stack((h_full + 4, w_full + 4)).squeeze())
             y = F.pad(src, [2, 2, 2, 2])
         else:
             x = F.interpolate(
-                x, paddle.concat((h_full, w_full)), mode='nearest')
+                x, paddle.stack((h_full, w_full)).squeeze(), mode='nearest')
             y = src
 
         x = self.conv3(paddle.concat([x, y], axis=1))