Spaces:
Runtime error
Runtime error
wzhouxiff
commited on
Commit
·
d5f3775
1
Parent(s):
5b1668b
rescale=0.0 if use camera pose
Browse files- app.py +1 -1
- objctrl_2_5d/utils/ui_utils.py +14 -5
app.py
CHANGED
@@ -680,7 +680,7 @@ with gr.Blocks() as demo:
|
|
680 |
get_camera_pose_button.click(
|
681 |
get_camera_pose(CAMERA_MODE),
|
682 |
[camera_option, selected_points, depth, mask, rescale, angle, Tx, Ty, Tz, speed],
|
683 |
-
[camera_pose, camera_pose_vis]
|
684 |
)
|
685 |
|
686 |
generated_button.click(
|
|
|
680 |
get_camera_pose_button.click(
|
681 |
get_camera_pose(CAMERA_MODE),
|
682 |
[camera_option, selected_points, depth, mask, rescale, angle, Tx, Ty, Tz, speed],
|
683 |
+
[camera_pose, camera_pose_vis, rescale]
|
684 |
)
|
685 |
|
686 |
generated_button.click(
|
objctrl_2_5d/utils/ui_utils.py
CHANGED
@@ -119,7 +119,11 @@ def traj2cam(traj, depth, rescale):
|
|
119 |
center_h_margin, center_w_margin = center_margin, center_margin
|
120 |
depth_center = np.mean(depth[height//2-center_h_margin:height//2+center_h_margin, width//2-center_w_margin:width//2+center_w_margin])
|
121 |
|
|
|
|
|
|
|
122 |
depth_rescale = round(depth_scale_ * rescale / depth_center, 2)
|
|
|
123 |
r_depth = depth * depth_rescale
|
124 |
|
125 |
zc = []
|
@@ -144,7 +148,7 @@ def traj2cam(traj, depth, rescale):
|
|
144 |
RTs = traj_w2c[:, :3]
|
145 |
fig = vis_camera_rescale(RTs)
|
146 |
|
147 |
-
return RTs, fig
|
148 |
|
149 |
def get_rotate_cam(angle, depth):
|
150 |
# mean_depth = np.mean(depth * mask)
|
@@ -188,12 +192,17 @@ def get_translate_cam(Tx, Ty, Tz, depth, mask, speed):
|
|
188 |
def get_camera_pose(camera_mode):
|
189 |
def trigger_camera_pose(camera_option, selected_points, depth, mask, rescale, angle, Tx, Ty, Tz, speed):
|
190 |
if camera_option == camera_mode[0]: # traj2cam
|
191 |
-
|
192 |
elif camera_option == camera_mode[1]: # rotate
|
193 |
-
|
|
|
194 |
elif camera_option == camera_mode[2]: # clockwise
|
195 |
-
|
|
|
196 |
elif camera_option == camera_mode[3]: # translate
|
197 |
-
|
|
|
|
|
|
|
198 |
|
199 |
return trigger_camera_pose
|
|
|
119 |
center_h_margin, center_w_margin = center_margin, center_margin
|
120 |
depth_center = np.mean(depth[height//2-center_h_margin:height//2+center_h_margin, width//2-center_w_margin:width//2+center_w_margin])
|
121 |
|
122 |
+
if rescale == 0:
|
123 |
+
rescale = 1
|
124 |
+
|
125 |
depth_rescale = round(depth_scale_ * rescale / depth_center, 2)
|
126 |
+
|
127 |
r_depth = depth * depth_rescale
|
128 |
|
129 |
zc = []
|
|
|
148 |
RTs = traj_w2c[:, :3]
|
149 |
fig = vis_camera_rescale(RTs)
|
150 |
|
151 |
+
return RTs, fig, rescale
|
152 |
|
153 |
def get_rotate_cam(angle, depth):
|
154 |
# mean_depth = np.mean(depth * mask)
|
|
|
192 |
def get_camera_pose(camera_mode):
|
193 |
def trigger_camera_pose(camera_option, selected_points, depth, mask, rescale, angle, Tx, Ty, Tz, speed):
|
194 |
if camera_option == camera_mode[0]: # traj2cam
|
195 |
+
RTs, fig, rescale = traj2cam(selected_points, depth, rescale)
|
196 |
elif camera_option == camera_mode[1]: # rotate
|
197 |
+
RTs, fig = get_rotate_cam(angle, depth)
|
198 |
+
rescale = 0.0
|
199 |
elif camera_option == camera_mode[2]: # clockwise
|
200 |
+
RTs, fig = get_clockwise_cam(angle, depth, mask)
|
201 |
+
rescale = 0.0
|
202 |
elif camera_option == camera_mode[3]: # translate
|
203 |
+
RTs, fig = get_translate_cam(Tx, Ty, Tz, depth, mask, speed)
|
204 |
+
rescale = 0.0
|
205 |
+
|
206 |
+
return RTs, fig, rescale
|
207 |
|
208 |
return trigger_camera_pose
|