Method load_model

qgis-samgeo-plugin/samgeo_plugin.py:770–824

Load the SamGeo model.

(self)

            self.output_path_edit.setText(file_path)

    def load_model(self):
        """Load the SamGeo model."""
        try:
            self.progress_bar.setVisible(True)
            self.progress_bar.setRange(0, 0)  # Indeterminate
            self.model_status.setText("Loading model...")
            self.model_status.setStyleSheet("color: orange;")
            QCoreApplication.processEvents()

            model_version = self.model_combo.currentText()
            backend = self.backend_combo.currentText()
            device = self.device_combo.currentText()
            if device == "auto":
                device = None

            confidence = self.conf_spin.value()
            enable_interactive = self.interactive_check.isChecked()

            # Import and initialize the appropriate model
            if "SamGeo3" in model_version:
                from samgeo import SamGeo3

                self.sam = SamGeo3(
                    backend=backend,
                    device=device,
                    confidence_threshold=confidence,
                    enable_inst_interactivity=enable_interactive,
                )
                model_name = "SamGeo3"
            elif "SamGeo2" in model_version:
                from samgeo import SamGeo2

                self.sam = SamGeo2(
                    device=device,
                )
                model_name = "SamGeo2"
            else:
                from samgeo import SamGeo

                self.sam = SamGeo(
                    device=device,
                )
                model_name = "SamGeo"

            self.model_status.setText(f"Model: {model_name} loaded")
            self.model_status.setStyleSheet("color: green;")
            self.log_message(f"{model_name} model loaded successfully")

        except Exception as e:
            self.model_status.setText("Model: Failed to load")
            self.model_status.setStyleSheet("color: red;")
            self.show_error(f"Failed to load model: {str(e)}")

        finally:
            self.progress_bar.setVisible(False)

    def set_image_from_layer(self):
        """Set the image from the selected QGIS layer."""