[Nodes] ColorCheckerDetection: Add color chart type option and add cn…

…n model folder setting.
alicevision · Sep 27, 2023 · d8362c1 · d8362c1
1 parent 75422c7
commit d8362c1
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/meshroom/nodes/aliceVision/ColorCheckerDetection.py b/meshroom/nodes/aliceVision/ColorCheckerDetection.py
@@ -35,6 +35,25 @@ class ColorCheckerDetection(desc.AVCommandLineNode):
             value="",
             uid=[0],
         ),
+        desc.ChoiceParam(
+            name="ccType",
+            label="Colorchart Type",
+            description="Colorchart type:\n"
+                        " - mcc24   classical macbeth 24 patches\n"
+                        " - sg140   Digital SG 140 patches\n"
+                        " - Vinyl18 DKK colorchart 12 patches + 6 rectangles)",
+            value="mcc24",
+            values=["mcc24", "sg140", "Vinyl18"],
+            exclusive=True,
+            uid=[0],
+        ),
+        desc.File(
+            name="modelFolder",
+            label="Model Folder",
+            description="DNN model directory path.",
+            value="${ALICEVISION_COLORCHARTDETECTION_MODEL_FOLDER}",
+            uid=[0],
+        ),
         desc.IntParam(
             name="maxCount",
             label="Max Count By Image",