RuntimeError: Tensors must have same number of dimensions: got 4 and 3

[WesternTrail]

When I set the Model = MVSNET (Refine = TRUE), an error occurs: Runtimeerror: Tensors Must has number of diarsions: Got 4 and 3?IntersectionIntersectionDoes the Internet do not support the Refinement part

[Innocence4822]

Have you solved this problem?

[mumuliu66]

me too

[Zhaoyibinn]

    if not self.refine:
        return {"depth": depth, "photometric_confidence": photometric_confidence}
    else:
        resize_transform = torchvision.transforms.Resize(size=(depth.shape[1], depth.shape[2]))
        resized_img = resize_transform(imgs[0][0])
        refined_depth = self.refine_network(resized_img.unsqueeze(0), depth.unsqueeze(0))
        return {"depth": depth, "refined_depth": refined_depth, "photometric_confidence": photometric_confidence}