File size: 1,302 Bytes
88f8b7c |
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 |
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()
|