Commit ·
6b79750
1
Parent(s): 9efc262
Adjust resolution and remesh iterations in generate3d function for optimized output
Browse files- Decreased the resolution from [2056, 2056] to [1024, 1024] to balance performance and quality.
- Reduced remesh iterations from 3 to 2 to streamline the mesh generation process while maintaining detail.
- inference.py +2 -2
inference.py
CHANGED
|
@@ -56,7 +56,7 @@ def generate3d(model, rgb, ccm, device):
|
|
| 56 |
|
| 57 |
with torch.no_grad():
|
| 58 |
data_config = {
|
| 59 |
-
'resolution': [
|
| 60 |
"triview_color": triplane_color.to(device),
|
| 61 |
}
|
| 62 |
|
|
@@ -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.005, 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()
|
|
|
|
| 56 |
|
| 57 |
with torch.no_grad():
|
| 58 |
data_config = {
|
| 59 |
+
'resolution': [1024, 1024],
|
| 60 |
"triview_color": triplane_color.to(device),
|
| 61 |
}
|
| 62 |
|
|
|
|
| 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.005, remesh_iters=2
|
| 74 |
)
|
| 75 |
data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()
|
| 76 |
data_config['faces'] = torch.from_numpy(faces).to(device).contiguous()
|