Carlos Vargas R commited on
Commit
3457f75
Β·
1 Parent(s): 70b0c25

Cambios efecto espejo en OBJ

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. 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(