Upload Colony_Analyzer_AI_zstack2_HF.py

Colony_Analyzer_AI_zstack2_HF.py

@@ -275,7 +275,8 @@ def main(args):
     for x in range(len(files)):
         if x == best:
             continue
-        mask = np.zeros_like(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
+        #mask = np.zeros_like(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
+        mask = np.zeros_like(img)
         contours = colonies[colonies["source"]==x]
         contours = list(contours["contour"])
         cv2.drawContours(mask, contours, -1, 255, thickness=cv2.FILLED)