diff options
Diffstat (limited to 'src/text_recognizer/networks/__init__.py')
| -rw-r--r-- | src/text_recognizer/networks/__init__.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/text_recognizer/networks/__init__.py b/src/text_recognizer/networks/__init__.py index f958672..2b624bb 100644 --- a/src/text_recognizer/networks/__init__.py +++ b/src/text_recognizer/networks/__init__.py @@ -3,19 +3,18 @@ from .cnn_transformer import CNNTransformer from .crnn import ConvolutionalRecurrentNetwork from .ctc import greedy_decoder from .densenet import DenseNet -from .fcn import FCN from .lenet import LeNet -from .metrics import accuracy, accuracy_ignore_pad, cer, wer +from .metrics import accuracy, cer, wer from .mlp import MLP from .residual_network import ResidualNetwork, ResidualNetworkEncoder from .transformer import Transformer from .unet import UNet from .util import sliding_window +from .vit import ViT from .wide_resnet import WideResidualNetwork __all__ = [ "accuracy", - "accuracy_ignore_pad", "cer", "CNNTransformer", "ConvolutionalRecurrentNetwork", @@ -29,6 +28,7 @@ __all__ = [ "sliding_window", "UNet", "Transformer", + "ViT", "wer", "WideResidualNetwork", ] |