Spaces:

zhigangjiang
/

3D-Room-Layout-Estimation_LGT-Net

Running

zhigangjiang commited on Nov 1, 2023

Commit

54dbfd8

•

1 Parent(s): 0f026b7

Update utils/conversion.py

Files changed (1) hide show

utils/conversion.py CHANGED Viewed

@@ -52,7 +52,7 @@ def get_lon(w, is_np, b=None):
 def pixel2uv(pixel, w=1024, h=512, axis=None):
-    pixel = pixel.astype(np.float) if isinstance(pixel, np.ndarray) else pixel.float()
     # +0.5 will make left/right and up/down coordinates symmetric
     if axis is None:
         u = (pixel[..., 0:1] + 0.5) / w

 def pixel2uv(pixel, w=1024, h=512, axis=None):
+    pixel = pixel.astype(np.float64) if isinstance(pixel, np.ndarray) else pixel.float()
     # +0.5 will make left/right and up/down coordinates symmetric
     if axis is None:
         u = (pixel[..., 0:1] + 0.5) / w