Update ppd/models/ppd.py

ppd/models/ppd.py

@@ -78,7 +78,7 @@ class PixelPerfectDepth(nn.Module):
             depth = self.forward_test(image)
         # depth = F.interpolate(depth, size=(h, w), mode='bilinear', align_corners=False)[0, 0]
 
-        return depth, resize_image
+        return depth.squeeze().cpu().numpy(), resize_image
     
     @torch.no_grad()
     def forward_test(self, image):