n2n_plugin

class N2NPlugin(BasePlugin):
    """CAREamics N2N 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 n2n config
        self.careamics_config = get_default_n2n_config()
        # advanced n2n config
        self.advanced_config = N2NAdvancedConfig()

        # assemble plugin ui
        self.add_careamics_banner(
            "CAREamics UI for training Noise2Noise (N2N) 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 = N2NConfigurationWindow(self, self.careamics_config, self.advanced_config)
        win.finished.connect(lambda: print(self.advanced_config, self.careamics_config))
        win.show()