tried normalization, did not work

This commit is contained in:
2023-01-06 11:19:34 +00:00
parent 51b6219021
commit 022c488180
5 changed files with 104 additions and 20 deletions
+5 -1
View File
@@ -47,4 +47,8 @@ def get_normals_from_depth(depth_image, depth_mask, camera_intrinsic):
depth_dilated = torch.permute(depth_dilated, [0,3,1,2])
normals = kornia.geometry.depth.depth_to_normals(depth_dilated, camera_intrinsic, normalize_points=True)
normals = torch.permute(normals, [0,2,3,1]) * depth_mask[:,:,:,None]
return normals
return normals
def center_crop_resize(img, new_center, size):
pass