fallback for std 0

brisvag · Aug 3, 2023 · aa42f18 · aa42f18
1 parent 3124d95
commit aa42f18
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stemia/imod/find_NAD_params.py b/stemia/imod/find_NAD_params.py
@@ -23,6 +23,9 @@ def cli(input, k_values, iterations, std):
     if std is None:
         with mrcfile.open(input, header_only=True) as mrc:
             std = mrc.header.rms.item()
+        if std == 0:
+            with mrcfile.open(input) as mrc:
+                std = mrc.data.std()
 
     ks = [float(k) for k in k_values.split(',')]
     max_it = max(float(it) for it in iterations.split(','))