| import imageio | |
| import numpy as np | |
| from oee.utils import plt_utils | |
| from oee.utils.elev_est_api import elev_est_api | |
| import argparse | |
| def visualize(img_paths, elev): | |
| imgs = [imageio.imread_v2(img_path) for img_path in img_paths] | |
| plt_utils.image_grid(imgs, 2, 2, label=f"elev={elev}") | |
| def main(): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--img_paths", type=str, nargs=4, help="image paths", | |
| default=["assets/example_data/0_0.png", | |
| "assets/example_data/0_1.png", | |
| "assets/example_data/0_2.png", | |
| "assets/example_data/0_3.png"]) | |
| parser.add_argument("--min_elev", type=float, default=30, help="min elevation") | |
| parser.add_argument("--max_elev", type=float, default=150, help="max elevation") | |
| parser.add_argument("--dbg", default=False, action="store_true", help="debug mode") | |
| parser.add_argument("--K_path", type=str, default=None, help="path to K") | |
| args = parser.parse_args() | |
| if args.K_path is not None: | |
| K = np.loadtxt(args.K_path) | |
| else: | |
| K = None | |
| elev = elev_est_api(args.img_paths, args.min_elev, args.max_elev, K, args.dbg) | |
| visualize(args.img_paths, elev) | |
| if __name__ == '__main__': | |
| main() | |