PyramidRescale

class mmocr.datasets.transforms.PyramidRescale(factor=4, base_shape=(128, 512), randomize_factor=True)

Resize the image to the base shape, downsample it with gaussian pyramid, and rescale it back to original size.

Adapted from https://github.com/FangShancheng/ABINet.

Required Keys:

• img (ndarray)

Modified Keys:

• img (ndarray)

Parameters

• factor (int) – The decay factor from base size, or the number of downsampling operations from the base layer.

• base_shape (tuple[int, int]) – The shape (width, height) of the base layer of the pyramid.

• randomize_factor (bool) – If True, the final factor would be a random integer in [0, factor].

Return type

None

transform(results)

Applying pyramid rescale on results.

Parameters

results (dict) – Result dict containing the data to transform.

Returns

The transformed data.

Return type

Dict