mmcv.visualization.flow2rgb

mmcv.visualization.flow2rgb(flow: numpy.ndarray, color_wheel: Optional[numpy.ndarray] = None, unknown_thr: float = 1000000.0) → numpy.ndarray

Convert flow map to RGB image.

参数

• 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.

返回

RGB image that can be visualized.

返回类型

ndarray