__init__.py 1.82 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
from .det_pse_loss import PSELoss
WenmuZhou's avatar
WenmuZhou committed
24

littletomatodonkey's avatar
littletomatodonkey committed
25
26
27
28
# rec loss
from .rec_ctc_loss import CTCLoss
from .rec_att_loss import AttentionLoss
from .rec_srn_loss import SRNLoss
Topdu's avatar
Topdu committed
29
from .rec_nrtr_loss import NRTRLoss
andyjpaddle's avatar
andyjpaddle committed
30
from .rec_sar_loss import SARLoss
tink2123's avatar
tink2123 committed
31
32
from .rec_aster_loss import AsterLoss

littletomatodonkey's avatar
littletomatodonkey committed
33
34
35
36
37
# cls loss
from .cls_loss import ClsLoss

# e2e loss
from .e2e_pg_loss import PGLoss
LDOUBLEV's avatar
LDOUBLEV committed
38
from .kie_sdmgr_loss import SDMGRLoss
WenmuZhou's avatar
WenmuZhou committed
39

littletomatodonkey's avatar
littletomatodonkey committed
40
41
# basic loss function
from .basic_loss import DistanceLoss
WenmuZhou's avatar
WenmuZhou committed
42

littletomatodonkey's avatar
littletomatodonkey committed
43
44
# combined loss function
from .combined_loss import CombinedLoss
zhoujun's avatar
zhoujun committed
45

MissPenguin's avatar
MissPenguin committed
46
47
# table loss
from .table_att_loss import TableAttentionLoss
LDOUBLEV's avatar
add kie  
LDOUBLEV committed
48
49


littletomatodonkey's avatar
littletomatodonkey committed
50
def build_loss(config):
LDOUBLEV's avatar
LDOUBLEV committed
51
    support_dict = [
WenmuZhou's avatar
WenmuZhou committed
52
53
        'DBLoss', 'PSELoss', 'EASTLoss', 'SASTLoss', 'CTCLoss', 'ClsLoss',
        'AttentionLoss', 'SRNLoss', 'PGLoss', 'CombinedLoss', 'NRTRLoss',
LDOUBLEV's avatar
LDOUBLEV committed
54
        'TableAttentionLoss', 'SARLoss', 'AsterLoss', 'SDMGRLoss'
LDOUBLEV's avatar
add kie  
LDOUBLEV committed
55
    ]
WenmuZhou's avatar
WenmuZhou committed
56
57
58
59
60
61
    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