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
|