10.6.7. visualize

Visualize code snippets for various task.

10.6.7.1. visualize

argoverse.ArgoverseViz

The visiualize method of Argoverse result.

cam3d.Cam3dViz

cls.ClsViz

The viz method of classification result.

utils.ANCBEVImgStitcher

Restore multi view pictures for bev tasks.

det.DetViz

The visiualize method of det result.

disparity.DispViz

The visualize method of disparity or depth pred result.

keypoints.KeypointsViz

The viz method of Keypoints predict result.

lane_lines.LanelineViz

The visualize method of Laneline detection result.

nuscenes.NuscenesViz

The visiualize method of Nuscenes result.

opticalflow.FlowViz

The visualize method of Flow_pred result.

seg.SegViz

The visualize method of segmentation result.

track.TrackViz

The visiualize method of object track result.

10.6.7.2. API Reference

class hat.visualize.argoverse.ArgoverseViz(traj_scale=50, is_plot=True)

The visiualize method of Argoverse result.

参数

  • traj_scale – Scale for traj input feature.

  • is_plot – Whether to plot image.

class hat.visualize.cls.ClsViz(is_plot: bool = False)

The viz method of classification result.

参数

is_plot (bool) – Whether to plot image.

class hat.visualize.utils.ANCBEVImgStitcher(camera_view_names: list, camera_layouts: Optional[list] = None, per_extra_img_size: tuple = (960, 512), is_bev_horizon: bool = False, interpolation: int = 1, **kwargs)

Restore multi view pictures for bev tasks.

参数

  • camera_view_names – the camera view names of the model.

  • camera_layouts – the customized layout of cameras for visualization. Support to visualize part of camera views. “tmp_img” denotes placeholder img with all zeros. Further, we support view expansion, if the name of the view in a line or a square area is the same.

  • per_extra_img_size – Multiview img sizes to visualize,(w, h).

  • is_bev_horizon – indicates whether the BEV image occupies the bottom row or the rightmost column.

  • interpolation – the interpolation mode of resizing bev_img.

class hat.visualize.det.DetViz(score_thresh: float = 0.5, absolute_coord: bool = True, is_plot: bool = True)

The visiualize method of det result.

参数

  • score_thresh (float) – Score thresh for filtering box in plot.

  • absolute_coord (bool) – Coordinates of pred_box is absolute or not.

  • is_plot (bool) – Whether to plot image.

class hat.visualize.disparity.DispViz(is_plot: bool = True)

The visualize method of disparity or depth pred result.

参数

is_plot – Whether to plot image.

class hat.visualize.keypoints.KeypointsViz(threshold: float = 0.0, is_plot: bool = False)

The viz method of Keypoints predict result.

参数

  • threshold – threshold of ploting keypoint prediction.

  • is_plot – Whether to plot image.

class hat.visualize.lane_lines.LanelineViz(is_plot: bool = True, thickness: int = 10, color: Tuple[int, int, int] = (255, 0, 0))

The visualize method of Laneline detection result.

参数

  • is_plot – Whether to plot image.

  • thickness – The thickness of lane line.

  • color – The color of lane line.

class hat.visualize.nuscenes.NuscenesViz(bev_size: Optional[Tuple[float, float, float]] = None, score_thresh: float = 0.4, is_plot: bool = True, use_bce: bool = False)

The visiualize method of Nuscenes result.

参数

  • bev_size – Size of Bev size.

  • score_thresh – Score thresh for filtering box in plot.

  • is_plot – Whether to plot image.

  • use_bce – Whether use bce as segmentation result.

class hat.visualize.opticalflow.FlowViz(is_plot: bool = True)

The visualize method of Flow_pred result.

参数

is_plot – Whether to plot image.

class hat.visualize.seg.SegViz(is_plot: bool, image_process: Callable = <function _default_image_process>, label_process: Callable = <function _seg_target_process>)

The visualize method of segmentation result.

参数

  • image_process (Callable) – Process of image.

  • label_process (Callable) – Process of label.

class hat.visualize.track.TrackViz(is_plot: bool = True)

The visiualize method of object track result.

参数

is_plot (bool) – Whether to plot image.