Update app.py
Browse files
app.py
CHANGED
|
@@ -120,6 +120,12 @@ def rayst3r_to_glb(img,depth_dict,mask,max_total_points=10e6,rotated=False):
|
|
| 120 |
|
| 121 |
rayst3r_points[:,1] = -rayst3r_points[:,1]
|
| 122 |
rayst3r_points[:,2] = -rayst3r_points[:,2]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 123 |
|
| 124 |
# make all points red
|
| 125 |
colors = colorize_points_with_turbo_all_dims(rayst3r_points)
|
|
@@ -154,8 +160,8 @@ def input_to_glb(outdir,img,depth_dict,mask,rotated=False):
|
|
| 154 |
|
| 155 |
if rotated:
|
| 156 |
pts_swapped = np.copy(pts)
|
| 157 |
-
pts_swapped[:,0] = pts[:,1]
|
| 158 |
-
pts_swapped[:,1] = pts[:,0]
|
| 159 |
pts = pts_swapped
|
| 160 |
|
| 161 |
pct = trimesh.PointCloud(pts, colors=col.reshape(-1,3))
|
|
|
|
| 120 |
|
| 121 |
rayst3r_points[:,1] = -rayst3r_points[:,1]
|
| 122 |
rayst3r_points[:,2] = -rayst3r_points[:,2]
|
| 123 |
+
|
| 124 |
+
if rotated:
|
| 125 |
+
pts_swapped = np.copy(rayst3r_points)
|
| 126 |
+
pts_swapped[:,0] = -rayst3r_points[:,1]
|
| 127 |
+
pts_swapped[:,1] = -rayst3r_points[:,0]
|
| 128 |
+
rayst3r_points = pts_swapped
|
| 129 |
|
| 130 |
# make all points red
|
| 131 |
colors = colorize_points_with_turbo_all_dims(rayst3r_points)
|
|
|
|
| 160 |
|
| 161 |
if rotated:
|
| 162 |
pts_swapped = np.copy(pts)
|
| 163 |
+
pts_swapped[:,0] = -pts[:,1]
|
| 164 |
+
pts_swapped[:,1] = -pts[:,0]
|
| 165 |
pts = pts_swapped
|
| 166 |
|
| 167 |
pct = trimesh.PointCloud(pts, colors=col.reshape(-1,3))
|