mmcv.visualization.flow2rgb¶

mmcv.visualization.flow2rgb(flow: ndarray, color_wheel: ndarray | None = None, unknown_thr: float = 1000000.0) → ndarray[source]¶

Convert flow map to RGB image.

Parameters:

flow (ndarray) – Array of optical flow.
color_wheel (ndarray or None) – Color wheel used to map flow field to RGB colorspace. Default color wheel will be used if not specified.
unknown_thr (float) – Values above this threshold will be marked as unknown and thus ignored.

Returns:

RGB image that can be visualized.

Return type:

ndarray