Upload folder using huggingface_hub

svrm/ldm/models/svrm.py

@@ -40,6 +40,29 @@ try:
     import open3d as o3d
 except:
     raise "failed to import 3d libraries "
+    
+try:
+    import cvcuda
+    def uv_padding(image, hole_mask, uv_padding_size = 2):
+
+        inpaint_image = (
+                torch.as_tensor(
+                    cvcuda.inpaint(
+                        src=cvcuda.as_tensor((image * 255).to(torch.uint8), 'HWC'),
+                        masks=cvcuda.as_tensor((hole_mask * 255).view(1024, 1024, 1).to(torch.uint8), 'HWC'),
+                        inpaintRadius=uv_padding_size,
+                    ).cuda()
+                )
+        )
+        return inpaint_image.cpu().numpy()
+except Exception as err:
+    def uv_padding(image, hole_mask, uv_padding_size = 2):
+        return cv2.inpaint(
+            (image.detach().cpu().numpy() * 255).astype(np.uint8),
+            (hole_mask.detach().cpu().numpy() * 255).astype(np.uint8),
+            uv_padding_size,
+            cv2.INPAINT_TELEA
+        )
 
 from ..modules.rendering_neus.mesh import Mesh
 from ..modules.rendering_neus.rasterize import NVDiffRasterizerContext
@@ -58,13 +81,6 @@ def unwrap_uv(v_pos, t_pos_idx):
     indices = indices.astype(np.int64, casting="same_kind")
     return uvs, indices
 
-def uv_padding(image, hole_mask, uv_padding_size = 2):
-    return cv2.inpaint(
-        (image.detach().cpu().numpy() * 255).astype(np.uint8),
-        (hole_mask.detach().cpu().numpy() * 255).astype(np.uint8),
-        uv_padding_size,
-        cv2.INPAINT_TELEA
-    )
 
 def refine_mesh(vtx_refine, faces_refine):
     mesh = o3d.geometry.TriangleMesh(