Spaces:
Running
on
Zero
Running
on
Zero
Carlos Vargas R
commited on
Commit
Β·
3457f75
1
Parent(s):
70b0c25
Cambios efecto espejo en OBJ
Browse files- app.py +1 -0
- src/utils/mesh_util.py +1 -1
app.py
CHANGED
@@ -28,6 +28,7 @@ from src.utils.train_util import instantiate_from_config
|
|
28 |
zero = torch.Tensor([0]).cuda()
|
29 |
print(zero.device) # <-- 'cpu' π€
|
30 |
print(zero.device) # <-- 'cuda:0' π€
|
|
|
31 |
def find_cuda():
|
32 |
cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
|
33 |
if cuda_home and os.path.exists(cuda_home):
|
|
|
28 |
zero = torch.Tensor([0]).cuda()
|
29 |
print(zero.device) # <-- 'cpu' π€
|
30 |
print(zero.device) # <-- 'cuda:0' π€
|
31 |
+
|
32 |
def find_cuda():
|
33 |
cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
|
34 |
if cuda_home and os.path.exists(cuda_home):
|
src/utils/mesh_util.py
CHANGED
@@ -17,7 +17,7 @@ from PIL import Image
|
|
17 |
|
18 |
def save_obj(pointnp_px3, facenp_fx3, colornp_px3, fpath):
|
19 |
|
20 |
-
pointnp_px3 = pointnp_px3 @ np.array([[1, 0, 0], [0, 1, 0], [0, 0, -1]])
|
21 |
facenp_fx3 = facenp_fx3[:, [2, 1, 0]]
|
22 |
|
23 |
mesh = trimesh.Trimesh(
|
|
|
17 |
|
18 |
def save_obj(pointnp_px3, facenp_fx3, colornp_px3, fpath):
|
19 |
|
20 |
+
pointnp_px3 = pointnp_px3 @ np.array([[-1, 0, 0], [0, 1, 0], [0, 0, -1]])
|
21 |
facenp_fx3 = facenp_fx3[:, [2, 1, 0]]
|
22 |
|
23 |
mesh = trimesh.Trimesh(
|