Spaces:

radames
/

Depth-Image-to-Autostereogram

Running

radames commited on May 4, 2022

Commit

a9bf5f9

1 Parent(s): b3ad117

invert depth map

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def process_image(image_path):
         align_corners=False,
     ).squeeze()
     output = prediction.cpu().numpy()
-    depth_image = (output * 255 / np.max(output)).astype('uint8')
     depth_image_padded = np.array(ImageOps.pad(
         Image.fromarray(depth_image), (1280, 720)))

         align_corners=False,
     ).squeeze()
     output = prediction.cpu().numpy()
+    depth_image = (255 - output * 255 / np.max(output)).astype('uint8')
     depth_image_padded = np.array(ImageOps.pad(
         Image.fromarray(depth_image), (1280, 720)))