Commit
·
bd713a9
1
Parent(s):
2625296
Enhance color processing in generate3d function by applying clipping and gamma correction
Browse files- Added clipping to ensure orig_colors are within the [0, 1] range.
- Implemented gamma correction to adjust color values for improved visual fidelity in the generated mesh.
- inference.py +2 -0
inference.py
CHANGED
@@ -72,6 +72,8 @@ def generate3d(model, rgb, ccm, device):
|
|
72 |
# Comment out the scaling below if orig_colors is already in [0, 1]
|
73 |
# orig_colors = (orig_colors * 0.5 + 0.5).clip(0, 1) # scale to [0, 1]
|
74 |
print('orig_colors min/max AFTER scaling:', orig_colors.min(), orig_colors.max())
|
|
|
|
|
75 |
verts, faces = clean_mesh(
|
76 |
orig_verts.astype(np.float32),
|
77 |
data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
|
|
|
72 |
# Comment out the scaling below if orig_colors is already in [0, 1]
|
73 |
# orig_colors = (orig_colors * 0.5 + 0.5).clip(0, 1) # scale to [0, 1]
|
74 |
print('orig_colors min/max AFTER scaling:', orig_colors.min(), orig_colors.max())
|
75 |
+
orig_colors = np.clip(orig_colors, 0, 1)
|
76 |
+
orig_colors = np.power(orig_colors, 1/2.2)
|
77 |
verts, faces = clean_mesh(
|
78 |
orig_verts.astype(np.float32),
|
79 |
data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
|