Upload Colony_Analyzer_AI_zstack2_HF.py

Colony_Analyzer_AI_zstack2_HF.py

@@ -284,7 +284,12 @@ def main(args):
         src_image = np.array(files[x])
         roi = cv2.bitwise_and(src_image, src_image, mask=mask)
         # Paste the extracted regions onto the destination image
-        np.copyto(img, roi, where=(mask[..., None] == 255))
+        print("ROI Shape:", roi.shape)
+        print("img Shape:", img.shape)
+        print("mask Shape:", mask.shape)
+        mask = np.repeat(mask[..., None], 3, axis=2)
+        print("mask Shape fixed:", mask.shape)    
+        np.copyto(img, roi, where=(mask== 255))
     try:
         del x, mask, src_image, roi, best, contours
     except: