Beispiel #1
0
 def depth_input(depth_image: np.array, **kwargs):
     c, h, w = depth_image.size()
     depth_image = depth_image[0:1, :, :]  # depth
     new_image = depth_image.expand(1, 3, h, w)
     return new_image