MMOCR2MMDet

class mmocr.datasets.transforms.MMOCR2MMDet(poly2mask=False)

Convert transforms’s data format from MMOCR to MMDet.

Required Keys:

• img_shape

• gt_polygons (List[ndarray]) (optional)

• gt_ignored (np.bool) (optional)

Added Keys:

• gt_masks (PolygonMasks | BitmapMasks) (optional)

• gt_ignore_flags (np.bool) (optional)

Parameters

poly2mask (bool) – Whether to convert mask to bitmap. Default: True.

Return type

None

transform(results)

Convert MMOCR’s data format to MMDet’s data format.

Parameters

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

Returns

The transformed data.

Return type

(Dict)