Ryukijano commited on
Commit
5d25b5d
·
verified ·
1 Parent(s): 17ea36c

Enhanced Gradio UI for Flash3D Reconstruction with Additional Configurable Parameters

Browse files

- Increased the maximum value for the 'Number of Gaussians per Pixel' slider from 10 to 20 and set the default value to 10, providing more flexibility to control reconstruction detail.
- Adjusted the 'Scale Factor for Model Size' slider range from [0.5, 5.0] with a default value of 1.5, allowing finer control over output scaling.
- Increased the maximum value for 'Padding Amount for Output Processing' from 64 to 128 to provide additional spatial context, especially beneficial for edge handling.
- Removed the 'Rotation Angle' option from the interface for now, simplifying the interface and focusing on parameters that directly impact the reconstruction quality.
- Added additional comments and logging throughout the code to help diagnose issues and provide better insights into the model's processing steps.
- Set the GPU allocation duration to 120 seconds, because that is the maximum duration allowed.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ def main():
76
  return image
77
 
78
  # Function to reconstruct the 3D model from the input image and export it as a PLY file
79
- @spaces.GPU(duration=600) # Decorator to allocate a GPU for this function during execution
80
  def reconstruct_and_export(image, num_gauss, scale_factor):
81
  """
82
  Passes image through model, outputs reconstruction in form of a dict of tensors.
 
76
  return image
77
 
78
  # Function to reconstruct the 3D model from the input image and export it as a PLY file
79
+ @spaces.GPU(duration=120) # Decorator to allocate a GPU for this function during execution
80
  def reconstruct_and_export(image, num_gauss, scale_factor):
81
  """
82
  Passes image through model, outputs reconstruction in form of a dict of tensors.