__init__.py 2.09 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

19
20
21
# basic_loss
from .basic_loss import LossFromOutput

littletomatodonkey's avatar
littletomatodonkey committed
22
23
24
25
# 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
26
from .det_pse_loss import PSELoss
zhiminzhang0830's avatar
zhiminzhang0830 committed
27
from .det_fce_loss import FCELoss
WenmuZhou's avatar
WenmuZhou committed
28

littletomatodonkey's avatar
littletomatodonkey committed
29
30
31
32
# 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
33
from .rec_nrtr_loss import NRTRLoss
andyjpaddle's avatar
andyjpaddle committed
34
from .rec_sar_loss import SARLoss
tink2123's avatar
tink2123 committed
35
from .rec_aster_loss import AsterLoss
36
from .rec_pren_loss import PRENLoss
tink2123's avatar
tink2123 committed
37

littletomatodonkey's avatar
littletomatodonkey committed
38
39
40
41
42
# cls loss
from .cls_loss import ClsLoss

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

littletomatodonkey's avatar
littletomatodonkey committed
45
46
# basic loss function
from .basic_loss import DistanceLoss
WenmuZhou's avatar
WenmuZhou committed
47

littletomatodonkey's avatar
littletomatodonkey committed
48
49
# combined loss function
from .combined_loss import CombinedLoss
zhoujun's avatar
zhoujun committed
50

MissPenguin's avatar
MissPenguin committed
51
52
# table loss
from .table_att_loss import TableAttentionLoss
LDOUBLEV's avatar
add kie  
LDOUBLEV committed
53

54
55
56
# vqa token loss
from .vqa_token_layoutlm_loss import VQASerTokenLayoutLMLoss

LDOUBLEV's avatar
add kie  
LDOUBLEV committed
57

littletomatodonkey's avatar
littletomatodonkey committed
58
def build_loss(config):
LDOUBLEV's avatar
LDOUBLEV committed
59
    support_dict = [
zhiminzhang0830's avatar
zhiminzhang0830 committed
60
61
62
        'DBLoss', 'PSELoss', 'EASTLoss', 'SASTLoss', 'FCELoss', 'CTCLoss',
        'ClsLoss', 'AttentionLoss', 'SRNLoss', 'PGLoss', 'CombinedLoss',
        'NRTRLoss', 'TableAttentionLoss', 'SARLoss', 'AsterLoss', 'SDMGRLoss',
63
        'VQASerTokenLayoutLMLoss', 'LossFromOutput', 'PRENLoss'
LDOUBLEV's avatar
add kie  
LDOUBLEV committed
64
    ]
WenmuZhou's avatar
WenmuZhou committed
65
66
67
68
69
70
    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