Spaces:
mashroo
/
Runtime error

YoussefAnso commited on
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.

Files changed (1) hide show
  1. 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),