eliphatfs commited on
Commit ·
44a6c30
1
Parent(s): 1caa49c
Fix.
Browse files- openshape/demo/misc_utils.py +11 -4
openshape/demo/misc_utils.py
CHANGED
|
@@ -99,10 +99,17 @@ def trimesh_to_pc(scene_or_mesh):
|
|
| 99 |
|
| 100 |
|
| 101 |
def input_3d_shape(key=None):
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
f32 = numpy.float32
|
| 107 |
|
| 108 |
def load_data(prog):
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
def input_3d_shape(key=None):
|
| 102 |
+
if key is None:
|
| 103 |
+
objaid_key = model_key = npy_key = swap_key = None
|
| 104 |
+
else:
|
| 105 |
+
objaid_key = key + "_objaid"
|
| 106 |
+
model_key = key + "_model"
|
| 107 |
+
npy_key = key + "_npy"
|
| 108 |
+
swap_key = key + "_swap"
|
| 109 |
+
objaid = st.text_input("Enter an Objaverse ID", key=objaid_key)
|
| 110 |
+
model = st.file_uploader("Or upload a model (.glb/.obj/.ply)", key=model_key)
|
| 111 |
+
npy = st.file_uploader("Or upload a point cloud numpy array (.npy of Nx3 XYZ or Nx6 XYZRGB)", key=npy_key)
|
| 112 |
+
swap_yz_axes = st.checkbox("Swap Y/Z axes of input (Y is up for OpenShape)", key=swap_key)
|
| 113 |
f32 = numpy.float32
|
| 114 |
|
| 115 |
def load_data(prog):
|