Changed to package without main.

pyproject.toml

@@ -16,6 +16,3 @@ imagelab = {path = "../imagelab", develop = true}
 [build-system]
 build-backend = "poetry.core.masonry.api"
 requires = ["poetry-core>=1.0.0", "setuptools"]
-
-[tool.poetry.scripts]
-photometric_stereo = "photometric_stereo:"

src/photometric_stereo/__init__.py

+from . import calibrate

src/photometric_stereo/__main__.py

-from imagelab import io, visionhardware
-
-cameras = visionhardware.getCameras()
-
-if len(cameras) < 1:
-    exit(1)
-
-im = cameras[0].capture_single_frame()
-
-io.imwrite("test.png", im)

src/photometric_stereo/calibrate.py

+from imagelab import imageanalysis, utilities
+import numpy as np
+
+
+def detect_spheres(*images):
+    boxes = []
+    for image in images:
+        gray = utilities.grayscale(image)
+        nonblack = np.zeros_like(gray.shape[-3:], dtype=bool)
+        for idx in np.nditer(gray.shape[:-3]):
+            nonblack |= gray[idx] > 10
+        boxes += [imageanalysis.detect_ellipse(255 * nonblack.astype('u1'))]
+    if len(boxes) == 1:
+        boxes = boxes[0]
+    return boxes