Draft Reconstruction pipeline

alicevision · Sep 13, 2021 · d30571c · d30571c
1 parent 1a9692d
commit d30571c
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 2 deletions.
diff --git a/bin/meshroom_batch b/bin/meshroom_batch
@@ -20,8 +20,8 @@ parser.add_argument('-I', '--inputRecursive', metavar='FOLDERS/IMAGES', type=str
                     default=[],
                     help='Input folders containing all images recursively.')
 
-parser.add_argument('-p', '--pipeline', metavar='photogrammetry/panoramaHdr/panoramaFisheyeHdr/MG_FILE', type=str, default='photogrammetry',
-                    help='"photogrammetry", "panoramaHdr", "panoramaFisheyeHdr", "cameraTracking" pipeline or a Meshroom file containing a custom pipeline to run on input images. '
+parser.add_argument('-p', '--pipeline', metavar='photogrammetry/panoramaHdr/panoramaFisheyeHdr/cameraTracking/photogrammetryDraft/MG_FILE', type=str, default='photogrammetry',
+                    help='"photogrammetry", "panoramaHdr", "panoramaFisheyeHdr", "cameraTracking", "photogrammetryDraft" pipeline or a Meshroom file containing a custom pipeline to run on input images. '
                          'Requirements: the graph must contain one CameraInit node, '
                          'and one Publish node if --output is set.')
 

diff --git a/meshroom/multiview.py b/meshroom/multiview.py
@@ -557,3 +557,47 @@ def photogrammetryAndCameraTracking(inputImages=list(), inputViewpoints=list(),
 
     return graph
 
+
+def photogrammetryDraft(inputImages=None, inputViewpoints=None, inputIntrinsics=None, output='', graph=None):
+    """
+    Create a new Graph with a complete photogrammetry pipeline without requiring a NVIDIA CUDA video card. Something also named Draft Meshing.
+    More information on that pipeline https://github.com/alicevision/meshroom/wiki/Draft-Meshing
+
+    Args:
+        inputImages (list of str, optional): list of image file paths
+        inputViewpoints (list of Viewpoint, optional): list of Viewpoints
+        output (str, optional): the path to export reconstructed model to
+
+    Returns:
+        Graph: the created graph
+    """
+    if not graph:
+        graph = Graph('PhotogrammetryDraft')
+    with GraphModification(graph):
+        sfmNodes = sfmPipeline(graph)
+        sfmNode = sfmNodes[-1]
+
+        meshing = graph.addNewNode('Meshing',
+                                   input=sfmNode.output)
+
+        meshFiltering = graph.addNewNode('MeshFiltering',
+                                         inputMesh=meshing.outputMesh)
+        texturing = graph.addNewNode('Texturing',
+                                     input=meshing.output,
+                                     inputMesh=meshFiltering.outputMesh)
+
+        cameraInit = sfmNodes[0]
+
+        if inputImages:
+            cameraInit.viewpoints.extend([{'path': image} for image in inputImages])
+        if inputViewpoints:
+            cameraInit.viewpoints.extend(inputViewpoints)
+        if inputIntrinsics:
+            cameraInit.intrinsics.extend(inputIntrinsics)
+
+        if output:
+            graph.addNewNode('Publish', output=output, inputFiles=[texturing.outputMesh,
+                                                                   texturing.outputMaterial,
+                                                                   texturing.outputTextures])
+
+    return graph
diff --git a/meshroom/ui/qml/main.qml b/meshroom/ui/qml/main.qml
@@ -436,6 +436,10 @@ ApplicationWindow {
                     text: "Camera Tracking (experimental)"
                     onTriggered: ensureSaved(function() { _reconstruction.new("cameratracking") })
                 }
+                Action {
+                    text: "Photogrammetry Draft (No CUDA)"
+                    onTriggered: ensureSaved(function() { _reconstruction.new("photogrammetrydraft") })
+                }
             }
             Action {
                 id: openActionItem

diff --git a/meshroom/ui/reconstruction.py b/meshroom/ui/reconstruction.py
@@ -496,6 +496,9 @@ def new(self, pipeline=None):
         elif p.lower() == "cameratracking":
             # default camera tracking pipeline
             self.setGraph(multiview.cameraTracking())
+        elif p.lower() == "photogrammetrydraft":
+            # photogrammetry pipeline in draft mode (no cuda)
+            self.setGraph(multiview.photogrammetryDraft())
         else:
             # use the user-provided default photogrammetry project file
             self.load(p, setupProjectFile=False)