Rename Colony_Analyzer_AI2_HF.py to Organoid_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py → Organoid_Analyzer_AI2_HF.py

@@ -243,7 +243,7 @@ def analyze_colonies(mask, size_cutoff, circ_cutoff, img):
                 nec_list.append(nec)
 
         data.append({
-            "colony_area": colony_area,
+            "organoid_area": colony_area,
             "necrotic_area": necrosis_area,
             "centroid": centroid,
             "percent_necrotic": necrosis_area/colony_area,
@@ -273,7 +273,7 @@ def main(args):
     	caption = np.ones((150, 2048, 3), dtype=np.uint8) * 255  # Multiply by 255 to make it white
     	cv2.putText(caption, 'No colonies detected.', (40, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 3)
     	cv2.imwrite('results.png', np.vstack((img, caption)))
-    	colonies = pd.DataFrame({"Colony Number":[], 'Colony volume':[], "colony_area":[],'mean_pixel_value':[], "centroid":[], "necrotic_area":[],"percent_necrotic":[]})
+    	colonies = pd.DataFrame({"organoid_number":[], 'organoid_volume':[], "organoid_area":[],'mean_pixel_value':[], "centroid":[], "necrotic_area":[],"percent_necrotic":[]})
     	with pd.ExcelWriter('results.xlsx') as writer:
         	colonies.to_excel(writer, sheet_name="Colony data", index=False)
     	return(np.vstack((img, caption)), 'results.png', 'results.xlsx')
@@ -283,8 +283,8 @@ def main(args):
     img = cv2.copyMakeBorder(img,top=0, bottom=10,left=0,right=10, borderType=cv2.BORDER_CONSTANT,  value=[255, 255, 255]) 
     #print(colonies.to_string())
     
-    colonies = colonies.sort_values(by=["colony_area"], ascending=False)
-    colonies = colonies[colonies["colony_area"]>= min_size]
+    colonies = colonies.sort_values(by=["organoid_area"], ascending=False)
+    colonies = colonies[colonies["organoid_area"]>= min_size]
     colonies.index = range(1,len(colonies.index)+1) 
     
     for i in range(len(colonies)): 
@@ -300,31 +300,31 @@ def main(args):
     colonies = colonies.drop('contour', axis=1)
     colonies = colonies.drop('nec_contours', axis=1)
     
-    colonies.insert(loc=0, column="Colony Number", value=[str(x) for x in range(1, len(colonies)+1)])
+    colonies.insert(loc=0, column="organoid_number", value=[str(x) for x in range(1, len(colonies)+1)])
     total_area_dark = sum(colonies['necrotic_area'])
-    total_area_light = sum(colonies['colony_area'])
+    total_area_light = sum(colonies['organoid_area'])
     ratio = total_area_dark/(abs(total_area_light)+1)
-    radii = [np.sqrt(x/3.1415) for x in list(colonies['colony_area'])]
+    radii = [np.sqrt(x/3.1415) for x in list(colonies['organoid_area'])]
     volumes = [4.189*(x**3) for x in radii]
-    colonies['Colony volume'] = volumes
+    colonies['organoid_volume'] = volumes
     del radii, volumes
-    meanpix = sum(colonies['mean_pixel_value'] * colonies['colony_area'])/total_area_light
-    colonies.loc[len(colonies)+1] = ["Total", total_area_light, total_area_dark, None, ratio, meanpix, sum(colonies['Colony volume'])]
+    meanpix = sum(colonies['mean_pixel_value'] * colonies['organoid_area'])/total_area_light
+    colonies.loc[len(colonies)+1] = ["Total", total_area_light, total_area_dark, None, ratio, meanpix, sum(colonies['organoid_volume'])]
     del meanpix
-    colonies = colonies[["Colony Number", 'Colony volume', "colony_area",'mean_pixel_value', "centroid", "necrotic_area","percent_necrotic"]]
+    colonies = colonies[["organoid_number", 'organoid_volume', "organoid_area",'mean_pixel_value', "centroid", "necrotic_area","percent_necrotic"]]
     if do_necrosis == False:
         colonies = colonies.drop('necrotic_area', axis=1)
         colonies = colonies.drop('percent_necrotic', axis=1)
-    Parameters = pd.DataFrame({"Minimum colony size in pixels":[min_size], "Minimum colony circularity":[min_circ]})
+    Parameters = pd.DataFrame({"Minimum organoid size in pixels":[min_size], "Minimum organoid circularity":[min_circ]})
     with pd.ExcelWriter('results.xlsx') as writer:
-        colonies.to_excel(writer, sheet_name="Colony data", index=False)
+        colonies.to_excel(writer, sheet_name="Organoid data", index=False)
         Parameters.to_excel(writer, sheet_name="Parameters", index=False)
     caption = np.ones((150, 2068, 3), dtype=np.uint8) * 255  # Multiply by 255 to make it white
     if do_necrosis == True:
         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)
     else:
         cv2.putText(caption, "Total area: "+str(total_area_light), (40, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 3)
-    cv2.putText(caption, "Total number of colonies: "+str(len(colonies)-1), (40, 110), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 3)
+    cv2.putText(caption, "Total number of organoids: "+str(len(colonies)-1), (40, 110), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 3)
 
 
     cv2.imwrite('results.png', np.vstack((img, caption)))