12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

parse command line arguments

release/4.3a0
Varun Agrawal 2025-02-09 11:53:12 -05:00
parent 6e587cfcb6
commit e67e4cd41d
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
examples/plot_city10000.py
View File

@ -8,15 +8,28 @@ See LICENSE for the license information
Script to plot City10000 results. Script to plot City10000 results.
Can be used to plot results from both C++ and python scripts. Can be used to plot results from both C++ and python scripts.
python plot_city10000.py python plot_city10000.py Data/ISAM2_GT_city10000.txt ../build/examples/ISAM2_city10000.txt ../build/examples/Hybrid_City10000.txt
Author: Varun Agrawal Author: Varun Agrawal
""" """
import argparse
import numpy as np import numpy as np
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
def parse_args():
"""Parse command line arguments"""
parser = argparse.ArgumentParser()
parser.add_argument("ground_truth", help="The ground truth data file.")
parser.add_argument(
"estimates",
nargs='+',
help="File(s) with estimates (as .txt), can be more than one.")
return parser.parse_args()
def plot_estimates(gt, def plot_estimates(gt,
estimates, estimates,
fignum: int, fignum: int,
@ -42,15 +55,13 @@ def plot_estimates(gt,
def main(): def main():
gt = np.loadtxt('Data/ISAM2_GT_city10000.txt', delimiter=" ") gt = np.loadtxt('', delimiter=" ")
# Generate by running `make ISAM2_City10000.run` # Generate by running `make ISAM2_City10000.run`
eh_poses = np.loadtxt('../build/examples/ISAM2_city10000.txt', eh_poses = np.loadtxt('', delimiter=" ")
delimiter=" ")
# Generate by running `make Hybrid_City10000.run` # Generate by running `make Hybrid_City10000.run`
h_poses = np.loadtxt('../build/examples/Hybrid_City10000.txt', h_poses = np.loadtxt('', delimiter=" ")
delimiter=" ")
# Plot the same number of GT poses as estimated ones # Plot the same number of GT poses as estimated ones
gt = gt[:h_poses.shape[0], :] gt = gt[:h_poses.shape[0], :]