aboutsummaryrefslogtreecommitdiff
path: root/src/argaze/utils/export_aruco_markers.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/argaze/utils/export_aruco_markers.py')
-rw-r--r--src/argaze/utils/export_aruco_markers.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/src/argaze/utils/export_aruco_markers.py b/src/argaze/utils/export_aruco_markers.py
new file mode 100644
index 0000000..e76b95e
--- /dev/null
+++ b/src/argaze/utils/export_aruco_markers.py
@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+
+import argparse
+import os
+
+from argaze.ArUcoMarkers import ArUcoMarkers
+
+def main():
+ """Generates ArUco markers to place into a scene."""
+
+ # manage arguments
+ parser = argparse.ArgumentParser(description=main.__doc__)
+ parser.add_argument('-o', '--output', metavar='OUT', type=str, default='.', help='destination path')
+ parser.add_argument('-d', '--dictionary', metavar='DICT', type=str, default='DICT_4X4_50', help='aruco marker dictionnary')
+ parser.add_argument('-r', '--resolution', metavar='RES', type=int, default=300, help='picture resolution in dpi')
+ args = parser.parse_args()
+
+ # manage destination folder
+ if not os.path.exists(args.output):
+ os.makedirs(args.output)
+ print(f'{args.output} folder created')
+
+ # create aruco markers
+ aruco_markers = ArUcoMarkers.ArUcoMarkers(args.dictionary)
+
+ # export markers
+ aruco_markers.export_all(args.output, args.resolution)
+
+if __name__ == '__main__':
+
+ main() \ No newline at end of file