fixing deprecated numpy type names

compphoto · Apr 29, 2024 · 2e5c57b · 2e5c57b
1 parent 2ea7651
commit 2e5c57b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/intrinsic_compositing/albedo/model/MiDaS/midas/transforms.py b/intrinsic_compositing/albedo/model/MiDaS/midas/transforms.py
@@ -36,7 +36,7 @@ def apply_min_size(sample, size, image_interpolation_method=cv2.INTER_AREA):
         sample["disparity"], tuple(shape[::-1]), interpolation=cv2.INTER_NEAREST
     )
     sample["mask"] = cv2.resize(
-        sample["mask"].astype(np.float32),
+        sample["mask"].astype(np.single),
         tuple(shape[::-1]),
         interpolation=cv2.INTER_NEAREST,
     )
@@ -185,7 +185,7 @@ def __call__(self, sample):
                 )
 
             sample["mask"] = cv2.resize(
-                sample["mask"].astype(np.float32),
+                sample["mask"].astype(np.single),
                 (width, height),
                 interpolation=cv2.INTER_NEAREST,
             )
@@ -217,18 +217,18 @@ def __init__(self):
 
     def __call__(self, sample):
         image = np.transpose(sample["image"], (2, 0, 1))
-        sample["image"] = np.ascontiguousarray(image).astype(np.float32)
+        sample["image"] = np.ascontiguousarray(image).astype(np.single)
 
         if "mask" in sample:
-            sample["mask"] = sample["mask"].astype(np.float32)
+            sample["mask"] = sample["mask"].astype(np.single)
             sample["mask"] = np.ascontiguousarray(sample["mask"])
 
         if "disparity" in sample:
-            disparity = sample["disparity"].astype(np.float32)
+            disparity = sample["disparity"].astype(np.single)
             sample["disparity"] = np.ascontiguousarray(disparity)
 
         if "depth" in sample:
-            depth = sample["depth"].astype(np.float32)
+            depth = sample["depth"].astype(np.single)
             sample["depth"] = np.ascontiguousarray(depth)
 
         return sample
diff --git a/intrinsic_compositing/shading/pipeline.py b/intrinsic_compositing/shading/pipeline.py
@@ -215,7 +215,7 @@ def compute_reshading(orig, msk, inv_shd, depth, normals, alb, coeffs, model):
     inv_shd = resize(inv_shd, (round_32(h), round_32(w)))
     dpt = resize(depth, (round_32(h), round_32(w)))
     nrm = resize(normals, (round_32(h), round_32(w)))
-    msk = msk.astype(np.float32)
+    msk = msk.astype(np.single)
 
     hard_msk = (msk > 0.5)