Shortcuts

mmcv.visualization.flow2rgb

mmcv.visualization.flow2rgb(flow: numpy.ndarray, color_wheel: Optional[numpy.ndarray] = None, unknown_thr: float = 1000000.0)numpy.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