[nodes] new node CameraCalibration

meshroom/nodes/aliceVision/CameraCalibration.py

@@ -0,0 +1,127 @@
+import sys
+from meshroom.core import desc
+
+
+class CameraCalibration(desc.CommandLineNode):
+    internalFolder = '{cache}/{nodeType}/{uid0}/'
+    commandLine = 'aliceVision_cameraCalibration {allParams}'
+
+    inputs = [
+        desc.File(
+            name='input',
+            label='Input',
+            description='''Input images in one of the following form:
+ - folder containing images
+ - image sequence like "/path/to/seq.@.jpg"
+ - video file''',
+            value='',
+            uid=[0],
+            ),
+        desc.ChoiceParam(
+            name='pattern',
+            label='Pattern',
+            description='''Type of pattern (CHESSBOARD, CIRCLES, ASYMMETRIC_CIRCLES, ASYMMETRIC_CCTAG).''',
+            value='CHESSBOARD',
+            values=['CHESSBOARD', 'CIRCLES', 'ASYMMETRIC_CIRCLES', 'ASYMMETRIC_CCTAG'],
+            exclusive=True,
+            uid=[0],
+            ),
+        desc.GroupAttribute(name="size", label="Size", description="Number of inner corners per one of board dimension like W H.", groupDesc=[
+            desc.IntParam(
+                name='width',
+                label='Width',
+                description='',
+                value=7,
+                range=(0, 10000, 1),
+                uid=[0],
+                ),
+            desc.IntParam(
+                name='height',
+                label='Height',
+                description='',
+                value=5,
+                range=(0, 10000, 1),
+                uid=[0],
+                ),
+            ]),
+        desc.FloatParam(
+            name='squareSize',
+            label='Square Size',
+            description='''Size of the grid's square cells (mm).''',
+            value=1.0,
+            range=(0.0, 100.0, 1.0),
+            uid=[0],
+            ),
+        desc.IntParam(
+            name='nbDistortionCoef',
+            label='Nb Distortion Coef',
+            description='''Number of distortion coefficient.''',
+            value=3,
+            range=(0, 5, 1),
+            uid=[0],
+            ),
+        desc.IntParam(
+            name='maxFrames',
+            label='Max Frames',
+            description='''Maximal number of frames to extract from the video file.''',
+            value=0,
+            range=(0, 5, 1),
+            uid=[0],
+            ),
+        desc.IntParam(
+            name='maxCalibFrames',
+            label='Max Calib Frames',
+            description='''Maximal number of frames to use to calibrate from the selected frames.''',
+            value=100,
+            range=(0, 1000, 1),
+            uid=[0],
+            ),
+        desc.IntParam(
+            name='calibGridSize',
+            label='Calib Grid Size',
+            description='''Define the number of cells per edge.''',
+            value=10,
+            range=(0, 50, 1),
+            uid=[0],
+            ),
+        desc.IntParam(
+            name='minInputFrames',
+            label='Min Input Frames',
+            description='''Minimal number of frames to limit the refinement loop.''',
+            value=10,
+            range=(0, 100, 1),
+            uid=[0],
+            ),
+        desc.FloatParam(
+            name='maxTotalAvgErr',
+            label='Max Total Avg Err',
+            description='''Max Total Average Error.''',
+            value=0.10000000000000001,
+            range=(0.0, 1.0, 0.01),
+            uid=[0],
+            ),
+        desc.File(
+            name='debugRejectedImgFolder',
+            label='Debug Rejected Img Folder',
+            description='''Folder to export delete images during the refinement loop.''',
+            value='',
+            uid=[0],
+            ),
+        desc.File(
+            name='debugSelectedImgFolder',
+            label='Debug Selected Img Folder',
+            description='''Folder to export debug images.''',
+            value='',
+            uid=[0],
+            ),
+    ]
+
+    outputs = [
+        desc.File(
+            name='output',
+            label='Output',
+            description='''Output filename for intrinsic [and extrinsic] parameters.''',
+            value='{cache}/{nodeType}/{uid0}/cameraCalibration.cal',
+            uid=[],
+            ),
+    ]