From 4e5e3b561b44554933a641bc81f6e73d86ec66b6 Mon Sep 17 00:00:00 2001
From: Fabien Servant <fabien.servant@technicolor.com>
Date: Thu, 7 Dec 2023 11:53:58 +0100
Subject: [PATCH] option for gpu in segmentation

---
 meshroom/nodes/aliceVision/ImageSegmentation.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/meshroom/nodes/aliceVision/ImageSegmentation.py b/meshroom/nodes/aliceVision/ImageSegmentation.py
index 05cf592251..4ce6e91025 100644
--- a/meshroom/nodes/aliceVision/ImageSegmentation.py
+++ b/meshroom/nodes/aliceVision/ImageSegmentation.py
@@ -57,6 +57,13 @@ class ImageSegmentation(desc.AVCommandLineNode):
             value=False,
             uid=[0],
         ),
+        desc.BoolParam(
+            name="useGpu",
+            label="Use GPU",
+            description="Use GPU for computation if available",
+            value=True,
+            uid=[],
+        ),
         desc.ChoiceParam(
             name="verboseLevel",
             label="Verbose Level",