Upload Colony_Analyzer_AI_zstack2_HF.py

Colony_Analyzer_AI_zstack2_HF.py

@@ -355,7 +355,9 @@ def main(args):
     caption = np.ones((150, 2068, 3), dtype=np.uint8) * 255  # Multiply by 255 to make it white
     cv2.putText(caption, "Total area necrotic: "+str(total_area_dark)+ ", Total area living: "+str(total_area_light)+", Ratio: "+str(ratio), (40, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 3)
 
-
-
+    print('img ndim: ' +str(img.ndim))
+    print('caption ndim: ' +str(caption.ndim))
+    if img.ndim == 2:
+        img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
     cv2.imwrite('Group_analysis_results.png', np.vstack((img, caption)))
     return(np.vstack((img, caption)), 'Group_analysis_results.png', 'Group_analysis_results.xlsx')