Zhengyi commited on
Commit
72fbb8b
·
1 Parent(s): bfe84c4

remove codes to inference

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