third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix markers parse in gen_pattern.py

Browse Source
This commit is contained in:
AleksandrPanov 2021-11-10 11:14:06 +03:00
parent 58043dac0f
commit 0c0a8392e5
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
doc/pattern_tools/gen_pattern.py
View File

@ -171,7 +171,7 @@ def main():
parser.add_argument("-m", "--markers", help="list of cells with markers for the radon checkerboard. Marker " parser.add_argument("-m", "--markers", help="list of cells with markers for the radon checkerboard. Marker "
"coordinates as list of numbers: -m 1 2 3 4 means markers in cells " "coordinates as list of numbers: -m 1 2 3 4 means markers in cells "
"[1, 2] and [3, 4]", "[1, 2] and [3, 4]",
action="store", dest="markers", nargs="+", type=int) default=argparse.SUPPRESS, action="store", dest="markers", nargs="+", type=int)
args = parser.parse_args() args = parser.parse_args()
show_help = args.show_help show_help = args.show_help
@ -195,14 +195,16 @@ def main():
"A5": [148, 210]} "A5": [148, 210]}
page_width = page_sizes[page_size][0] page_width = page_sizes[page_size][0]
page_height = page_sizes[page_size][1] page_height = page_sizes[page_size][1]
if len(args.markers) % 2 == 1: markers = None
raise ValueError("The length of the markers array={} must be even".format(len(args.markers))) if p_type == "radon_checkerboard" and "markers" in args:
markers = set() if len(args.markers) % 2 == 1:
for x, y in zip(args.markers[::2], args.markers[1::2]): raise ValueError("The length of the markers array={} must be even".format(len(args.markers)))
if x in range(0, columns) and y in range(0, rows): markers = set()
markers.add((x, y)) for x, y in zip(args.markers[::2], args.markers[1::2]):
else: if x in range(0, columns) and y in range(0, rows):
raise ValueError("The marker {},{} is outside the checkerboard".format(x, y)) markers.add((x, y))
else:
raise ValueError("The marker {},{} is outside the checkerboard".format(x, y))
pm = PatternMaker(columns, rows, output, units, square_size, radius_rate, page_width, page_height, markers) pm = PatternMaker(columns, rows, output, units, square_size, radius_rate, page_width, page_height, markers)
# dict for easy lookup of pattern type # dict for easy lookup of pattern type