From f587b69f2736a795ce09bbabe18e3267a466cab7 Mon Sep 17 00:00:00 2001
From: ptresson <paul.tresson@ird.fr>
Date: Wed, 4 Sep 2024 18:20:36 +0200
Subject: [PATCH] add cancel feedbacks during preprocessing, solves #2
---
encoder.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/encoder.py b/encoder.py
index 7375b9e..0b6ff43 100644
--- a/encoder.py
+++ b/encoder.py
@@ -364,6 +364,11 @@ class EncoderAlgorithm(QgsProcessingAlgorithm):
mint=dataset.index.bounds[4], maxt=dataset.index.bounds[5])
+ if feedback.isCanceled():
+ feedback.pushWarning(
+ self.tr("\n !!!Processing is canceled by user!!! \n"))
+ return
+
feedback.pushInfo(f'create model')
model = timm.create_model(
self.backbone_name,
@@ -372,6 +377,11 @@ class EncoderAlgorithm(QgsProcessingAlgorithm):
num_classes=0
)
+ if feedback.isCanceled():
+ feedback.pushWarning(
+ self.tr("\n !!!Processing is canceled by user!!! \n"))
+ return
+
feedback.pushInfo(f'model done')
data_config = timm.data.resolve_model_data_config(model)
_, h, w, = data_config['input_size']
--
GitLab