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.
@@ -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=
|
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.
|