Tong Gao 2d743cfa19
[Model] SPTS (#1696)
* [Feature] Add RepeatAugSampler

* initial commit

* spts inference done

* merge repeat_aug (bug in multi-node?)

* fix inference

* train done

* rm readme

* Revert "merge repeat_aug (bug in multi-node?)"

This reverts commit 393506a97cbe6d75ad1f28611ea10eba6b8fa4b3.

* Revert "[Feature] Add RepeatAugSampler"

This reverts commit 2089b02b4844157670033766f257b5d1bca452ce.

* remove utils

* readme & conversion script

* update readme

* fix

* optimize

* rename cfg & del compose

* fix

* fix
2023-02-01 11:58:03 +08:00

11 lines
446 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .adel_dataset import AdelDataset
from .transforms.spts_transforms import (Bezier2Polygon, ConvertText,
LoadOCRAnnotationsWithBezier,
Polygon2Bezier, RescaleToShortSide)
__all__ = [
'AdelDataset', 'LoadOCRAnnotationsWithBezier', 'Bezier2Polygon',
'Polygon2Bezier', 'ConvertText', 'RescaleToShortSide'
]