Parameterize deterministic multi-view fixtures

demo_pipeline.py

@@ -18,14 +18,16 @@ from simam3d_core import (
 )
 
 
-def build_demo_views(size: int = 48):
+def build_demo_views(size: int = 48, view_count: int = 3):
+    if view_count < 1:
+        raise ValueError("view_count must be positive")
     yy, xx = np.mgrid[0:size, 0:size]
     depth = (0.25 + 0.65 * ((xx / max(size - 1, 1) - 0.5) ** 2 + (yy / max(size - 1, 1) - 0.5) ** 2)).astype(np.float32)
     views = []
-    poses = synthetic_orbit_poses(3)
+    poses = synthetic_orbit_poses(view_count)
     for index, pose in enumerate(poses):
         colors = np.zeros((size, size, 3), dtype=np.uint8)
-        colors[..., index] = 220
+        colors[..., index % 3] = 220
         colors[..., (index + 1) % 3] = (xx * 255 // max(size - 1, 1)).astype(np.uint8)
         confidence = np.full_like(depth, 1.0 - index * 0.1)
         views.append((depth, confidence, None, pose, colors))