Spaces:
Paused
Paused
Upload simam3d_core.py with huggingface_hub
Browse files- simam3d_core.py +2 -0
simam3d_core.py
CHANGED
|
@@ -77,6 +77,8 @@ def synthetic_orbit_poses(view_count: int, radius: float = 0.35) -> np.ndarray:
|
|
| 77 |
"""
|
| 78 |
if view_count < 1:
|
| 79 |
return np.empty((0, 4, 4), dtype=np.float32)
|
|
|
|
|
|
|
| 80 |
poses = []
|
| 81 |
up = np.array([0.0, 1.0, 0.0], dtype=np.float32)
|
| 82 |
for index in range(view_count):
|
|
|
|
| 77 |
"""
|
| 78 |
if view_count < 1:
|
| 79 |
return np.empty((0, 4, 4), dtype=np.float32)
|
| 80 |
+
if view_count == 1:
|
| 81 |
+
return np.eye(4, dtype=np.float32)[None, ...]
|
| 82 |
poses = []
|
| 83 |
up = np.array([0.0, 1.0, 0.0], dtype=np.float32)
|
| 84 |
for index in range(view_count):
|