Merge pull request #2083 from alicevision/build/sphereDetectionSetup

Set `ALICEVISION_SPHERE_DETECTION_MODEL` variable during the initialisation
alicevision · Jun 26, 2023 · bdb3a0a · bdb3a0a
2 parents 851099d + b8b6568
commit bdb3a0a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/meshroom/__init__.py b/meshroom/__init__.py
@@ -120,6 +120,7 @@ def addToEnvPath(var, val, index=-1):
         qtPluginsDir = os.path.join(rootDir, "qtPlugins")
         sensorDBPath = os.path.join(aliceVisionShareDir, "cameraSensors.db")
         voctreePath = os.path.join(aliceVisionShareDir, "vlfeat_K80L3.SIFT.tree")
+        sphereDetectionModel = os.path.join(aliceVisionShareDir, "sphereDetection_Mask-RCNN.onnx")
 
         env = {
             'PATH': aliceVisionBinDir,
@@ -134,7 +135,8 @@ def addToEnvPath(var, val, index=-1):
         variables = {
             "ALICEVISION_ROOT": aliceVisionDir,
             "ALICEVISION_SENSOR_DB": sensorDBPath,
-            "ALICEVISION_VOCTREE": voctreePath
+            "ALICEVISION_VOCTREE": voctreePath,
+            "ALICEVISION_SPHERE_DETECTION_MODEL": sphereDetectionModel
         }
 
         for key, value in variables.items():