From 04d476d85b0ca45b0806fa702b4f8ba91f8c6371 Mon Sep 17 00:00:00 2001 From: kasra-hosseini Date: Tue, 11 Jan 2022 17:22:30 +0000 Subject: [PATCH] Add min_mean_pixel argument --- mapreader/annotate/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/mapreader/annotate/utils.py b/mapreader/annotate/utils.py index 03b36c78..da4c3357 100644 --- a/mapreader/annotate/utils.py +++ b/mapreader/annotate/utils.py @@ -247,6 +247,7 @@ def prepare_annotation(userID, tree_level="child", sortby=None, min_alpha_channel=None, + min_mean_pixel=None, context_image=False, xoffset=500, yoffset=500, @@ -315,7 +316,7 @@ def prepare_annotation(userID, mymaps.add_metadata(metadata=annot_file, index_col=-1, delimiter=",", tree_level=tree_level) # Calculate mean before converting to pandas so the dataframe contains information about mean pixel intensity - if sortby == "mean" or isinstance(min_alpha_channel, float): + if sortby == "mean" or isinstance(min_alpha_channel, float) or isinstance(min_mean_pixel, float): mymaps.calc_pixel_stats(calc_std=False) # convert images to dataframe @@ -327,6 +328,10 @@ def prepare_annotation(userID, if isinstance(min_alpha_channel, float): if "mean_pixel_A" in sliced_df.columns: sliced_df = sliced_df[sliced_df["mean_pixel_A"] >= min_alpha_channel] + + if isinstance(min_mean_pixel, float): + if "mean_pixel_RGB" in sliced_df.columns: + sliced_df = sliced_df[sliced_df["mean_pixel_RGB"] >= min_mean_pixel] col_names = ["image_path", "parent_id"] else: