visualize_ground_truth.py 961 Bytes
Newer Older
Clément Pinard's avatar
Clément Pinard committed
1
2
3
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from path import Path

Clément Pinard's avatar
Clément Pinard committed
4
parser = ArgumentParser(description='Create visualization for specified video',
Clément Pinard's avatar
Clément Pinard committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
                        formatter_class=ArgumentDefaultsHelpFormatter)

parser.add_argument('--gt', metavar='PATH',
                    help='path to folder with raw groundtruth', type=Path)
parser.add_argument('--images', metavar='PATH',
                    help='path to folder with images', type=Path)
parser.add_argument('--range', metavar='R', default=10, type=int,
                    help='')
parser.add_argument('--output', metavar='DIR', default=None, type=Path)


def main():
    args = parser.parse_args()
    depth_paths = sorted(args.gt.files())
    image_paths = sorted(args.images.files())

    if len(depth_paths) != len(image_paths):
        print("{} depth groundtruth maps for {} images".format(len(depth_paths), len(image_paths)))


if __name__ == '__main__':
    main()