mmocr/projects/SPTS/spts/model/__init__.py
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

13 lines
403 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .spts import SPTS
from .spts_decoder import SPTSDecoder
from .spts_dictionary import SPTSDictionary
from .spts_encoder import SPTSEncoder
from .spts_module_loss import SPTSModuleLoss
from .spts_postprocessor import SPTSPostprocessor
__all__ = [
'SPTSEncoder', 'SPTSDecoder', 'SPTSPostprocessor', 'SPTS',
'SPTSDictionary', 'SPTSModuleLoss'
]