Spaces:
Running
Running
File size: 1,514 Bytes
a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import argparse
import os, sys
from SensorData import SensorData
# params
parser = argparse.ArgumentParser()
# data paths
parser.add_argument("--filename", required=True, help="path to sens file to read")
parser.add_argument("--output_path", required=True, help="path to output folder")
parser.add_argument(
"--export_depth_images", dest="export_depth_images", action="store_true"
)
parser.add_argument(
"--export_color_images", dest="export_color_images", action="store_true"
)
parser.add_argument("--export_poses", dest="export_poses", action="store_true")
parser.add_argument(
"--export_intrinsics", dest="export_intrinsics", action="store_true"
)
parser.set_defaults(
export_depth_images=False,
export_color_images=False,
export_poses=False,
export_intrinsics=False,
)
opt = parser.parse_args()
print(opt)
def main():
if not os.path.exists(opt.output_path):
os.makedirs(opt.output_path)
# load the data
sys.stdout.write("loading %s..." % opt.filename)
sd = SensorData(opt.filename)
sys.stdout.write("loaded!\n")
if opt.export_depth_images:
sd.export_depth_images(os.path.join(opt.output_path, "depth"))
if opt.export_color_images:
sd.export_color_images(os.path.join(opt.output_path, "color"))
if opt.export_poses:
sd.export_poses(os.path.join(opt.output_path, "pose"))
if opt.export_intrinsics:
sd.export_intrinsics(os.path.join(opt.output_path, "intrinsic"))
if __name__ == "__main__":
main()
|