Commit
·
293fbf3
1
Parent(s):
b913122
Add debug logging for color statistics in CRM model
Browse files- 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
|