Skip to content

Commit e7146be

Browse files
committed
add resolution adjust with browser size or sidebar config
1 parent 9c61aae commit e7146be

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

threedgrut/utils/viser_gui_util.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,16 @@ def _init_ui(self):
106106
initial_value=True
107107
)
108108

109+
self.adjust_resolution_checkbox = self.server.gui.add_checkbox(
110+
"Adjust Browser Size",
111+
initial_value=False
112+
)
113+
114+
self.resolution_slider = self.server.gui.add_slider(
115+
"Resolution", min=384, max=4096,
116+
step=2, initial_value=1024
117+
)
118+
109119
self.subsample_slider = self.server.gui.add_slider(
110120
"Subsample",
111121
min=1,
@@ -194,8 +204,15 @@ def render_from_current_view(self, client) -> Tuple[torch.Tensor, torch.Tensor,
194204
camera = client.camera
195205

196206
# Get window size and apply subsample
197-
window_w = self.render_width // self.viz_render_subsample
198-
window_h = self.render_height // self.viz_render_subsample
207+
208+
if self.adjust_resolution_checkbox.value:
209+
window_w = self.render_width // self.viz_render_subsample
210+
window_h = self.render_height // self.viz_render_subsample
211+
else:
212+
window_w = self.resolution_slider.value
213+
window_h = int(self.resolution_slider.value / camera.aspect)
214+
215+
199216

200217
# Get camera parameters from viser
201218
view_matrix = self.get_c2w(camera) # This is W2C (world to camera)

0 commit comments

Comments
 (0)