[multiview] add arguments to directly provide input images to the pipeline

This commit is contained in:
Fabien Castan 2017-10-31 10:48:06 +01:00
parent 46f6865a25
commit 2e035d5a7a
2 changed files with 5 additions and 6 deletions

View file

@ -3,10 +3,12 @@ import os
from .core.graph import Graph
def photogrammetryPipeline():
def photogrammetryPipeline(inputFolder='', inputImages=[], inputViewpoints=[]):
# type: () -> Graph
graph = Graph('pipeline')
cameraInit = graph.addNewNode('CameraInit')
cameraInit = graph.addNewNode('CameraInit', imageDirectory=inputFolder)
cameraInit.viewpoints.value = [{'image': image, 'focal': -1} for image in inputImages]
cameraInit.viewpoints.extend(inputViewpoints)
featureExtraction = graph.addNewNode('FeatureExtraction',
input=cameraInit.outputSfm)
imageMatching = graph.addNewNode('ImageMatching',