Shortcuts

mmcv.arraymisc.quantize

mmcv.arraymisc.quantize(arr: ~numpy.ndarray, min_val: int | float, max_val: 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