care_plugin

class CAREPlugin(BasePlugin):
    """CAREamics CARE plugin.

    Parameters
    ----------
    napari_viewer : napari.Viewer or None, default=None
        Napari viewer.
    """

    def __init__(
        self,
        napari_viewer: napari.Viewer | None = None,
    ) -> None:
        """Initialize the plugin.

        Parameters
        ----------
        napari_viewer : napari.Viewer or None, default=None
            Napari viewer.
        """
        super().__init__(napari_viewer)
        self.viewer = napari_viewer

        # create a care config
        self.careamics_config = get_default_care_config()
        # advanced care config
        self.advanced_config = CAREAdvancedConfig()

        # assemble plugin ui
        self.add_careamics_banner("CAREamics UI for training CARE denoising models.")
        self.add_train_input_ui(use_target=self.careamics_config.needs_gt)
        self.add_config_ui()
        self.add_train_button_ui()
        self.add_prediction_ui()
        self.add_model_export_ui()

    def show_advanced_config(self) -> None:
        """Show advanced configuration."""
        # update axes in configuration
        self.config_widget.axes_widget.update_config()

        # show window with advanced options
        win = CAREConfigurationWindow(self, self.careamics_config, self.advanced_config)
        win.finished.connect(lambda: print(self.advanced_config, self.careamics_config))
        win.show()