| 16 | |
| 17 | |
| 18 | def build_backbone(config, model_type): |
| 19 | if model_type == "det" or model_type == "table": |
| 20 | from .det_mobilenet_v3 import MobileNetV3 |
| 21 | from .det_resnet import ResNet |
| 22 | from .det_resnet_vd import ResNet_vd |
| 23 | from .det_resnet_vd_sast import ResNet_SAST |
| 24 | from .det_pp_lcnet import PPLCNet |
| 25 | from .rec_lcnetv3 import PPLCNetV3 |
| 26 | from .rec_lcnetv4 import PPLCNetV4 |
| 27 | from .rec_hgnet import PPHGNet_small |
| 28 | from .rec_vit import ViT |
| 29 | from .det_pp_lcnet_v2 import PPLCNetV2_base |
| 30 | from .rec_repvit import RepSVTR_det |
| 31 | from .rec_vary_vit import Vary_VIT_B |
| 32 | from .rec_pphgnetv2 import PPHGNetV2_B4 |
| 33 | |
| 34 | support_dict = [ |
| 35 | "MobileNetV3", |
| 36 | "ResNet", |
| 37 | "ResNet_vd", |
| 38 | "ResNet_SAST", |
| 39 | "PPLCNet", |
| 40 | "PPLCNetV3", |
| 41 | "PPLCNetV4", |
| 42 | "PPHGNet_small", |
| 43 | "PPLCNetV2_base", |
| 44 | "RepSVTR_det", |
| 45 | "Vary_VIT_B", |
| 46 | "PPHGNetV2_B4", |
| 47 | ] |
| 48 | if model_type == "table": |
| 49 | from .table_master_resnet import TableResNetExtra |
| 50 | |
| 51 | support_dict.append("TableResNetExtra") |
| 52 | elif model_type == "rec" or model_type == "cls": |
| 53 | from .rec_mobilenet_v3 import MobileNetV3 |
| 54 | from .rec_resnet_vd import ResNet |
| 55 | from .rec_resnet_fpn import ResNetFPN |
| 56 | from .rec_mv1_enhance import MobileNetV1Enhance |
| 57 | from .rec_nrtr_mtb import MTB |
| 58 | from .rec_resnet_31 import ResNet31 |
| 59 | from .rec_resnet_32 import ResNet32 |
| 60 | from .rec_resnet_45 import ResNet45 |
| 61 | from .rec_resnet_aster import ResNet_ASTER |
| 62 | from .rec_micronet import MicroNet |
| 63 | from .rec_efficientb3_pren import EfficientNetb3_PREN |
| 64 | from .rec_svtrnet import SVTRNet |
| 65 | from .rec_vitstr import ViTSTR |
| 66 | from .rec_resnet_rfl import ResNetRFL |
| 67 | from .rec_densenet import DenseNet |
| 68 | from .rec_resnetv2 import ResNetV2 |
| 69 | from .rec_hybridvit import HybridTransformer |
| 70 | from .rec_donut_swin import DonutSwinModel |
| 71 | from .rec_shallow_cnn import ShallowCNN |
| 72 | from .rec_lcnetv3 import PPLCNetV3 |
| 73 | from .rec_lcnetv4 import PPLCNetV4 |
| 74 | from .rec_hgnet import PPHGNet_small |
| 75 | from .rec_vit_parseq import ViTParseQ |