Spaces:

radames
/

dpt-depth-estimation-3d-obj

Running

nielsr HF staff commited on Mar 23, 2022

Commit

e3f8b3e

•

1 Parent(s): 0cc576d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,8 @@ def write_depth(depth, bits):
   else:
       out = np.zeros(depth.shape, dtype=depth.dtype)
-  cv2.imwrite("result.png", out.astype("uint16"), [cv2.IMWRITE_PNG_COMPRESSION, 0])
   return

   else:
       out = np.zeros(depth.shape, dtype=depth.dtype)
+  out = out/256
+  cv2.imwrite("result.png", out.astype("uint8"), [cv2.IMWRITE_PNG_COMPRESSION, 0])
   return