__init__.py 1.63 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
WenmuZhou's avatar
WenmuZhou committed
14
15

import copy
littletomatodonkey's avatar
littletomatodonkey committed
16
17
import paddle
import paddle.nn as nn
WenmuZhou's avatar
WenmuZhou committed
18

littletomatodonkey's avatar
littletomatodonkey committed
19
20
21
22
# det loss
from .det_db_loss import DBLoss
from .det_east_loss import EASTLoss
from .det_sast_loss import SASTLoss
WenmuZhou's avatar
WenmuZhou committed
23

littletomatodonkey's avatar
littletomatodonkey committed
24
25
26
27
28
29
30
31
32
33
# rec loss
from .rec_ctc_loss import CTCLoss
from .rec_att_loss import AttentionLoss
from .rec_srn_loss import SRNLoss

# cls loss
from .cls_loss import ClsLoss

# e2e loss
from .e2e_pg_loss import PGLoss
WenmuZhou's avatar
WenmuZhou committed
34

littletomatodonkey's avatar
littletomatodonkey committed
35
36
# basic loss function
from .basic_loss import DistanceLoss
WenmuZhou's avatar
WenmuZhou committed
37

littletomatodonkey's avatar
littletomatodonkey committed
38
39
# combined loss function
from .combined_loss import CombinedLoss
zhoujun's avatar
zhoujun committed
40

MissPenguin's avatar
MissPenguin committed
41
42
# table loss
from .table_att_loss import TableAttentionLoss
littletomatodonkey's avatar
littletomatodonkey committed
43

tink2123's avatar
tink2123 committed
44
45
46
from .rec_aster_loss import AsterLoss


littletomatodonkey's avatar
littletomatodonkey committed
47
def build_loss(config):
LDOUBLEV's avatar
LDOUBLEV committed
48
    support_dict = [
LDOUBLEV's avatar
LDOUBLEV committed
49
        'DBLoss', 'EASTLoss', 'SASTLoss', 'CTCLoss', 'ClsLoss', 'AttentionLoss',
tink2123's avatar
tink2123 committed
50
        'SRNLoss', 'PGLoss', 'CombinedLoss', 'TableAttentionLoss', 'AsterLoss'
littletomatodonkey's avatar
littletomatodonkey committed
51
    ]
WenmuZhou's avatar
WenmuZhou committed
52
53
54
55
56
57
    config = copy.deepcopy(config)
    module_name = config.pop('name')
    assert module_name in support_dict, Exception('loss only support {}'.format(
        support_dict))
    module_class = eval(module_name)(**config)
    return module_class