Spaces:
mashroo
/
Runtime error

YoussefAnso commited on
Commit
8773786
·
1 Parent(s): 96bbb06

Reduce remesh iterations in generate3d function from 10 to 7 for optimized mesh generation

Browse files
Files changed (2) hide show
  1. inference.py +1 -1
  2. model/archs/mlp_head.py +0 -1
inference.py CHANGED
@@ -70,7 +70,7 @@ def generate3d(model, rgb, ccm, device):
70
  verts, faces = clean_mesh(
71
  data_config['verts'].squeeze().cpu().numpy().astype(np.float32),
72
  data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
73
- repair=False, remesh=True, remesh_size=0.01, remesh_iters=10
74
  )
75
  data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()
76
  data_config['faces'] = torch.from_numpy(faces).to(device).contiguous()
 
70
  verts, faces = clean_mesh(
71
  data_config['verts'].squeeze().cpu().numpy().astype(np.float32),
72
  data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
73
+ repair=False, remesh=True, remesh_size=0.01, remesh_iters=7
74
  )
75
  data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()
76
  data_config['faces'] = torch.from_numpy(faces).to(device).contiguous()
model/archs/mlp_head.py CHANGED
@@ -35,7 +35,6 @@ class RgbMlp(nn.Module):
35
  x = F.relu(self.fc1(input))
36
  x = F.relu(self.fc2(x))
37
  out = self.fc3(x)
38
- out = torch.tanh(out)
39
 
40
  return out
41
 
 
35
  x = F.relu(self.fc1(input))
36
  x = F.relu(self.fc2(x))
37
  out = self.fc3(x)
 
38
 
39
  return out
40