Commit
·
8773786
1
Parent(s):
96bbb06
Reduce remesh iterations in generate3d function from 10 to 7 for optimized mesh generation
Browse files- inference.py +1 -1
- 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=
|
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 |
|