fixing brush size bug with reset of image, and train size window increas

MouseLand · May 23, 2024 · c03958c · c03958c
1 parent 40a0c7d
commit c03958c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cellpose/gui/gui.py b/cellpose/gui/gui.py
@@ -824,6 +824,7 @@ def make_buttons(self):
         b0 += 3
         self.norm3D_cb = QCheckBox("norm3D")
         self.norm3D_cb.setFont(self.medfont)
+        self.norm3D_cb.setChecked(True)
         self.norm3D_cb.setToolTip("run same normalization across planes")
         self.filtBoxG.addWidget(self.norm3D_cb, b0, 0, 1, 3)
 
@@ -1233,7 +1234,6 @@ def reset(self):
         self.ViewDropDown.model().item(self.ViewDropDown.count() - 1).setEnabled(False)
         self.delete_restore()
 
-        self.BrushChoose.setCurrentIndex(1)
         self.clear_all()
 
         #self.update_plot()
@@ -1942,12 +1942,13 @@ def get_normalize_params(self):
         ]
         self.check_percentile_params(percentile)
         normalize_params = {"percentile": percentile}
+        norm3D = self.norm3D_cb.isChecked()
+        normalize_params["norm3D"] = norm3D
         if self.restore == "filter":
             sharpen = float(self.filt_edits[0].text())
             smooth = float(self.filt_edits[1].text())
             tile_norm = float(self.filt_edits[2].text())
             smooth3D = float(self.filt_edits[3].text())
-            norm3D = self.norm3D_cb.isChecked()
             invert = self.invert_cb.isChecked()
             out = self.check_filter_params(sharpen, smooth, tile_norm, smooth3D, norm3D,
                                            invert)
@@ -1956,7 +1957,6 @@ def get_normalize_params(self):
             normalize_params["smooth_radius"] = smooth
             normalize_params["tile_norm_blocksize"] = tile_norm
             normalize_params["tile_norm_smooth3D"] = smooth3D
-            normalize_params["norm3D"] = norm3D
             normalize_params["invert"] = invert
 
         from cellpose.models import normalize_default

diff --git a/cellpose/gui/guiparts.py b/cellpose/gui/guiparts.py
@@ -185,7 +185,7 @@ class TrainWindow(QDialog):
 
     def __init__(self, parent, model_strings):
         super().__init__(parent)
-        self.setGeometry(100, 100, 900, 350)
+        self.setGeometry(100, 100, 900, 550)
         self.setWindowTitle("train settings")
         self.win = QWidget(self)
         self.l0 = QGridLayout()