mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-05-17 11:06:27 +02:00
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
__version__ = "1.1"
|
|
|
|
import os
|
|
from meshroom.core import desc
|
|
|
|
|
|
class ImageMatching(desc.CommandLineNode):
|
|
commandLine = 'aliceVision_imageMatching {allParams}'
|
|
size = desc.DynamicNodeSize('input')
|
|
|
|
inputs = [
|
|
desc.File(
|
|
name='input',
|
|
label='Input',
|
|
description='SfMData file .',
|
|
value='',
|
|
uid=[0],
|
|
),
|
|
desc.ListAttribute(
|
|
elementDesc=desc.File(
|
|
name="featuresFolder",
|
|
label="Features Folder",
|
|
description="",
|
|
value="",
|
|
uid=[0],
|
|
),
|
|
name="featuresFolders",
|
|
label="Features Folders",
|
|
description="Folder(s) containing the extracted features and descriptors."
|
|
),
|
|
desc.ChoiceParam(
|
|
name='method',
|
|
label='Method',
|
|
description='Method used to select the image pairs to match.',
|
|
value='VocabularyTree',
|
|
values=['VocabularyTree', 'Sequential', 'SequentialAndVocabularyTree','Exhaustive','Frustum'],
|
|
exclusive=True,
|
|
uid=[0],
|
|
),
|
|
desc.File(
|
|
name='tree',
|
|
label='Tree',
|
|
description='Input name for the vocabulary tree file.',
|
|
value=os.environ.get('ALICEVISION_VOCTREE', ''),
|
|
uid=[],
|
|
),
|
|
desc.File(
|
|
name='weights',
|
|
label='Weights',
|
|
description='Input name for the weight file, if not provided the weights will be computed on the database built with the provided set.',
|
|
value='',
|
|
uid=[0],
|
|
advanced=True,
|
|
),
|
|
desc.IntParam(
|
|
name='minNbImages',
|
|
label='Minimal Number of Images',
|
|
description='Minimal number of images to use the vocabulary tree. If we have less features than this threshold, we will compute all matching combinations.',
|
|
value=200,
|
|
range=(0, 500, 1),
|
|
uid=[0],
|
|
advanced=True,
|
|
),
|
|
desc.IntParam(
|
|
name='maxDescriptors',
|
|
label='Max Descriptors',
|
|
description='Limit the number of descriptors you load per image. Zero means no limit.',
|
|
value=500,
|
|
range=(0, 100000, 1),
|
|
uid=[0],
|
|
advanced=True,
|
|
),
|
|
desc.IntParam(
|
|
name='nbMatches',
|
|
label='Nb Matches',
|
|
description='The number of matches to retrieve for each image (If 0 it will retrieve all the matches).',
|
|
value=50,
|
|
range=(0, 1000, 1),
|
|
uid=[0],
|
|
advanced=True,
|
|
),
|
|
desc.IntParam(
|
|
name='nbNeighbors',
|
|
label='Nb Neighbors',
|
|
description='The number of neighbors to retrieve for each image (If 0 it will retrieve all the neighbors).',
|
|
value=50,
|
|
range=(0, 1000, 1),
|
|
uid=[0],
|
|
advanced=True,
|
|
),
|
|
desc.ChoiceParam(
|
|
name='verboseLevel',
|
|
label='Verbose Level',
|
|
description='verbosity level (fatal, error, warning, info, debug, trace).',
|
|
value='info',
|
|
values=['fatal', 'error', 'warning', 'info', 'debug', 'trace'],
|
|
exclusive=True,
|
|
uid=[],
|
|
)
|
|
]
|
|
|
|
outputs = [
|
|
desc.File(
|
|
name='output',
|
|
label='Output List File',
|
|
description='Filepath to the output file with the list of selected image pairs.',
|
|
value=desc.Node.internalFolder + 'imageMatches.txt',
|
|
uid=[],
|
|
),
|
|
]
|