diff --git a/unidepth/models/unidepthv1/decoder.py b/unidepth/models/unidepthv1/decoder.py index 7b86657..c29fd09 100644 --- a/unidepth/models/unidepthv1/decoder.py +++ b/unidepth/models/unidepthv1/decoder.py @@ -209,21 +209,9 @@ def forward( # camera_embedding # torch.cuda.synchronize() # start = time() - rays_embedding_16 = F.normalize( - flat_interpolate(rays_hr, old=self.original_shapes, new=shapes), dim=-1 - ) - rays_embedding_8 = F.normalize( - flat_interpolate( - rays_hr, old=self.original_shapes, new=[x * 2 for x in shapes] - ), - dim=-1, - ) - rays_embedding_4 = F.normalize( - flat_interpolate( - rays_hr, old=self.original_shapes, new=[x * 4 for x in shapes] - ), - dim=-1, - ) + rays_embedding_16 = flat_interpolate(rays_hr, old=self.original_shapes, new=shapes) + rays_embedding_8 = flat_interpolate(rays_hr, old=self.original_shapes, new=[x * 2 for x in shapes]) + rays_embedding_4 = flat_interpolate(rays_hr, old=self.original_shapes, new=[x * 4 for x in shapes]) rays_embedding_16 = self.project_rays16(rsh_cart_8(rays_embedding_16)) rays_embedding_8 = self.project_rays8(rsh_cart_8(rays_embedding_8)) rays_embedding_4 = self.project_rays4(rsh_cart_8(rays_embedding_4))