3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-04-29 10:17:27 +02:00
Code Issues Projects Releases Packages Wiki Activity

[bin] meshroom_batch: support multiple init nodes

Browse source
This commit is contained in:
Loïc Vital 2023-08-01 11:44:48 +02:00
parent 24b60b3d42
commit 91f3abdeaa
1 changed files with 36 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
bin/meshroom_batch
View file

@ -3,6 +3,8 @@ import argparse
import os
import sys
import distutils.util
import json
import re
import meshroom
meshroom.setupEnvironment()
@ -14,11 +16,11 @@ import logging
parser = argparse.ArgumentParser(description='Launch the full photogrammetry or Panorama HDR pipeline.')
parser.add_argument('-i', '--input', metavar='SFM/FOLDERS/IMAGES', type=str, nargs='*',
parser.add_argument('-i', '--input', metavar='NODEINSTANCE:SFM/FOLDERS/IMAGES;...', type=str, nargs='*',
default=[],
help='Input folder containing images or folders of images or file (.sfm or .json) '
'with images paths and optionally predefined camera intrinsics.')
parser.add_argument('-I', '--inputRecursive', metavar='FOLDERS/IMAGES', type=str, nargs='*',
parser.add_argument('-I', '--inputRecursive', metavar='NODEINSTANCE:FOLDERS/IMAGES;...', type=str, nargs='*',
default=[],
help='Input folders containing all images recursively.')
@ -90,21 +92,6 @@ logStringToPython = {
if args.verbose:
logging.getLogger().setLevel(logStringToPython[args.verbose])
def getInitNode(g):
"""
Helper function to get the Init node in the graph 'g' and raise an exception if there is no or
multiple candidates.
"""
nodes = g.findInitNodes()
if len(nodes) == 0:
raise RuntimeError("meshroom_batch requires an Init node in the pipeline.")
elif len(nodes) > 1:
raise RuntimeError("meshroom_batch requires exactly one Init node in the pipeline, {} found: {}"
.format(len(nodes), str(nodes)))
return nodes[0]
if not args.input and not args.inputRecursive:
print('Nothing to compute. You need to set --input or --inputRecursive.')
sys.exit(1)
@ -120,9 +107,38 @@ with multiview.GraphModification(graph):
# custom pipeline
graph.load(args.pipeline, setupProjectFile=False, publishOutputs=True if args.output else False)
# get init node and initialize it
initNode = getInitNode(graph)
initNode.nodeDesc.initialize(initNode, args.input, args.inputRecursive)
# get inputs for each init node
mapInput = {}
for inp in args.input:
inputGroup = inp.split(':')
if len(inputGroup) != 2:
print('Syntax error in input argument')
sys.exit(1)
nodeName = inputGroup[0]
nodeInputs = inputGroup[1].split(';')
mapInput[nodeName] = nodeInputs
# get recursive inputs for each init node
mapInputRecursive = {}
for inp in args.inputRecursive:
inputGroup = inp.split(':')
if len(inputGroup) != 2:
print('Syntax error in inputRecursive argument')
sys.exit(1)
nodeName = inputGroup[0]
nodeInputs = inputGroup[1].split(';')
mapInputRecursive[nodeName] = nodeInputs
# get init nodes and initialize them
initNodes = graph.findInitNodes()
for initNode in initNodes:
nodeName = initNode.getName()
inp = mapInput.get(nodeName)
inpRec = mapInputRecursive.get(nodeName)
if not inp and not inpRec:
print(f'Nothing to compute for node {nodeName}.')
sys.exit(1)
initNode.nodeDesc.initialize(initNode, inp, inpRec)
if not graph.canComputeLeaves:
raise RuntimeError("Graph cannot be computed. Check for compatibility issues.")
@ -142,7 +158,6 @@ with multiview.GraphModification(graph):
raise RuntimeError("meshroom_batch requires a pipeline graph with at least one Publish node, none found.")
if args.overrides:
import json
with open(args.overrides, 'r', encoding='utf-8', errors='ignore') as f:
data = json.load(f)
for nodeName, overrides in data.items():
@ -151,7 +166,6 @@ with multiview.GraphModification(graph):
if args.paramOverrides:
print("\n")
import re
reExtract = re.compile('(\w+)([:.])(\w[\w.]*)=(.*)')
for p in args.paramOverrides:
result = reExtract.match(p)