mmocr.utils.rescale_polygon

mmocr.utils.rescale_polygon(polygon, scale_factor, mode='mul')

Rescale a polygon according to scale_factor.

The behavior is different depending on the mode. When mode is ‘mul’, the coordinates will be multiplied by scale_factor, which is usually used in preprocessing transforms such as Resize(). The coordinates will be divided by scale_factor if mode is ‘div’. It can be used in postprocessors to recover the polygon in the original image size.

Parameters

• polygon (ArrayLike) – A polygon. In any form can be converted to an 1-D numpy array. E.g. list[float], np.ndarray, or torch.Tensor. Polygon is written in [x1, y1, x2, y2, …].

• scale_factor (tuple(int, int)) – (w_scale, h_scale).

• model (str) – Rescale mode. Can be ‘mul’ or ‘div’. Defaults to ‘mul’.

• mode (str) –

Returns

Rescaled polygon.

Return type

np.ndarray