57e82c37bd1e40f9fb37b21719fda4ebadd095de,preparation.py,,overlay_contours,#Any#Any#Any#,49

Before Change

    for mask_dirname in tqdm(glob.glob("{}/*/masks".format(train_dir))):
        masks = []
        for image_filepath in glob.glob("{}/*".format(mask_dirname)):
            image = plt.imread(image_filepath)
            masks.append(get_contour(image))
        overlayed_masks = np.where(np.sum(masks, axis=0) > 128., 255., 0.).astype(np.uint8)

        target_filepath = "/".join(mask_dirname.replace(images_dir, target_dir).split("/")[:-1]) + ".png"

After Change

    for mask_dirname in tqdm(glob.glob("{}/*/masks".format(train_dir))):
        masks = []
        for image_filepath in glob.glob("{}/*".format(mask_dirname)):
            image = np.asarray(Image.open(image_filepath))
            image = image / image.max(axis=None).astype(np.float32)
            masks.append(get_contour(image))
        overlayed_masks = np.where(np.sum(masks, axis=0) > 128., 255., 0.).astype(np.uint8)
        target_filepath = "/".join(mask_dirname.replace(images_dir, target_dir).split("/")[:-1]) + ".png"
        os.makedirs(os.path.dirname(target_filepath), exist_ok=True)

Instances