From 49771dfe3eb1ed6bf0a75d3846541e4c3a5634f4 Mon Sep 17 00:00:00 2001 From: Carsen Stringer Date: Thu, 15 Feb 2024 16:39:16 -0500 Subject: [PATCH] bug in normalize params for training --- cellpose/gui/gui.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cellpose/gui/gui.py b/cellpose/gui/gui.py index a9f4534c..acf19331 100644 --- a/cellpose/gui/gui.py +++ b/cellpose/gui/gui.py @@ -1742,15 +1742,16 @@ def set_normalize_params(self, normalize_params): def check_percentile_params(self, percentile): # check normalization params - if not (percentile[0] >= 0 and percentile[1] > 0 and percentile[0] < 100 and percentile[1] <= 100 + if percentile is not None and not (percentile[0] >= 0 and percentile[1] > 0 and percentile[0] < 100 and percentile[1] <= 100 and percentile[1] > percentile[0]): print('GUI_ERROR: percentiles need be between 0 and 100, and upper > lower, using defaults') self.norm_edits[0].setText('1.') self.norm_edits[1].setText('99.') percentile = [1., 99.] - else: - self.norm_edits[0].setText(str(percentile[0])) - self.norm_edits[1].setText(str(percentile[1])) + elif percentile is None: + percentile = [1., 99.] + self.norm_edits[0].setText(str(percentile[0])) + self.norm_edits[1].setText(str(percentile[1])) return percentile def check_filter_params(self, sharpen, smooth, tile_norm, smooth3D, norm3D, invert): @@ -1967,7 +1968,6 @@ def train_model(self, restore=None, normalize_params=normalize_default): model_type = None self.logger.info(f'training new model starting from scratch') self.current_model = model_type - self.channels = self.get_channels() self.logger.info(f'training with chan = {self.ChannelChoose[0].currentText()}, chan2 = {self.ChannelChoose[1].currentText()}') @@ -1980,7 +1980,7 @@ def train_model(self, restore=None, normalize_params=normalize_default): self.new_model_path = train.train_seg(self.model.net, train_data=self.train_data, train_labels=self.train_labels, channels=self.channels, - normalize=self.get_normalize_params(), + normalize=normalize_params, min_train_masks=0, save_path=save_path, nimg_per_epoch=8,