Spaces:
Running
on
Zero
Running
on
Zero
remove codes to inference
Browse files- inference.py +4 -0
- model/crm/model.py +2 -2
inference.py
CHANGED
@@ -6,8 +6,12 @@ from util.utils import get_tri
|
|
6 |
import tempfile
|
7 |
from mesh import Mesh
|
8 |
import zipfile
|
|
|
9 |
def generate3d(model, rgb, ccm, device):
|
10 |
|
|
|
|
|
|
|
11 |
color_tri = torch.from_numpy(rgb)/255
|
12 |
xyz_tri = torch.from_numpy(ccm[:,:,(2,1,0)])/255
|
13 |
color = color_tri.permute(2,0,1)
|
|
|
6 |
import tempfile
|
7 |
from mesh import Mesh
|
8 |
import zipfile
|
9 |
+
from util.renderer import Renderer
|
10 |
def generate3d(model, rgb, ccm, device):
|
11 |
|
12 |
+
model.renderer = Renderer(tet_grid_size=model.tet_grid_size, camera_angle_num=model.camera_angle_num,
|
13 |
+
scale=model.input.scale, geo_type = model.geo_type)
|
14 |
+
|
15 |
color_tri = torch.from_numpy(rgb)/255
|
16 |
xyz_tri = torch.from_numpy(ccm[:,:,(2,1,0)])/255
|
17 |
color = color_tri.permute(2,0,1)
|
model/crm/model.py
CHANGED
@@ -56,8 +56,8 @@ class CRM(nn.Module):
|
|
56 |
|
57 |
self.sdfMlp = SdfMlp(mlp_chnl_s * 32, 512, bias=self.arch.mlp_bias)
|
58 |
self.rgbMlp = RgbMlp(mlp_chnl_s * 32, 512, bias=self.arch.mlp_bias)
|
59 |
-
self.renderer = Renderer(tet_grid_size=self.tet_grid_size, camera_angle_num=self.camera_angle_num,
|
60 |
-
|
61 |
|
62 |
|
63 |
self.spob = True if specs['Pretrain']['mode'] is None else False # whether to add sphere
|
|
|
56 |
|
57 |
self.sdfMlp = SdfMlp(mlp_chnl_s * 32, 512, bias=self.arch.mlp_bias)
|
58 |
self.rgbMlp = RgbMlp(mlp_chnl_s * 32, 512, bias=self.arch.mlp_bias)
|
59 |
+
# self.renderer = Renderer(tet_grid_size=self.tet_grid_size, camera_angle_num=self.camera_angle_num,
|
60 |
+
# scale=self.input.scale, geo_type = self.geo_type)
|
61 |
|
62 |
|
63 |
self.spob = True if specs['Pretrain']['mode'] is None else False # whether to add sphere
|