Passing ai, img_st, mask_st and inpain as arrays

gfreychet · Jan 26, 2024 · 93985ff · 93985ff
1 parent 94d6ffb
commit 93985ff
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/smi_analysis/SMI_beamline.py b/smi_analysis/SMI_beamline.py
@@ -36,11 +36,11 @@ def __init__(self,
         self.det_angle_step = det_angle_step
         self.det_angles = det_angles
 
-        self.ai = []
+        self.ai = np.array([])
         self.masks = []
         self.cake = []
-        self.inpaints, self.mask_inpaints = [], []
-        self.img_st, self.mask_st = [], []
+        self.inpaints, self.mask_inpaints = np.array([]), np.array([])
+        self.img_st, self.mask_st = np.array([]), np.array([])
         self.bs_kind = bs_kind
         self.scales = 1
 
@@ -223,8 +223,8 @@ def calculate_integrator_gi(self, det_rots):
 
     def stitching_data(self, flag_scale=True, interp_factor=1):
         self.img_st, self.qp, self.qz = [], [], []
-
-        if self.ai == []:
+        
+        if not(self.ai.size):
             if len(self.det_angles) != len(self.imgs):
                 if self.detector != 'Pilatus900kw':
                     if len(self.det_angles) !=0 and len(self.det_angles) > len(self.imgs):