Shortcuts

mmcv.arraymisc.quantize

mmcv.arraymisc.quantize(arr: numpy.ndarray, min_val: Union[int, float], max_val: Union[int, float], levels: int, dtype=<class 'numpy.int64'>)tuple[source]

Quantize an array of (-inf, inf) to [0, levels-1].

Parameters
  • arr (ndarray) – Input array.

  • min_val (int or float) – Minimum value to be clipped.

  • max_val (int or float) – Maximum value to be clipped.

  • levels (int) – Quantization levels.

  • dtype (np.type) – The type of the quantized array.

Returns

Quantized array.

Return type

tuple