fix: guess initial values for image method when edge detection method…

… fails
RI-imaging · Aug 23, 2019 · f04ff78 · f04ff78
1 parent 9090868
commit f04ff78
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,6 @@
+0.5.2
+ - fix: guess initial values for image method when edge detection
+   method fails
 0.5.1
  - maintenance release
 0.5.0

diff --git a/qpsphere/cnvnc.py b/qpsphere/cnvnc.py
@@ -87,12 +87,18 @@ def analyze(qpi, r0, method="edge", model="projection", edgekw={}, imagekw={},
                                center=c)
             res.append(qpi_sim)
     elif method == "image":
-        n0, r0, c0 = edgefit.analyze(qpi=qpi,
-                                     r0=r0,
-                                     edgekw=edgekw,
-                                     ret_center=True,
-                                     ret_edge=False,
-                                     )
+        try:
+            n0, r0, c0 = edgefit.analyze(qpi=qpi,
+                                         r0=r0,
+                                         edgekw=edgekw,
+                                         ret_center=True,
+                                         ret_edge=False,
+                                         )
+        except edgefit.EdgeDetectionError:
+            # proceed with best guess
+            c0 = np.array(qpi.shape) / 2
+            n0 = qpi["medium index"] + np.sign(np.sum(qpi.pha)) * .01
+
         res = imagefit.analyze(qpi=qpi,
                                model=model,
                                n0=n0,