Spaces:
mashroo
/
Runtime error

YoussefAnso commited on
Commit
293fbf3
·
1 Parent(s): b913122

Add debug logging for color statistics in CRM model

Browse files
Files changed (1) hide show
  1. model/crm/model.py +3 -0
model/crm/model.py CHANGED
@@ -95,8 +95,11 @@ class CRM(nn.Module):
95
 
96
  dec_verts = self.decoder(tri_fea_2, verts.unsqueeze(0))
97
  colors = self.rgbMlp(dec_verts).squeeze().detach().cpu().numpy()
 
 
98
  # Expect predicted colors value range from [-1, 1]
99
  colors = (colors * 0.5 + 0.5).clip(0, 1)
 
100
 
101
  verts = verts[..., [0, 2, 1]]
102
  verts[..., 0]*= -1
 
95
 
96
  dec_verts = self.decoder(tri_fea_2, verts.unsqueeze(0))
97
  colors = self.rgbMlp(dec_verts).squeeze().detach().cpu().numpy()
98
+ print("[DEBUG] Raw color min/max/mean:", colors.min(), colors.max(), colors.mean())
99
+
100
  # Expect predicted colors value range from [-1, 1]
101
  colors = (colors * 0.5 + 0.5).clip(0, 1)
102
+ print("[DEBUG] Scaled color min/max/mean:", colors.min(), colors.max(), colors.mean())
103
 
104
  verts = verts[..., [0, 2, 1]]
105
  verts[..., 0]*= -1