From cf69b5a73be37d57c89c537e7ad65f1cb8842482 Mon Sep 17 00:00:00 2001 From: yutaka Date: Wed, 17 Nov 2021 03:08:29 +0900 Subject: [PATCH 1/2] update --- argoverse/visualization/visualize_sequences.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/argoverse/visualization/visualize_sequences.py b/argoverse/visualization/visualize_sequences.py index 1acbca28..ef825c68 100644 --- a/argoverse/visualization/visualize_sequences.py +++ b/argoverse/visualization/visualize_sequences.py @@ -1,8 +1,9 @@ #!/usr/bin/env python """A simple python script template.""" +import os from collections import defaultdict -from typing import Dict, List, Optional +from typing import Dict, List, Optional, Union import matplotlib.lines as mlines import matplotlib.pyplot as plt @@ -14,6 +15,7 @@ _ZORDER = {"AGENT": 15, "AV": 10, "OTHERS": 5} +_PathLike = Union[str, "os.PathLike[str]", None] def interpolate_polyline(polyline: np.ndarray, num_points: int) -> np.ndarray: duplicates = [] @@ -31,6 +33,7 @@ def interpolate_polyline(polyline: np.ndarray, num_points: int) -> np.ndarray: def viz_sequence( df: pd.DataFrame, + map_root: _PathLike = None, lane_centerlines: Optional[List[np.ndarray]] = None, show: bool = True, smoothen: bool = False, @@ -41,7 +44,10 @@ def viz_sequence( if lane_centerlines is None: # Get API for Argo Dataset map - avm = ArgoverseMap() + if map_root is not None: + avm = ArgoverseMap(map_root) + else: + avm = ArgoverseMap() seq_lane_props = avm.city_lane_centerlines_dict[city_name] plt.figure(0, figsize=(8, 7)) From f3328b33ee2e756e8b14aaeb4f33cfba66986c65 Mon Sep 17 00:00:00 2001 From: yutaka Date: Fri, 19 Nov 2021 01:11:14 +0900 Subject: [PATCH 2/2] modify pathlike expression --- argoverse/visualization/visualize_sequences.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/argoverse/visualization/visualize_sequences.py b/argoverse/visualization/visualize_sequences.py index ef825c68..4233a28b 100644 --- a/argoverse/visualization/visualize_sequences.py +++ b/argoverse/visualization/visualize_sequences.py @@ -15,7 +15,7 @@ _ZORDER = {"AGENT": 15, "AV": 10, "OTHERS": 5} -_PathLike = Union[str, "os.PathLike[str]", None] +_PathLike = Union[str, "os.PathLike[str]"] def interpolate_polyline(polyline: np.ndarray, num_points: int) -> np.ndarray: duplicates = []