Spaces:
Running
on
Zero
Running
on
Zero
ban gpu for zoedepth
Browse files- demo/mesh_recon.py +1 -4
demo/mesh_recon.py
CHANGED
|
@@ -4,10 +4,8 @@ import numpy as np
|
|
| 4 |
import torch
|
| 5 |
import trimesh
|
| 6 |
|
| 7 |
-
import spaces
|
| 8 |
|
| 9 |
-
|
| 10 |
-
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
| 11 |
|
| 12 |
# use torch hub
|
| 13 |
model = torch.hub.load("isl-org/ZoeDepth", "ZoeD_NK", pretrained=True).to(device).eval()
|
|
@@ -99,7 +97,6 @@ def depth_edges_mask(depth):
|
|
| 99 |
return mask
|
| 100 |
|
| 101 |
|
| 102 |
-
@spaces.GPU
|
| 103 |
def mesh_reconstruction(
|
| 104 |
masked_image: np.ndarray,
|
| 105 |
mask: np.ndarray,
|
|
|
|
| 4 |
import torch
|
| 5 |
import trimesh
|
| 6 |
|
|
|
|
| 7 |
|
| 8 |
+
device = torch.device('cpu')
|
|
|
|
| 9 |
|
| 10 |
# use torch hub
|
| 11 |
model = torch.hub.load("isl-org/ZoeDepth", "ZoeD_NK", pretrained=True).to(device).eval()
|
|
|
|
| 97 |
return mask
|
| 98 |
|
| 99 |
|
|
|
|
| 100 |
def mesh_reconstruction(
|
| 101 |
masked_image: np.ndarray,
|
| 102 |
mask: np.ndarray,
|