junaid-simamdigital commited on
Commit
6cac5c3
·
verified ·
1 Parent(s): 07188e3

Upload simam3d_core.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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):