paddlex 2.0.0rc4__py3-none-any.whl → 3.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- paddlex/.version +1 -0
- paddlex/__init__.py +35 -18
- paddlex/__main__.py +39 -0
- paddlex/configs/modules/3d_bev_detection/BEVFusion.yaml +38 -0
- paddlex/configs/modules/chart_parsing/PP-Chart2Table.yaml +13 -0
- paddlex/configs/modules/doc_text_orientation/PP-LCNet_x1_0_doc_ori.yaml +41 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-2B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-7B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee2-3B.yaml +14 -0
- paddlex/configs/modules/face_detection/BlazeFace-FPN-SSH.yaml +40 -0
- paddlex/configs/modules/face_detection/BlazeFace.yaml +40 -0
- paddlex/configs/modules/face_detection/PP-YOLOE_plus-S_face.yaml +40 -0
- paddlex/configs/modules/face_detection/PicoDet_LCNet_x2_5_face.yaml +40 -0
- paddlex/configs/modules/face_feature/MobileFaceNet.yaml +41 -0
- paddlex/configs/modules/face_feature/ResNet50_face.yaml +41 -0
- paddlex/configs/modules/formula_recognition/LaTeX_OCR_rec.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-M.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/UniMERNet.yaml +40 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-L_human.yaml +42 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-S_human.yaml +42 -0
- paddlex/configs/modules/image_anomaly_detection/STFPM.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_base_patch16_224.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_large_patch14_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_small.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/FasterNet-L.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-M.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-S.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T0.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T1.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T2.yaml +40 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_small.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B1.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B2.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B3.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B4.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B6.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_large.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_5.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet200_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S1.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S2.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S3.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S4.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_small_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_tiny_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_base.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_large.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/CLIP_vit_base_patch16_448_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B4_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B6_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-LCNet_x1_0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/ResNet50_ML.yaml +41 -0
- paddlex/configs/modules/image_unwarping/UVDoc.yaml +12 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-M.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNeXt101-vd-FPN.yaml +39 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/PP-YOLOE_seg-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/SOLOv2.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_128x96.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_256x192.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocBlockLayout.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-M.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-S.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout_plus-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x_table.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_3cls.yaml +40 -0
- paddlex/configs/modules/mainbody_detection/PP-ShiTuV2_det.yaml +41 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_base.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_large.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_medium.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_small.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_tiny.yaml +12 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-FPN.yaml +41 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/CenterNet-DLA-34.yaml +41 -0
- paddlex/configs/modules/object_detection/CenterNet-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/Co-DINO-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-DINO-Swin-L.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-Swin-T.yaml +40 -0
- paddlex/configs/modules/object_detection/DETR-R50.yaml +42 -0
- paddlex/configs/modules/object_detection/FCOS-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNeXt101-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet34-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-Swin-Tiny-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-M.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-X.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-M.yaml +42 -0
- paddlex/configs/modules/object_detection/PicoDet-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-XS.yaml +42 -0
- paddlex/configs/modules/object_detection/RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R18.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-L.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-M.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-N.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-S.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-T.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-DarkNet53.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-MobileNetV3.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-ResNet50_vd_DCN.yaml +40 -0
- paddlex/configs/modules/open_vocabulary_detection/GroundingDINO-T.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_detection/YOLO-Worldv2-L.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_box.yaml +17 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_point.yaml +15 -0
- paddlex/configs/modules/pedestrian_attribute_recognition/PP-LCNet_x1_0_pedestrian_attribute.yaml +41 -0
- paddlex/configs/modules/rotated_object_detection/PP-YOLOE-R-L.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_mobile_seal_det.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_server_seal_det.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_small.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_tiny.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W18.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W48.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-B.yaml +41 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-T.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_base.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_large.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_small.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_tiny.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B0.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B1.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B2.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B3.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B4.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B5.yaml +40 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-L.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-S.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml +42 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wired_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wireless_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_classification/PP-LCNet_x1_0_table_cls.yaml +41 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wired.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wireless.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet_plus.yaml +39 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_server_det.yaml +40 -0
- paddlex/configs/modules/text_recognition/PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec_doc.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/arabic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_RepSVTR_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_SVTRv2_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/chinese_cht_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/cyrillic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/devanagari_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/japan_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ka_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/korean_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/latin_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ta_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/te_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/textline_orientation/PP-LCNet_x0_25_textline_ori.yaml +41 -0
- paddlex/configs/modules/ts_anomaly_detection/AutoEncoder_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/DLinear_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/Nonstationary_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/PatchTST_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/TimesNet_ad.yaml +37 -0
- paddlex/configs/modules/ts_classification/TimesNet_cls.yaml +37 -0
- paddlex/configs/modules/ts_forecast/DLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/NLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/Nonstationary.yaml +38 -0
- paddlex/configs/modules/ts_forecast/PatchTST.yaml +38 -0
- paddlex/configs/modules/ts_forecast/RLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TiDE.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TimesNet.yaml +38 -0
- paddlex/configs/modules/vehicle_attribute_recognition/PP-LCNet_x1_0_vehicle_attribute.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-L_vehicle.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-S_vehicle.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSM-R50_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_16frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_detection/YOWO.yaml +40 -0
- paddlex/configs/pipelines/3d_bev_detection.yaml +9 -0
- paddlex/configs/pipelines/OCR.yaml +45 -0
- paddlex/configs/pipelines/PP-ChatOCRv3-doc.yaml +151 -0
- paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml +237 -0
- paddlex/configs/pipelines/PP-ShiTuV2.yaml +18 -0
- paddlex/configs/pipelines/PP-StructureV3.yaml +226 -0
- paddlex/configs/pipelines/anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/doc_preprocessor.yaml +15 -0
- paddlex/configs/pipelines/doc_understanding.yaml +9 -0
- paddlex/configs/pipelines/face_recognition.yaml +18 -0
- paddlex/configs/pipelines/formula_recognition.yaml +39 -0
- paddlex/configs/pipelines/human_keypoint_detection.yaml +17 -0
- paddlex/configs/pipelines/image_classification.yaml +10 -0
- paddlex/configs/pipelines/image_multilabel_classification.yaml +9 -0
- paddlex/configs/pipelines/instance_segmentation.yaml +10 -0
- paddlex/configs/pipelines/layout_parsing.yaml +102 -0
- paddlex/configs/pipelines/multilingual_speech_recognition.yaml +9 -0
- paddlex/configs/pipelines/object_detection.yaml +10 -0
- paddlex/configs/pipelines/open_vocabulary_detection.yaml +12 -0
- paddlex/configs/pipelines/open_vocabulary_segmentation.yaml +13 -0
- paddlex/configs/pipelines/pedestrian_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/rotated_object_detection.yaml +10 -0
- paddlex/configs/pipelines/seal_recognition.yaml +52 -0
- paddlex/configs/pipelines/semantic_segmentation.yaml +10 -0
- paddlex/configs/pipelines/small_object_detection.yaml +10 -0
- paddlex/configs/pipelines/table_recognition.yaml +57 -0
- paddlex/configs/pipelines/table_recognition_v2.yaml +82 -0
- paddlex/configs/pipelines/ts_anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/ts_classification.yaml +8 -0
- paddlex/configs/pipelines/ts_forecast.yaml +8 -0
- paddlex/configs/pipelines/vehicle_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/video_classification.yaml +9 -0
- paddlex/configs/pipelines/video_detection.yaml +10 -0
- paddlex/constants.py +17 -0
- paddlex/engine.py +56 -0
- paddlex/hpip_links.html +31 -0
- paddlex/inference/__init__.py +19 -0
- paddlex/inference/common/__init__.py +13 -0
- paddlex/inference/common/batch_sampler/__init__.py +21 -0
- paddlex/inference/common/batch_sampler/audio_batch_sampler.py +83 -0
- paddlex/inference/common/batch_sampler/base_batch_sampler.py +94 -0
- paddlex/inference/common/batch_sampler/det_3d_batch_sampler.py +144 -0
- paddlex/inference/common/batch_sampler/doc_vlm_batch_sampler.py +87 -0
- paddlex/inference/common/batch_sampler/image_batch_sampler.py +121 -0
- paddlex/inference/common/batch_sampler/ts_batch_sampler.py +109 -0
- paddlex/inference/common/batch_sampler/video_batch_sampler.py +74 -0
- paddlex/inference/common/reader/__init__.py +19 -0
- paddlex/inference/common/reader/audio_reader.py +46 -0
- paddlex/inference/common/reader/det_3d_reader.py +241 -0
- paddlex/inference/common/reader/image_reader.py +73 -0
- paddlex/inference/common/reader/ts_reader.py +46 -0
- paddlex/inference/common/reader/video_reader.py +42 -0
- paddlex/inference/common/result/__init__.py +29 -0
- paddlex/inference/common/result/base_cv_result.py +41 -0
- paddlex/inference/common/result/base_result.py +72 -0
- paddlex/inference/common/result/base_ts_result.py +41 -0
- paddlex/inference/common/result/base_video_result.py +36 -0
- paddlex/inference/common/result/mixin.py +709 -0
- paddlex/inference/models/__init__.py +86 -0
- paddlex/inference/models/anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/anomaly_detection/predictor.py +135 -0
- paddlex/inference/models/anomaly_detection/processors.py +53 -0
- paddlex/inference/models/anomaly_detection/result.py +71 -0
- paddlex/inference/models/base/__init__.py +15 -0
- paddlex/inference/models/base/predictor/__init__.py +15 -0
- paddlex/inference/models/base/predictor/base_predictor.py +414 -0
- paddlex/inference/models/common/__init__.py +26 -0
- paddlex/inference/models/common/static_infer.py +801 -0
- paddlex/inference/models/common/tokenizer/__init__.py +21 -0
- paddlex/inference/models/common/tokenizer/bert_tokenizer.py +655 -0
- paddlex/inference/models/common/tokenizer/clip_tokenizer.py +609 -0
- paddlex/inference/models/common/tokenizer/gpt_tokenizer.py +453 -0
- paddlex/inference/models/common/tokenizer/qwen2_5_tokenizer.py +112 -0
- paddlex/inference/models/common/tokenizer/qwen2_tokenizer.py +438 -0
- paddlex/inference/models/common/tokenizer/qwen_tokenizer.py +288 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils.py +2149 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils_base.py +3720 -0
- paddlex/inference/models/common/tokenizer/utils.py +66 -0
- paddlex/inference/models/common/tokenizer/vocab.py +647 -0
- paddlex/inference/models/common/ts/__init__.py +15 -0
- paddlex/inference/models/common/ts/funcs.py +540 -0
- paddlex/inference/models/common/ts/processors.py +322 -0
- paddlex/inference/models/common/vision/__init__.py +23 -0
- paddlex/inference/models/common/vision/funcs.py +98 -0
- paddlex/inference/models/common/vision/processors.py +285 -0
- paddlex/inference/models/common/vlm/__init__.py +13 -0
- paddlex/inference/models/common/vlm/activations.py +189 -0
- paddlex/inference/models/common/vlm/bert_padding.py +127 -0
- paddlex/inference/models/common/vlm/conversion_utils.py +99 -0
- paddlex/inference/models/common/vlm/distributed.py +229 -0
- paddlex/inference/models/common/vlm/flash_attn_utils.py +119 -0
- paddlex/inference/models/common/vlm/fusion_ops.py +205 -0
- paddlex/inference/models/common/vlm/generation/__init__.py +34 -0
- paddlex/inference/models/common/vlm/generation/configuration_utils.py +533 -0
- paddlex/inference/models/common/vlm/generation/logits_process.py +730 -0
- paddlex/inference/models/common/vlm/generation/stopping_criteria.py +106 -0
- paddlex/inference/models/common/vlm/generation/utils.py +2162 -0
- paddlex/inference/models/common/vlm/transformers/__init__.py +16 -0
- paddlex/inference/models/common/vlm/transformers/configuration_utils.py +1037 -0
- paddlex/inference/models/common/vlm/transformers/conversion_utils.py +408 -0
- paddlex/inference/models/common/vlm/transformers/model_outputs.py +1612 -0
- paddlex/inference/models/common/vlm/transformers/model_utils.py +2014 -0
- paddlex/inference/models/common/vlm/transformers/utils.py +178 -0
- paddlex/inference/models/common/vlm/utils.py +109 -0
- paddlex/inference/models/doc_vlm/__init__.py +15 -0
- paddlex/inference/models/doc_vlm/modeling/GOT_ocr_2_0.py +830 -0
- paddlex/inference/models/doc_vlm/modeling/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2.py +1606 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_5_vl.py +3006 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_vl.py +2495 -0
- paddlex/inference/models/doc_vlm/predictor.py +253 -0
- paddlex/inference/models/doc_vlm/processors/GOT_ocr_2_0.py +97 -0
- paddlex/inference/models/doc_vlm/processors/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/processors/common.py +561 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_5_vl.py +548 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_vl.py +543 -0
- paddlex/inference/models/doc_vlm/result.py +21 -0
- paddlex/inference/models/face_feature/__init__.py +15 -0
- paddlex/inference/models/face_feature/predictor.py +66 -0
- paddlex/inference/models/formula_recognition/__init__.py +15 -0
- paddlex/inference/models/formula_recognition/predictor.py +193 -0
- paddlex/inference/models/formula_recognition/processors.py +1015 -0
- paddlex/inference/models/formula_recognition/result.py +411 -0
- paddlex/inference/models/image_classification/__init__.py +15 -0
- paddlex/inference/models/image_classification/predictor.py +172 -0
- paddlex/inference/models/image_classification/processors.py +89 -0
- paddlex/inference/models/image_classification/result.py +93 -0
- paddlex/inference/models/image_feature/__init__.py +15 -0
- paddlex/inference/models/image_feature/predictor.py +146 -0
- paddlex/inference/models/image_feature/processors.py +31 -0
- paddlex/inference/models/image_feature/result.py +32 -0
- paddlex/inference/models/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/models/image_multilabel_classification/predictor.py +95 -0
- paddlex/inference/models/image_multilabel_classification/processors.py +89 -0
- paddlex/inference/models/image_multilabel_classification/result.py +96 -0
- paddlex/inference/models/image_unwarping/__init__.py +15 -0
- paddlex/inference/models/image_unwarping/predictor.py +97 -0
- paddlex/inference/models/image_unwarping/processors.py +92 -0
- paddlex/inference/models/image_unwarping/result.py +47 -0
- paddlex/inference/models/instance_segmentation/__init__.py +15 -0
- paddlex/inference/models/instance_segmentation/predictor.py +202 -0
- paddlex/inference/models/instance_segmentation/processors.py +102 -0
- paddlex/inference/models/instance_segmentation/result.py +162 -0
- paddlex/inference/models/keypoint_detection/__init__.py +15 -0
- paddlex/inference/models/keypoint_detection/predictor.py +190 -0
- paddlex/inference/models/keypoint_detection/processors.py +367 -0
- paddlex/inference/models/keypoint_detection/result.py +197 -0
- paddlex/inference/models/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/models/m_3d_bev_detection/predictor.py +303 -0
- paddlex/inference/models/m_3d_bev_detection/processors.py +990 -0
- paddlex/inference/models/m_3d_bev_detection/result.py +68 -0
- paddlex/inference/models/m_3d_bev_detection/visualizer_3d.py +169 -0
- paddlex/inference/models/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/models/multilingual_speech_recognition/predictor.py +137 -0
- paddlex/inference/models/multilingual_speech_recognition/processors.py +1933 -0
- paddlex/inference/models/multilingual_speech_recognition/result.py +21 -0
- paddlex/inference/models/object_detection/__init__.py +15 -0
- paddlex/inference/models/object_detection/predictor.py +344 -0
- paddlex/inference/models/object_detection/processors.py +885 -0
- paddlex/inference/models/object_detection/result.py +114 -0
- paddlex/inference/models/object_detection/utils.py +70 -0
- paddlex/inference/models/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_detection/predictor.py +172 -0
- paddlex/inference/models/open_vocabulary_detection/processors/__init__.py +16 -0
- paddlex/inference/models/open_vocabulary_detection/processors/common.py +114 -0
- paddlex/inference/models/open_vocabulary_detection/processors/groundingdino_processors.py +496 -0
- paddlex/inference/models/open_vocabulary_detection/processors/yoloworld_processors.py +209 -0
- paddlex/inference/models/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/predictor.py +113 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/sam_processer.py +249 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/sam_result.py +149 -0
- paddlex/inference/models/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/models/semantic_segmentation/predictor.py +158 -0
- paddlex/inference/models/semantic_segmentation/processors.py +117 -0
- paddlex/inference/models/semantic_segmentation/result.py +73 -0
- paddlex/inference/models/table_structure_recognition/__init__.py +15 -0
- paddlex/inference/models/table_structure_recognition/predictor.py +161 -0
- paddlex/inference/models/table_structure_recognition/processors.py +229 -0
- paddlex/inference/models/table_structure_recognition/result.py +63 -0
- paddlex/inference/models/text_detection/__init__.py +15 -0
- paddlex/inference/models/text_detection/predictor.py +191 -0
- paddlex/inference/models/text_detection/processors.py +538 -0
- paddlex/inference/models/text_detection/result.py +46 -0
- paddlex/inference/models/text_recognition/__init__.py +15 -0
- paddlex/inference/models/text_recognition/predictor.py +98 -0
- paddlex/inference/models/text_recognition/processors.py +245 -0
- paddlex/inference/models/text_recognition/result.py +76 -0
- paddlex/inference/models/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/ts_anomaly_detection/predictor.py +141 -0
- paddlex/inference/models/ts_anomaly_detection/processors.py +98 -0
- paddlex/inference/models/ts_anomaly_detection/result.py +83 -0
- paddlex/inference/models/ts_classification/__init__.py +15 -0
- paddlex/inference/models/ts_classification/predictor.py +122 -0
- paddlex/inference/models/ts_classification/processors.py +122 -0
- paddlex/inference/models/ts_classification/result.py +87 -0
- paddlex/inference/models/ts_forecasting/__init__.py +15 -0
- paddlex/inference/models/ts_forecasting/predictor.py +154 -0
- paddlex/inference/models/ts_forecasting/processors.py +158 -0
- paddlex/inference/models/ts_forecasting/result.py +96 -0
- paddlex/inference/models/video_classification/__init__.py +15 -0
- paddlex/inference/models/video_classification/predictor.py +141 -0
- paddlex/inference/models/video_classification/processors.py +409 -0
- paddlex/inference/models/video_classification/result.py +96 -0
- paddlex/inference/models/video_detection/__init__.py +15 -0
- paddlex/inference/models/video_detection/predictor.py +129 -0
- paddlex/inference/models/video_detection/processors.py +463 -0
- paddlex/inference/models/video_detection/result.py +109 -0
- paddlex/inference/pipelines/__init__.py +239 -0
- paddlex/inference/pipelines/_parallel.py +172 -0
- paddlex/inference/pipelines/anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/anomaly_detection/pipeline.py +82 -0
- paddlex/inference/pipelines/attribute_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/attribute_recognition/pipeline.py +120 -0
- paddlex/inference/pipelines/attribute_recognition/result.py +102 -0
- paddlex/inference/pipelines/base.py +156 -0
- paddlex/inference/pipelines/components/__init__.py +29 -0
- paddlex/inference/pipelines/components/chat_server/__init__.py +16 -0
- paddlex/inference/pipelines/components/chat_server/base.py +39 -0
- paddlex/inference/pipelines/components/chat_server/openai_bot_chat.py +236 -0
- paddlex/inference/pipelines/components/common/__init__.py +19 -0
- paddlex/inference/pipelines/components/common/base_operator.py +37 -0
- paddlex/inference/pipelines/components/common/base_result.py +66 -0
- paddlex/inference/pipelines/components/common/convert_points_and_boxes.py +45 -0
- paddlex/inference/pipelines/components/common/crop_image_regions.py +556 -0
- paddlex/inference/pipelines/components/common/seal_det_warp.py +972 -0
- paddlex/inference/pipelines/components/common/sort_boxes.py +85 -0
- paddlex/inference/pipelines/components/common/warp_image.py +50 -0
- paddlex/inference/pipelines/components/faisser.py +357 -0
- paddlex/inference/pipelines/components/prompt_engineering/__init__.py +16 -0
- paddlex/inference/pipelines/components/prompt_engineering/base.py +35 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_ensemble_prompt.py +128 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_kie_prompt.py +148 -0
- paddlex/inference/pipelines/components/retriever/__init__.py +16 -0
- paddlex/inference/pipelines/components/retriever/base.py +228 -0
- paddlex/inference/pipelines/components/retriever/openai_bot_retriever.py +70 -0
- paddlex/inference/pipelines/components/retriever/qianfan_bot_retriever.py +166 -0
- paddlex/inference/pipelines/components/utils/__init__.py +13 -0
- paddlex/inference/pipelines/components/utils/mixin.py +206 -0
- paddlex/inference/pipelines/doc_preprocessor/__init__.py +15 -0
- paddlex/inference/pipelines/doc_preprocessor/pipeline.py +209 -0
- paddlex/inference/pipelines/doc_preprocessor/result.py +98 -0
- paddlex/inference/pipelines/doc_understanding/__init__.py +15 -0
- paddlex/inference/pipelines/doc_understanding/pipeline.py +71 -0
- paddlex/inference/pipelines/face_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/face_recognition/pipeline.py +63 -0
- paddlex/inference/pipelines/face_recognition/result.py +44 -0
- paddlex/inference/pipelines/formula_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/formula_recognition/pipeline.py +347 -0
- paddlex/inference/pipelines/formula_recognition/result.py +282 -0
- paddlex/inference/pipelines/image_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_classification/pipeline.py +90 -0
- paddlex/inference/pipelines/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_multilabel_classification/pipeline.py +97 -0
- paddlex/inference/pipelines/instance_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/instance_segmentation/pipeline.py +91 -0
- paddlex/inference/pipelines/keypoint_detection/__init__.py +15 -0
- paddlex/inference/pipelines/keypoint_detection/pipeline.py +158 -0
- paddlex/inference/pipelines/layout_parsing/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/pipeline.py +568 -0
- paddlex/inference/pipelines/layout_parsing/pipeline_v2.py +1382 -0
- paddlex/inference/pipelines/layout_parsing/result.py +191 -0
- paddlex/inference/pipelines/layout_parsing/result_v2.py +745 -0
- paddlex/inference/pipelines/layout_parsing/setting.py +87 -0
- paddlex/inference/pipelines/layout_parsing/utils.py +951 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/utils.py +1143 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/xycuts.py +562 -0
- paddlex/inference/pipelines/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/pipelines/m_3d_bev_detection/pipeline.py +74 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/pipeline.py +78 -0
- paddlex/inference/pipelines/object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/object_detection/pipeline.py +115 -0
- paddlex/inference/pipelines/ocr/__init__.py +15 -0
- paddlex/inference/pipelines/ocr/pipeline.py +463 -0
- paddlex/inference/pipelines/ocr/result.py +255 -0
- paddlex/inference/pipelines/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_detection/pipeline.py +86 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/pipeline.py +100 -0
- paddlex/inference/pipelines/pp_chatocr/__init__.py +16 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_base.py +111 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v3.py +781 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v4.py +992 -0
- paddlex/inference/pipelines/pp_shitu_v2/__init__.py +15 -0
- paddlex/inference/pipelines/pp_shitu_v2/pipeline.py +156 -0
- paddlex/inference/pipelines/pp_shitu_v2/result.py +126 -0
- paddlex/inference/pipelines/rotated_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/rotated_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/seal_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/seal_recognition/pipeline.py +335 -0
- paddlex/inference/pipelines/seal_recognition/result.py +89 -0
- paddlex/inference/pipelines/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/semantic_segmentation/pipeline.py +95 -0
- paddlex/inference/pipelines/small_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/small_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/table_recognition/__init__.py +16 -0
- paddlex/inference/pipelines/table_recognition/pipeline.py +486 -0
- paddlex/inference/pipelines/table_recognition/pipeline_v2.py +1395 -0
- paddlex/inference/pipelines/table_recognition/result.py +218 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing.py +366 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing_v2.py +488 -0
- paddlex/inference/pipelines/table_recognition/utils.py +44 -0
- paddlex/inference/pipelines/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/ts_anomaly_detection/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_classification/__init__.py +15 -0
- paddlex/inference/pipelines/ts_classification/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_forecasting/__init__.py +15 -0
- paddlex/inference/pipelines/ts_forecasting/pipeline.py +72 -0
- paddlex/inference/pipelines/video_classification/__init__.py +15 -0
- paddlex/inference/pipelines/video_classification/pipeline.py +79 -0
- paddlex/inference/pipelines/video_detection/__init__.py +15 -0
- paddlex/inference/pipelines/video_detection/pipeline.py +86 -0
- paddlex/inference/serving/__init__.py +17 -0
- paddlex/inference/serving/basic_serving/__init__.py +18 -0
- paddlex/inference/serving/basic_serving/_app.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/__init__.py +44 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/__init__.py +13 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/common.py +104 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/image_recognition.py +36 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/ocr.py +95 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/anomaly_detection.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_preprocessor.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_understanding.py +153 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/face_recognition.py +226 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/formula_recognition.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/human_keypoint_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_classification.py +69 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_multilabel_classification.py +73 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/instance_segmentation.py +87 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/layout_parsing.py +117 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/m_3d_bev_detection.py +79 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/multilingual_speech_recognition.py +92 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/object_detection.py +77 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ocr.py +102 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_segmentation.py +91 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pedestrian_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv3_doc.py +193 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv4_doc.py +223 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_shituv2.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_structurev3.py +143 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/rotated_object_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/seal_recognition.py +106 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/semantic_segmentation.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/small_object_detection.py +72 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition.py +108 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition_v2.py +113 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_anomaly_detection.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_classification.py +64 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_forecast.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/vehicle_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_classification.py +76 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_detection.py +92 -0
- paddlex/inference/serving/basic_serving/_server.py +40 -0
- paddlex/inference/serving/infra/__init__.py +13 -0
- paddlex/inference/serving/infra/config.py +36 -0
- paddlex/inference/serving/infra/models.py +79 -0
- paddlex/inference/serving/infra/storage.py +180 -0
- paddlex/inference/serving/infra/utils.py +285 -0
- paddlex/inference/serving/schemas/__init__.py +13 -0
- paddlex/inference/serving/schemas/anomaly_detection.py +39 -0
- paddlex/inference/serving/schemas/doc_preprocessor.py +54 -0
- paddlex/inference/serving/schemas/doc_understanding.py +78 -0
- paddlex/inference/serving/schemas/face_recognition.py +124 -0
- paddlex/inference/serving/schemas/formula_recognition.py +56 -0
- paddlex/inference/serving/schemas/human_keypoint_detection.py +55 -0
- paddlex/inference/serving/schemas/image_classification.py +45 -0
- paddlex/inference/serving/schemas/image_multilabel_classification.py +47 -0
- paddlex/inference/serving/schemas/instance_segmentation.py +53 -0
- paddlex/inference/serving/schemas/layout_parsing.py +71 -0
- paddlex/inference/serving/schemas/m_3d_bev_detection.py +48 -0
- paddlex/inference/serving/schemas/multilingual_speech_recognition.py +57 -0
- paddlex/inference/serving/schemas/object_detection.py +52 -0
- paddlex/inference/serving/schemas/ocr.py +60 -0
- paddlex/inference/serving/schemas/open_vocabulary_detection.py +52 -0
- paddlex/inference/serving/schemas/open_vocabulary_segmentation.py +52 -0
- paddlex/inference/serving/schemas/pedestrian_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/pp_chatocrv3_doc.py +133 -0
- paddlex/inference/serving/schemas/pp_chatocrv4_doc.py +150 -0
- paddlex/inference/serving/schemas/pp_shituv2.py +124 -0
- paddlex/inference/serving/schemas/pp_structurev3.py +88 -0
- paddlex/inference/serving/schemas/rotated_object_detection.py +52 -0
- paddlex/inference/serving/schemas/seal_recognition.py +62 -0
- paddlex/inference/serving/schemas/semantic_segmentation.py +45 -0
- paddlex/inference/serving/schemas/shared/__init__.py +13 -0
- paddlex/inference/serving/schemas/shared/classification.py +23 -0
- paddlex/inference/serving/schemas/shared/image_segmentation.py +28 -0
- paddlex/inference/serving/schemas/shared/object_detection.py +24 -0
- paddlex/inference/serving/schemas/shared/ocr.py +25 -0
- paddlex/inference/serving/schemas/small_object_detection.py +52 -0
- paddlex/inference/serving/schemas/table_recognition.py +64 -0
- paddlex/inference/serving/schemas/table_recognition_v2.py +69 -0
- paddlex/inference/serving/schemas/ts_anomaly_detection.py +37 -0
- paddlex/inference/serving/schemas/ts_classification.py +38 -0
- paddlex/inference/serving/schemas/ts_forecast.py +37 -0
- paddlex/inference/serving/schemas/vehicle_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/video_classification.py +44 -0
- paddlex/inference/serving/schemas/video_detection.py +56 -0
- paddlex/inference/utils/__init__.py +13 -0
- paddlex/inference/utils/benchmark.py +379 -0
- paddlex/inference/utils/color_map.py +123 -0
- paddlex/inference/utils/get_pipeline_path.py +27 -0
- paddlex/inference/utils/hpi.py +254 -0
- paddlex/inference/utils/hpi_model_info_collection.json +2331 -0
- paddlex/inference/utils/io/__init__.py +36 -0
- paddlex/inference/utils/io/readers.py +504 -0
- paddlex/inference/utils/io/style.py +381 -0
- paddlex/inference/utils/io/tablepyxl.py +157 -0
- paddlex/inference/utils/io/writers.py +458 -0
- paddlex/inference/utils/model_paths.py +48 -0
- paddlex/inference/utils/new_ir_blocklist.py +27 -0
- paddlex/inference/utils/official_models.py +367 -0
- paddlex/inference/utils/pp_option.py +339 -0
- paddlex/inference/utils/trt_blocklist.py +43 -0
- paddlex/inference/utils/trt_config.py +420 -0
- paddlex/model.py +131 -0
- paddlex/modules/__init__.py +115 -0
- paddlex/modules/anomaly_detection/__init__.py +18 -0
- paddlex/modules/anomaly_detection/dataset_checker/__init__.py +94 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +233 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/visualizer.py +76 -0
- paddlex/modules/anomaly_detection/evaluator.py +58 -0
- paddlex/modules/anomaly_detection/exportor.py +22 -0
- paddlex/modules/anomaly_detection/model_list.py +16 -0
- paddlex/modules/anomaly_detection/trainer.py +70 -0
- paddlex/modules/base/__init__.py +18 -0
- paddlex/modules/base/build_model.py +33 -0
- paddlex/modules/base/dataset_checker/__init__.py +16 -0
- paddlex/modules/base/dataset_checker/dataset_checker.py +169 -0
- paddlex/modules/base/dataset_checker/utils.py +108 -0
- paddlex/modules/base/evaluator.py +170 -0
- paddlex/modules/base/exportor.py +145 -0
- paddlex/modules/base/trainer.py +144 -0
- paddlex/modules/base/utils/__init__.py +13 -0
- paddlex/modules/base/utils/cinn_setting.py +89 -0
- paddlex/modules/base/utils/coco_eval.py +94 -0
- paddlex/modules/base/utils/topk_eval.py +118 -0
- paddlex/modules/doc_vlm/__init__.py +18 -0
- paddlex/modules/doc_vlm/dataset_checker.py +29 -0
- paddlex/modules/doc_vlm/evaluator.py +29 -0
- paddlex/modules/doc_vlm/exportor.py +29 -0
- paddlex/modules/doc_vlm/model_list.py +16 -0
- paddlex/modules/doc_vlm/trainer.py +41 -0
- paddlex/modules/face_recognition/__init__.py +18 -0
- paddlex/modules/face_recognition/dataset_checker/__init__.py +71 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/__init__.py +16 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/check_dataset.py +172 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/face_recognition/evaluator.py +52 -0
- paddlex/modules/face_recognition/exportor.py +22 -0
- paddlex/modules/face_recognition/model_list.py +15 -0
- paddlex/modules/face_recognition/trainer.py +75 -0
- paddlex/modules/formula_recognition/__init__.py +18 -0
- paddlex/modules/formula_recognition/dataset_checker/__init__.py +113 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/analyse_dataset.py +158 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/check_dataset.py +76 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/formula_recognition/evaluator.py +80 -0
- paddlex/modules/formula_recognition/exportor.py +22 -0
- paddlex/modules/formula_recognition/model_list.py +23 -0
- paddlex/modules/formula_recognition/trainer.py +123 -0
- paddlex/modules/general_recognition/__init__.py +18 -0
- paddlex/modules/general_recognition/dataset_checker/__init__.py +107 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/analyse_dataset.py +96 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/check_dataset.py +99 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/convert_dataset.py +100 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/visualizer.py +147 -0
- paddlex/modules/general_recognition/evaluator.py +31 -0
- paddlex/modules/general_recognition/exportor.py +22 -0
- paddlex/modules/general_recognition/model_list.py +19 -0
- paddlex/modules/general_recognition/trainer.py +52 -0
- paddlex/modules/image_classification/__init__.py +18 -0
- paddlex/modules/image_classification/dataset_checker/__init__.py +104 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/analyse_dataset.py +92 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/convert_dataset.py +51 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/image_classification/evaluator.py +43 -0
- paddlex/modules/image_classification/exportor.py +22 -0
- paddlex/modules/image_classification/model_list.py +99 -0
- paddlex/modules/image_classification/trainer.py +82 -0
- paddlex/modules/image_unwarping/__init__.py +13 -0
- paddlex/modules/image_unwarping/model_list.py +17 -0
- paddlex/modules/instance_segmentation/__init__.py +18 -0
- paddlex/modules/instance_segmentation/dataset_checker/__init__.py +107 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/check_dataset.py +95 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/convert_dataset.py +241 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/split_dataset.py +122 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/visualizer.py +223 -0
- paddlex/modules/instance_segmentation/evaluator.py +32 -0
- paddlex/modules/instance_segmentation/exportor.py +22 -0
- paddlex/modules/instance_segmentation/model_list.py +33 -0
- paddlex/modules/instance_segmentation/trainer.py +31 -0
- paddlex/modules/keypoint_detection/__init__.py +18 -0
- paddlex/modules/keypoint_detection/dataset_checker/__init__.py +56 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/__init__.py +15 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/visualizer.py +124 -0
- paddlex/modules/keypoint_detection/evaluator.py +41 -0
- paddlex/modules/keypoint_detection/exportor.py +22 -0
- paddlex/modules/keypoint_detection/model_list.py +16 -0
- paddlex/modules/keypoint_detection/trainer.py +39 -0
- paddlex/modules/m_3d_bev_detection/__init__.py +18 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/__init__.py +95 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/analyse_dataset.py +106 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/check_dataset.py +101 -0
- paddlex/modules/m_3d_bev_detection/evaluator.py +46 -0
- paddlex/modules/m_3d_bev_detection/exportor.py +22 -0
- paddlex/modules/m_3d_bev_detection/model_list.py +18 -0
- paddlex/modules/m_3d_bev_detection/trainer.py +68 -0
- paddlex/modules/multilabel_classification/__init__.py +18 -0
- paddlex/modules/multilabel_classification/dataset_checker/__init__.py +106 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/analyse_dataset.py +94 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/convert_dataset.py +120 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/visualizer.py +149 -0
- paddlex/modules/multilabel_classification/evaluator.py +43 -0
- paddlex/modules/multilabel_classification/exportor.py +22 -0
- paddlex/modules/multilabel_classification/model_list.py +24 -0
- paddlex/modules/multilabel_classification/trainer.py +85 -0
- paddlex/modules/multilingual_speech_recognition/__init__.py +18 -0
- paddlex/modules/multilingual_speech_recognition/dataset_checker.py +27 -0
- paddlex/modules/multilingual_speech_recognition/evaluator.py +27 -0
- paddlex/modules/multilingual_speech_recognition/exportor.py +27 -0
- paddlex/modules/multilingual_speech_recognition/model_list.py +22 -0
- paddlex/modules/multilingual_speech_recognition/trainer.py +42 -0
- paddlex/modules/object_detection/__init__.py +18 -0
- paddlex/modules/object_detection/dataset_checker/__init__.py +106 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/convert_dataset.py +438 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/split_dataset.py +123 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/visualizer.py +193 -0
- paddlex/modules/object_detection/evaluator.py +57 -0
- paddlex/modules/object_detection/exportor.py +22 -0
- paddlex/modules/object_detection/model_list.py +86 -0
- paddlex/modules/object_detection/trainer.py +98 -0
- paddlex/modules/open_vocabulary_detection/__init__.py +18 -0
- paddlex/modules/open_vocabulary_detection/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_detection/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_detection/exportor.py +29 -0
- paddlex/modules/open_vocabulary_detection/model_list.py +16 -0
- paddlex/modules/open_vocabulary_detection/trainer.py +44 -0
- paddlex/modules/open_vocabulary_segmentation/__init__.py +18 -0
- paddlex/modules/open_vocabulary_segmentation/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/exportor.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/model_list.py +19 -0
- paddlex/modules/open_vocabulary_segmentation/trainer.py +44 -0
- paddlex/modules/semantic_segmentation/__init__.py +18 -0
- paddlex/modules/semantic_segmentation/dataset_checker/__init__.py +109 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/analyse_dataset.py +76 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/check_dataset.py +80 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/convert_dataset.py +165 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/visualizer.py +75 -0
- paddlex/modules/semantic_segmentation/evaluator.py +58 -0
- paddlex/modules/semantic_segmentation/exportor.py +31 -0
- paddlex/modules/semantic_segmentation/model_list.py +37 -0
- paddlex/modules/semantic_segmentation/trainer.py +72 -0
- paddlex/modules/table_recognition/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/__init__.py +98 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/analyse_dataset.py +59 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py +87 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/table_recognition/evaluator.py +43 -0
- paddlex/modules/table_recognition/exportor.py +22 -0
- paddlex/modules/table_recognition/model_list.py +21 -0
- paddlex/modules/table_recognition/trainer.py +67 -0
- paddlex/modules/text_detection/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/__init__.py +107 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/analyse_dataset.py +220 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py +106 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/split_dataset.py +140 -0
- paddlex/modules/text_detection/evaluator.py +41 -0
- paddlex/modules/text_detection/exportor.py +22 -0
- paddlex/modules/text_detection/model_list.py +26 -0
- paddlex/modules/text_detection/trainer.py +65 -0
- paddlex/modules/text_recognition/__init__.py +18 -0
- paddlex/modules/text_recognition/dataset_checker/__init__.py +125 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/analyse_dataset.py +162 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py +104 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/text_recognition/evaluator.py +64 -0
- paddlex/modules/text_recognition/exportor.py +22 -0
- paddlex/modules/text_recognition/model_list.py +36 -0
- paddlex/modules/text_recognition/trainer.py +105 -0
- paddlex/modules/ts_anomaly_detection/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_anomaly_detection/evaluator.py +67 -0
- paddlex/modules/ts_anomaly_detection/exportor.py +44 -0
- paddlex/modules/ts_anomaly_detection/model_list.py +22 -0
- paddlex/modules/ts_anomaly_detection/trainer.py +113 -0
- paddlex/modules/ts_classification/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/analyse_dataset.py +77 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/split_dataset.py +88 -0
- paddlex/modules/ts_classification/evaluator.py +66 -0
- paddlex/modules/ts_classification/exportor.py +44 -0
- paddlex/modules/ts_classification/model_list.py +18 -0
- paddlex/modules/ts_classification/trainer.py +108 -0
- paddlex/modules/ts_forecast/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/convert_dataset.py +73 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_forecast/evaluator.py +66 -0
- paddlex/modules/ts_forecast/exportor.py +44 -0
- paddlex/modules/ts_forecast/model_list.py +24 -0
- paddlex/modules/ts_forecast/trainer.py +108 -0
- paddlex/modules/video_classification/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/__init__.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/analyse_dataset.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/check_dataset.py +120 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/video_classification/evaluator.py +44 -0
- paddlex/modules/video_classification/exportor.py +22 -0
- paddlex/modules/video_classification/model_list.py +19 -0
- paddlex/modules/video_classification/trainer.py +88 -0
- paddlex/modules/video_detection/__init__.py +18 -0
- paddlex/modules/video_detection/dataset_checker/__init__.py +86 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/analyse_dataset.py +100 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/video_detection/evaluator.py +42 -0
- paddlex/modules/video_detection/exportor.py +22 -0
- paddlex/modules/video_detection/model_list.py +15 -0
- paddlex/modules/video_detection/trainer.py +82 -0
- paddlex/ops/__init__.py +152 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.cpp +266 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.h +28 -0
- paddlex/ops/iou3d_nms/iou3d_nms.cpp +206 -0
- paddlex/ops/iou3d_nms/iou3d_nms.h +35 -0
- paddlex/ops/iou3d_nms/iou3d_nms_api.cpp +114 -0
- paddlex/ops/iou3d_nms/iou3d_nms_kernel.cu +484 -0
- paddlex/ops/setup.py +37 -0
- paddlex/ops/voxel/voxelize_op.cc +194 -0
- paddlex/ops/voxel/voxelize_op.cu +346 -0
- paddlex/paddlex_cli.py +476 -0
- paddlex/repo_apis/Paddle3D_api/__init__.py +17 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/__init__.py +18 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/config.py +118 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py +238 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/register.py +55 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/runner.py +104 -0
- paddlex/repo_apis/Paddle3D_api/pp3d_config.py +145 -0
- paddlex/repo_apis/PaddleClas_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleClas_api/cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleClas_api/cls/config.py +595 -0
- paddlex/repo_apis/PaddleClas_api/cls/model.py +355 -0
- paddlex/repo_apis/PaddleClas_api/cls/register.py +907 -0
- paddlex/repo_apis/PaddleClas_api/cls/runner.py +218 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/__init__.py +18 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/config.py +141 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/model.py +20 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/register.py +68 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/runner.py +50 -0
- paddlex/repo_apis/PaddleDetection_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleDetection_api/config_helper.py +280 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/__init__.py +18 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/config.py +457 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py +403 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/register.py +262 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/runner.py +225 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/config.py +540 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/model.py +429 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py +245 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/register.py +1135 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/runner.py +225 -0
- paddlex/repo_apis/PaddleNLP_api/__init__.py +13 -0
- paddlex/repo_apis/PaddleOCR_api/__init__.py +22 -0
- paddlex/repo_apis/PaddleOCR_api/config_utils.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/config.py +571 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/register.py +99 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/config.py +64 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/model.py +126 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/register.py +70 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/runner.py +51 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/config.py +62 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/model.py +72 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/register.py +107 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/runner.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/config.py +564 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/register.py +216 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleSeg_api/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/base_seg_config.py +134 -0
- paddlex/repo_apis/PaddleSeg_api/seg/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/seg/config.py +183 -0
- paddlex/repo_apis/PaddleSeg_api/seg/model.py +491 -0
- paddlex/repo_apis/PaddleSeg_api/seg/register.py +272 -0
- paddlex/repo_apis/PaddleSeg_api/seg/runner.py +261 -0
- paddlex/repo_apis/PaddleTS_api/__init__.py +20 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/config.py +88 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/register.py +146 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/__init__.py +13 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/config.py +244 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/model.py +276 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/config.py +72 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/register.py +59 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/config.py +136 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/register.py +186 -0
- paddlex/repo_apis/PaddleVideo_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleVideo_api/config_utils.py +51 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/config.py +548 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/model.py +346 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/register.py +70 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/runner.py +204 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/config.py +549 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/model.py +298 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/register.py +44 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/runner.py +199 -0
- paddlex/repo_apis/__init__.py +13 -0
- paddlex/repo_apis/base/__init__.py +22 -0
- paddlex/repo_apis/base/config.py +237 -0
- paddlex/repo_apis/base/model.py +563 -0
- paddlex/repo_apis/base/register.py +135 -0
- paddlex/repo_apis/base/runner.py +390 -0
- paddlex/repo_apis/base/utils/__init__.py +13 -0
- paddlex/repo_apis/base/utils/arg.py +64 -0
- paddlex/repo_apis/base/utils/subprocess.py +107 -0
- paddlex/repo_manager/__init__.py +17 -0
- paddlex/repo_manager/core.py +253 -0
- paddlex/repo_manager/meta.py +180 -0
- paddlex/repo_manager/repo.py +425 -0
- paddlex/repo_manager/utils.py +148 -0
- paddlex/utils/__init__.py +1 -12
- paddlex/utils/cache.py +146 -0
- paddlex/utils/config.py +216 -0
- paddlex/utils/custom_device_list.py +311 -0
- paddlex/utils/deps.py +249 -0
- paddlex/utils/device.py +195 -0
- paddlex/utils/download.py +168 -182
- paddlex/utils/env.py +31 -48
- paddlex/utils/errors/__init__.py +17 -0
- paddlex/utils/errors/dataset_checker.py +78 -0
- paddlex/utils/errors/others.py +138 -0
- paddlex/utils/file_interface.py +211 -0
- paddlex/utils/flags.py +70 -0
- paddlex/utils/fonts/__init__.py +97 -0
- paddlex/utils/func_register.py +41 -0
- paddlex/utils/install.py +87 -0
- paddlex/utils/interactive_get_pipeline.py +55 -0
- paddlex/utils/lazy_loader.py +68 -0
- paddlex/utils/logging.py +140 -33
- paddlex/utils/misc.py +201 -0
- paddlex/utils/pipeline_arguments.py +719 -0
- paddlex/utils/result_saver.py +58 -0
- paddlex/utils/subclass_register.py +99 -0
- paddlex/version.py +55 -0
- paddlex-3.0.0.dist-info/METADATA +1168 -0
- paddlex-3.0.0.dist-info/RECORD +1093 -0
- paddlex-3.0.0.dist-info/WHEEL +5 -0
- paddlex-3.0.0.dist-info/entry_points.txt +2 -0
- paddlex-3.0.0.dist-info/licenses/LICENSE +169 -0
- paddlex-3.0.0.dist-info/top_level.txt +1 -0
- PaddleClas/__init__.py +0 -16
- PaddleClas/paddleclas.py +0 -375
- PaddleClas/ppcls/__init__.py +0 -20
- PaddleClas/ppcls/data/__init__.py +0 -15
- PaddleClas/ppcls/data/imaug/__init__.py +0 -94
- PaddleClas/ppcls/data/imaug/autoaugment.py +0 -264
- PaddleClas/ppcls/data/imaug/batch_operators.py +0 -117
- PaddleClas/ppcls/data/imaug/cutout.py +0 -41
- PaddleClas/ppcls/data/imaug/fmix.py +0 -217
- PaddleClas/ppcls/data/imaug/grid.py +0 -89
- PaddleClas/ppcls/data/imaug/hide_and_seek.py +0 -44
- PaddleClas/ppcls/data/imaug/operators.py +0 -244
- PaddleClas/ppcls/data/imaug/randaugment.py +0 -106
- PaddleClas/ppcls/data/imaug/random_erasing.py +0 -55
- PaddleClas/ppcls/data/reader.py +0 -318
- PaddleClas/ppcls/modeling/__init__.py +0 -20
- PaddleClas/ppcls/modeling/architectures/__init__.py +0 -51
- PaddleClas/ppcls/modeling/architectures/alexnet.py +0 -132
- PaddleClas/ppcls/modeling/architectures/darknet.py +0 -161
- PaddleClas/ppcls/modeling/architectures/densenet.py +0 -308
- PaddleClas/ppcls/modeling/architectures/distillation_models.py +0 -65
- PaddleClas/ppcls/modeling/architectures/distilled_vision_transformer.py +0 -196
- PaddleClas/ppcls/modeling/architectures/dpn.py +0 -425
- PaddleClas/ppcls/modeling/architectures/efficientnet.py +0 -901
- PaddleClas/ppcls/modeling/architectures/ghostnet.py +0 -331
- PaddleClas/ppcls/modeling/architectures/googlenet.py +0 -207
- PaddleClas/ppcls/modeling/architectures/hrnet.py +0 -742
- PaddleClas/ppcls/modeling/architectures/inception_v3.py +0 -481
- PaddleClas/ppcls/modeling/architectures/inception_v4.py +0 -455
- PaddleClas/ppcls/modeling/architectures/mixnet.py +0 -782
- PaddleClas/ppcls/modeling/architectures/mobilenet_v1.py +0 -266
- PaddleClas/ppcls/modeling/architectures/mobilenet_v2.py +0 -248
- PaddleClas/ppcls/modeling/architectures/mobilenet_v3.py +0 -359
- PaddleClas/ppcls/modeling/architectures/regnet.py +0 -383
- PaddleClas/ppcls/modeling/architectures/repvgg.py +0 -339
- PaddleClas/ppcls/modeling/architectures/res2net.py +0 -272
- PaddleClas/ppcls/modeling/architectures/res2net_vd.py +0 -295
- PaddleClas/ppcls/modeling/architectures/resnest.py +0 -705
- PaddleClas/ppcls/modeling/architectures/resnet.py +0 -316
- PaddleClas/ppcls/modeling/architectures/resnet_vc.py +0 -309
- PaddleClas/ppcls/modeling/architectures/resnet_vd.py +0 -354
- PaddleClas/ppcls/modeling/architectures/resnext.py +0 -253
- PaddleClas/ppcls/modeling/architectures/resnext101_wsl.py +0 -447
- PaddleClas/ppcls/modeling/architectures/resnext_vd.py +0 -266
- PaddleClas/ppcls/modeling/architectures/rexnet.py +0 -240
- PaddleClas/ppcls/modeling/architectures/se_resnet_vd.py +0 -378
- PaddleClas/ppcls/modeling/architectures/se_resnext.py +0 -290
- PaddleClas/ppcls/modeling/architectures/se_resnext_vd.py +0 -285
- PaddleClas/ppcls/modeling/architectures/shufflenet_v2.py +0 -320
- PaddleClas/ppcls/modeling/architectures/squeezenet.py +0 -154
- PaddleClas/ppcls/modeling/architectures/vgg.py +0 -152
- PaddleClas/ppcls/modeling/architectures/vision_transformer.py +0 -402
- PaddleClas/ppcls/modeling/architectures/xception.py +0 -345
- PaddleClas/ppcls/modeling/architectures/xception_deeplab.py +0 -386
- PaddleClas/ppcls/modeling/loss.py +0 -154
- PaddleClas/ppcls/modeling/utils.py +0 -53
- PaddleClas/ppcls/optimizer/__init__.py +0 -19
- PaddleClas/ppcls/optimizer/learning_rate.py +0 -159
- PaddleClas/ppcls/optimizer/optimizer.py +0 -165
- PaddleClas/ppcls/utils/__init__.py +0 -27
- PaddleClas/ppcls/utils/check.py +0 -151
- PaddleClas/ppcls/utils/config.py +0 -201
- PaddleClas/ppcls/utils/logger.py +0 -120
- PaddleClas/ppcls/utils/metrics.py +0 -107
- PaddleClas/ppcls/utils/misc.py +0 -62
- PaddleClas/ppcls/utils/model_zoo.py +0 -213
- PaddleClas/ppcls/utils/save_load.py +0 -163
- PaddleClas/setup.py +0 -55
- PaddleClas/tools/__init__.py +0 -15
- PaddleClas/tools/download.py +0 -50
- PaddleClas/tools/ema.py +0 -58
- PaddleClas/tools/eval.py +0 -112
- PaddleClas/tools/export_model.py +0 -85
- PaddleClas/tools/export_serving_model.py +0 -76
- PaddleClas/tools/infer/__init__.py +0 -16
- PaddleClas/tools/infer/infer.py +0 -94
- PaddleClas/tools/infer/predict.py +0 -117
- PaddleClas/tools/infer/utils.py +0 -233
- PaddleClas/tools/program.py +0 -444
- PaddleClas/tools/test_hubserving.py +0 -113
- PaddleClas/tools/train.py +0 -141
- paddlex/cls.py +0 -76
- paddlex/command.py +0 -215
- paddlex/cv/__init__.py +0 -17
- paddlex/cv/datasets/__init__.py +0 -18
- paddlex/cv/datasets/coco.py +0 -169
- paddlex/cv/datasets/imagenet.py +0 -88
- paddlex/cv/datasets/seg_dataset.py +0 -91
- paddlex/cv/datasets/voc.py +0 -301
- paddlex/cv/models/__init__.py +0 -18
- paddlex/cv/models/base.py +0 -623
- paddlex/cv/models/classifier.py +0 -814
- paddlex/cv/models/detector.py +0 -1747
- paddlex/cv/models/load_model.py +0 -126
- paddlex/cv/models/segmenter.py +0 -673
- paddlex/cv/models/slim/__init__.py +0 -13
- paddlex/cv/models/slim/prune.py +0 -55
- paddlex/cv/models/utils/__init__.py +0 -13
- paddlex/cv/models/utils/det_metrics/__init__.py +0 -15
- paddlex/cv/models/utils/det_metrics/coco_utils.py +0 -217
- paddlex/cv/models/utils/det_metrics/metrics.py +0 -220
- paddlex/cv/models/utils/ema.py +0 -48
- paddlex/cv/models/utils/seg_metrics.py +0 -62
- paddlex/cv/models/utils/visualize.py +0 -394
- paddlex/cv/transforms/__init__.py +0 -46
- paddlex/cv/transforms/batch_operators.py +0 -286
- paddlex/cv/transforms/box_utils.py +0 -41
- paddlex/cv/transforms/functions.py +0 -193
- paddlex/cv/transforms/operators.py +0 -1402
- paddlex/det.py +0 -43
- paddlex/paddleseg/__init__.py +0 -17
- paddlex/paddleseg/core/__init__.py +0 -20
- paddlex/paddleseg/core/infer.py +0 -289
- paddlex/paddleseg/core/predict.py +0 -145
- paddlex/paddleseg/core/train.py +0 -258
- paddlex/paddleseg/core/val.py +0 -172
- paddlex/paddleseg/cvlibs/__init__.py +0 -17
- paddlex/paddleseg/cvlibs/callbacks.py +0 -279
- paddlex/paddleseg/cvlibs/config.py +0 -359
- paddlex/paddleseg/cvlibs/manager.py +0 -142
- paddlex/paddleseg/cvlibs/param_init.py +0 -91
- paddlex/paddleseg/datasets/__init__.py +0 -21
- paddlex/paddleseg/datasets/ade.py +0 -112
- paddlex/paddleseg/datasets/cityscapes.py +0 -86
- paddlex/paddleseg/datasets/cocostuff.py +0 -79
- paddlex/paddleseg/datasets/dataset.py +0 -164
- paddlex/paddleseg/datasets/mini_deep_globe_road_extraction.py +0 -95
- paddlex/paddleseg/datasets/optic_disc_seg.py +0 -97
- paddlex/paddleseg/datasets/pascal_context.py +0 -80
- paddlex/paddleseg/datasets/voc.py +0 -113
- paddlex/paddleseg/models/__init__.py +0 -39
- paddlex/paddleseg/models/ann.py +0 -436
- paddlex/paddleseg/models/attention_unet.py +0 -189
- paddlex/paddleseg/models/backbones/__init__.py +0 -18
- paddlex/paddleseg/models/backbones/hrnet.py +0 -815
- paddlex/paddleseg/models/backbones/mobilenetv3.py +0 -365
- paddlex/paddleseg/models/backbones/resnet_vd.py +0 -364
- paddlex/paddleseg/models/backbones/xception_deeplab.py +0 -415
- paddlex/paddleseg/models/bisenet.py +0 -311
- paddlex/paddleseg/models/danet.py +0 -220
- paddlex/paddleseg/models/decoupled_segnet.py +0 -233
- paddlex/paddleseg/models/deeplab.py +0 -258
- paddlex/paddleseg/models/dnlnet.py +0 -231
- paddlex/paddleseg/models/emanet.py +0 -219
- paddlex/paddleseg/models/fast_scnn.py +0 -318
- paddlex/paddleseg/models/fcn.py +0 -135
- paddlex/paddleseg/models/gcnet.py +0 -223
- paddlex/paddleseg/models/gscnn.py +0 -357
- paddlex/paddleseg/models/hardnet.py +0 -309
- paddlex/paddleseg/models/isanet.py +0 -202
- paddlex/paddleseg/models/layers/__init__.py +0 -19
- paddlex/paddleseg/models/layers/activation.py +0 -73
- paddlex/paddleseg/models/layers/attention.py +0 -146
- paddlex/paddleseg/models/layers/layer_libs.py +0 -168
- paddlex/paddleseg/models/layers/nonlocal2d.py +0 -155
- paddlex/paddleseg/models/layers/pyramid_pool.py +0 -182
- paddlex/paddleseg/models/losses/__init__.py +0 -27
- paddlex/paddleseg/models/losses/binary_cross_entropy_loss.py +0 -174
- paddlex/paddleseg/models/losses/bootstrapped_cross_entropy.py +0 -73
- paddlex/paddleseg/models/losses/cross_entropy_loss.py +0 -94
- paddlex/paddleseg/models/losses/decoupledsegnet_relax_boundary_loss.py +0 -129
- paddlex/paddleseg/models/losses/dice_loss.py +0 -61
- paddlex/paddleseg/models/losses/edge_attention_loss.py +0 -78
- paddlex/paddleseg/models/losses/gscnn_dual_task_loss.py +0 -141
- paddlex/paddleseg/models/losses/l1_loss.py +0 -76
- paddlex/paddleseg/models/losses/lovasz_loss.py +0 -222
- paddlex/paddleseg/models/losses/mean_square_error_loss.py +0 -65
- paddlex/paddleseg/models/losses/mixed_loss.py +0 -58
- paddlex/paddleseg/models/losses/ohem_cross_entropy_loss.py +0 -99
- paddlex/paddleseg/models/losses/ohem_edge_attention_loss.py +0 -114
- paddlex/paddleseg/models/ocrnet.py +0 -248
- paddlex/paddleseg/models/pspnet.py +0 -147
- paddlex/paddleseg/models/sfnet.py +0 -236
- paddlex/paddleseg/models/shufflenet_slim.py +0 -268
- paddlex/paddleseg/models/u2net.py +0 -574
- paddlex/paddleseg/models/unet.py +0 -155
- paddlex/paddleseg/models/unet_3plus.py +0 -316
- paddlex/paddleseg/models/unet_plusplus.py +0 -237
- paddlex/paddleseg/transforms/__init__.py +0 -16
- paddlex/paddleseg/transforms/functional.py +0 -161
- paddlex/paddleseg/transforms/transforms.py +0 -937
- paddlex/paddleseg/utils/__init__.py +0 -22
- paddlex/paddleseg/utils/config_check.py +0 -60
- paddlex/paddleseg/utils/download.py +0 -163
- paddlex/paddleseg/utils/env/__init__.py +0 -16
- paddlex/paddleseg/utils/env/seg_env.py +0 -56
- paddlex/paddleseg/utils/env/sys_env.py +0 -122
- paddlex/paddleseg/utils/logger.py +0 -48
- paddlex/paddleseg/utils/metrics.py +0 -146
- paddlex/paddleseg/utils/progbar.py +0 -212
- paddlex/paddleseg/utils/timer.py +0 -53
- paddlex/paddleseg/utils/utils.py +0 -120
- paddlex/paddleseg/utils/visualize.py +0 -90
- paddlex/ppcls/__init__.py +0 -20
- paddlex/ppcls/data/__init__.py +0 -15
- paddlex/ppcls/data/imaug/__init__.py +0 -94
- paddlex/ppcls/data/imaug/autoaugment.py +0 -264
- paddlex/ppcls/data/imaug/batch_operators.py +0 -117
- paddlex/ppcls/data/imaug/cutout.py +0 -41
- paddlex/ppcls/data/imaug/fmix.py +0 -217
- paddlex/ppcls/data/imaug/grid.py +0 -89
- paddlex/ppcls/data/imaug/hide_and_seek.py +0 -44
- paddlex/ppcls/data/imaug/operators.py +0 -256
- paddlex/ppcls/data/imaug/randaugment.py +0 -106
- paddlex/ppcls/data/imaug/random_erasing.py +0 -55
- paddlex/ppcls/data/reader.py +0 -318
- paddlex/ppcls/modeling/__init__.py +0 -20
- paddlex/ppcls/modeling/architectures/__init__.py +0 -51
- paddlex/ppcls/modeling/architectures/alexnet.py +0 -132
- paddlex/ppcls/modeling/architectures/darknet.py +0 -161
- paddlex/ppcls/modeling/architectures/densenet.py +0 -308
- paddlex/ppcls/modeling/architectures/distillation_models.py +0 -65
- paddlex/ppcls/modeling/architectures/distilled_vision_transformer.py +0 -196
- paddlex/ppcls/modeling/architectures/dpn.py +0 -425
- paddlex/ppcls/modeling/architectures/efficientnet.py +0 -901
- paddlex/ppcls/modeling/architectures/ghostnet.py +0 -331
- paddlex/ppcls/modeling/architectures/googlenet.py +0 -207
- paddlex/ppcls/modeling/architectures/hrnet.py +0 -742
- paddlex/ppcls/modeling/architectures/inception_v3.py +0 -541
- paddlex/ppcls/modeling/architectures/inception_v4.py +0 -455
- paddlex/ppcls/modeling/architectures/mixnet.py +0 -782
- paddlex/ppcls/modeling/architectures/mobilenet_v1.py +0 -266
- paddlex/ppcls/modeling/architectures/mobilenet_v2.py +0 -248
- paddlex/ppcls/modeling/architectures/mobilenet_v3.py +0 -359
- paddlex/ppcls/modeling/architectures/regnet.py +0 -383
- paddlex/ppcls/modeling/architectures/repvgg.py +0 -339
- paddlex/ppcls/modeling/architectures/res2net.py +0 -272
- paddlex/ppcls/modeling/architectures/res2net_vd.py +0 -295
- paddlex/ppcls/modeling/architectures/resnest.py +0 -705
- paddlex/ppcls/modeling/architectures/resnet.py +0 -317
- paddlex/ppcls/modeling/architectures/resnet_vc.py +0 -309
- paddlex/ppcls/modeling/architectures/resnet_vd.py +0 -354
- paddlex/ppcls/modeling/architectures/resnext.py +0 -259
- paddlex/ppcls/modeling/architectures/resnext101_wsl.py +0 -447
- paddlex/ppcls/modeling/architectures/resnext_vd.py +0 -266
- paddlex/ppcls/modeling/architectures/rexnet.py +0 -240
- paddlex/ppcls/modeling/architectures/se_resnet_vd.py +0 -378
- paddlex/ppcls/modeling/architectures/se_resnext.py +0 -290
- paddlex/ppcls/modeling/architectures/se_resnext_vd.py +0 -285
- paddlex/ppcls/modeling/architectures/shufflenet_v2.py +0 -320
- paddlex/ppcls/modeling/architectures/squeezenet.py +0 -154
- paddlex/ppcls/modeling/architectures/vgg.py +0 -152
- paddlex/ppcls/modeling/architectures/vision_transformer.py +0 -402
- paddlex/ppcls/modeling/architectures/xception.py +0 -345
- paddlex/ppcls/modeling/architectures/xception_deeplab.py +0 -386
- paddlex/ppcls/modeling/loss.py +0 -158
- paddlex/ppcls/modeling/utils.py +0 -53
- paddlex/ppcls/optimizer/__init__.py +0 -19
- paddlex/ppcls/optimizer/learning_rate.py +0 -159
- paddlex/ppcls/optimizer/optimizer.py +0 -165
- paddlex/ppcls/utils/__init__.py +0 -27
- paddlex/ppcls/utils/check.py +0 -151
- paddlex/ppcls/utils/config.py +0 -201
- paddlex/ppcls/utils/logger.py +0 -120
- paddlex/ppcls/utils/metrics.py +0 -112
- paddlex/ppcls/utils/misc.py +0 -62
- paddlex/ppcls/utils/model_zoo.py +0 -213
- paddlex/ppcls/utils/save_load.py +0 -163
- paddlex/ppdet/__init__.py +0 -16
- paddlex/ppdet/core/__init__.py +0 -15
- paddlex/ppdet/core/config/__init__.py +0 -13
- paddlex/ppdet/core/config/schema.py +0 -248
- paddlex/ppdet/core/config/yaml_helpers.py +0 -118
- paddlex/ppdet/core/workspace.py +0 -279
- paddlex/ppdet/data/__init__.py +0 -21
- paddlex/ppdet/data/reader.py +0 -304
- paddlex/ppdet/data/shm_utils.py +0 -67
- paddlex/ppdet/data/source/__init__.py +0 -27
- paddlex/ppdet/data/source/category.py +0 -823
- paddlex/ppdet/data/source/coco.py +0 -243
- paddlex/ppdet/data/source/dataset.py +0 -192
- paddlex/ppdet/data/source/keypoint_coco.py +0 -656
- paddlex/ppdet/data/source/mot.py +0 -360
- paddlex/ppdet/data/source/voc.py +0 -204
- paddlex/ppdet/data/source/widerface.py +0 -180
- paddlex/ppdet/data/transform/__init__.py +0 -28
- paddlex/ppdet/data/transform/autoaugment_utils.py +0 -1593
- paddlex/ppdet/data/transform/batch_operators.py +0 -758
- paddlex/ppdet/data/transform/gridmask_utils.py +0 -83
- paddlex/ppdet/data/transform/keypoint_operators.py +0 -665
- paddlex/ppdet/data/transform/mot_operators.py +0 -636
- paddlex/ppdet/data/transform/op_helper.py +0 -468
- paddlex/ppdet/data/transform/operators.py +0 -2103
- paddlex/ppdet/engine/__init__.py +0 -29
- paddlex/ppdet/engine/callbacks.py +0 -262
- paddlex/ppdet/engine/env.py +0 -47
- paddlex/ppdet/engine/export_utils.py +0 -118
- paddlex/ppdet/engine/tracker.py +0 -425
- paddlex/ppdet/engine/trainer.py +0 -535
- paddlex/ppdet/metrics/__init__.py +0 -23
- paddlex/ppdet/metrics/coco_utils.py +0 -184
- paddlex/ppdet/metrics/json_results.py +0 -151
- paddlex/ppdet/metrics/keypoint_metrics.py +0 -202
- paddlex/ppdet/metrics/map_utils.py +0 -396
- paddlex/ppdet/metrics/metrics.py +0 -300
- paddlex/ppdet/metrics/mot_eval_utils.py +0 -192
- paddlex/ppdet/metrics/mot_metrics.py +0 -184
- paddlex/ppdet/metrics/widerface_utils.py +0 -393
- paddlex/ppdet/model_zoo/__init__.py +0 -18
- paddlex/ppdet/model_zoo/model_zoo.py +0 -86
- paddlex/ppdet/model_zoo/tests/__init__.py +0 -13
- paddlex/ppdet/model_zoo/tests/test_get_model.py +0 -48
- paddlex/ppdet/model_zoo/tests/test_list_model.py +0 -68
- paddlex/ppdet/modeling/__init__.py +0 -41
- paddlex/ppdet/modeling/architectures/__init__.py +0 -40
- paddlex/ppdet/modeling/architectures/cascade_rcnn.py +0 -144
- paddlex/ppdet/modeling/architectures/centernet.py +0 -103
- paddlex/ppdet/modeling/architectures/deepsort.py +0 -111
- paddlex/ppdet/modeling/architectures/fairmot.py +0 -107
- paddlex/ppdet/modeling/architectures/faster_rcnn.py +0 -106
- paddlex/ppdet/modeling/architectures/fcos.py +0 -105
- paddlex/ppdet/modeling/architectures/jde.py +0 -125
- paddlex/ppdet/modeling/architectures/keypoint_hrhrnet.py +0 -286
- paddlex/ppdet/modeling/architectures/keypoint_hrnet.py +0 -203
- paddlex/ppdet/modeling/architectures/mask_rcnn.py +0 -135
- paddlex/ppdet/modeling/architectures/meta_arch.py +0 -45
- paddlex/ppdet/modeling/architectures/s2anet.py +0 -103
- paddlex/ppdet/modeling/architectures/solov2.py +0 -110
- paddlex/ppdet/modeling/architectures/ssd.py +0 -84
- paddlex/ppdet/modeling/architectures/ttfnet.py +0 -98
- paddlex/ppdet/modeling/architectures/yolo.py +0 -104
- paddlex/ppdet/modeling/backbones/__init__.py +0 -37
- paddlex/ppdet/modeling/backbones/blazenet.py +0 -322
- paddlex/ppdet/modeling/backbones/darknet.py +0 -341
- paddlex/ppdet/modeling/backbones/dla.py +0 -244
- paddlex/ppdet/modeling/backbones/ghostnet.py +0 -476
- paddlex/ppdet/modeling/backbones/hrnet.py +0 -724
- paddlex/ppdet/modeling/backbones/mobilenet_v1.py +0 -410
- paddlex/ppdet/modeling/backbones/mobilenet_v3.py +0 -497
- paddlex/ppdet/modeling/backbones/name_adapter.py +0 -69
- paddlex/ppdet/modeling/backbones/res2net.py +0 -358
- paddlex/ppdet/modeling/backbones/resnet.py +0 -606
- paddlex/ppdet/modeling/backbones/senet.py +0 -140
- paddlex/ppdet/modeling/backbones/vgg.py +0 -216
- paddlex/ppdet/modeling/bbox_utils.py +0 -464
- paddlex/ppdet/modeling/heads/__init__.py +0 -41
- paddlex/ppdet/modeling/heads/bbox_head.py +0 -379
- paddlex/ppdet/modeling/heads/cascade_head.py +0 -285
- paddlex/ppdet/modeling/heads/centernet_head.py +0 -194
- paddlex/ppdet/modeling/heads/face_head.py +0 -113
- paddlex/ppdet/modeling/heads/fcos_head.py +0 -270
- paddlex/ppdet/modeling/heads/keypoint_hrhrnet_head.py +0 -108
- paddlex/ppdet/modeling/heads/mask_head.py +0 -253
- paddlex/ppdet/modeling/heads/roi_extractor.py +0 -111
- paddlex/ppdet/modeling/heads/s2anet_head.py +0 -845
- paddlex/ppdet/modeling/heads/solov2_head.py +0 -537
- paddlex/ppdet/modeling/heads/ssd_head.py +0 -175
- paddlex/ppdet/modeling/heads/ttf_head.py +0 -314
- paddlex/ppdet/modeling/heads/yolo_head.py +0 -124
- paddlex/ppdet/modeling/keypoint_utils.py +0 -302
- paddlex/ppdet/modeling/layers.py +0 -1142
- paddlex/ppdet/modeling/losses/__init__.py +0 -35
- paddlex/ppdet/modeling/losses/ctfocal_loss.py +0 -67
- paddlex/ppdet/modeling/losses/fairmot_loss.py +0 -41
- paddlex/ppdet/modeling/losses/fcos_loss.py +0 -225
- paddlex/ppdet/modeling/losses/iou_aware_loss.py +0 -48
- paddlex/ppdet/modeling/losses/iou_loss.py +0 -210
- paddlex/ppdet/modeling/losses/jde_loss.py +0 -182
- paddlex/ppdet/modeling/losses/keypoint_loss.py +0 -228
- paddlex/ppdet/modeling/losses/solov2_loss.py +0 -101
- paddlex/ppdet/modeling/losses/ssd_loss.py +0 -163
- paddlex/ppdet/modeling/losses/yolo_loss.py +0 -212
- paddlex/ppdet/modeling/mot/__init__.py +0 -25
- paddlex/ppdet/modeling/mot/matching/__init__.py +0 -19
- paddlex/ppdet/modeling/mot/matching/deepsort_matching.py +0 -382
- paddlex/ppdet/modeling/mot/matching/jde_matching.py +0 -145
- paddlex/ppdet/modeling/mot/motion/__init__.py +0 -17
- paddlex/ppdet/modeling/mot/motion/kalman_filter.py +0 -270
- paddlex/ppdet/modeling/mot/tracker/__init__.py +0 -23
- paddlex/ppdet/modeling/mot/tracker/base_jde_tracker.py +0 -267
- paddlex/ppdet/modeling/mot/tracker/base_sde_tracker.py +0 -145
- paddlex/ppdet/modeling/mot/tracker/deepsort_tracker.py +0 -165
- paddlex/ppdet/modeling/mot/tracker/jde_tracker.py +0 -262
- paddlex/ppdet/modeling/mot/utils.py +0 -181
- paddlex/ppdet/modeling/mot/visualization.py +0 -130
- paddlex/ppdet/modeling/necks/__init__.py +0 -25
- paddlex/ppdet/modeling/necks/centernet_fpn.py +0 -185
- paddlex/ppdet/modeling/necks/fpn.py +0 -233
- paddlex/ppdet/modeling/necks/hrfpn.py +0 -131
- paddlex/ppdet/modeling/necks/ttf_fpn.py +0 -243
- paddlex/ppdet/modeling/necks/yolo_fpn.py +0 -1034
- paddlex/ppdet/modeling/ops.py +0 -1599
- paddlex/ppdet/modeling/post_process.py +0 -449
- paddlex/ppdet/modeling/proposal_generator/__init__.py +0 -2
- paddlex/ppdet/modeling/proposal_generator/anchor_generator.py +0 -135
- paddlex/ppdet/modeling/proposal_generator/proposal_generator.py +0 -81
- paddlex/ppdet/modeling/proposal_generator/rpn_head.py +0 -269
- paddlex/ppdet/modeling/proposal_generator/target.py +0 -671
- paddlex/ppdet/modeling/proposal_generator/target_layer.py +0 -476
- paddlex/ppdet/modeling/reid/__init__.py +0 -23
- paddlex/ppdet/modeling/reid/fairmot_embedding_head.py +0 -117
- paddlex/ppdet/modeling/reid/jde_embedding_head.py +0 -189
- paddlex/ppdet/modeling/reid/pyramidal_embedding.py +0 -151
- paddlex/ppdet/modeling/reid/resnet.py +0 -320
- paddlex/ppdet/modeling/shape_spec.py +0 -33
- paddlex/ppdet/modeling/tests/__init__.py +0 -13
- paddlex/ppdet/modeling/tests/test_architectures.py +0 -59
- paddlex/ppdet/modeling/tests/test_base.py +0 -75
- paddlex/ppdet/modeling/tests/test_ops.py +0 -839
- paddlex/ppdet/modeling/tests/test_yolov3_loss.py +0 -420
- paddlex/ppdet/optimizer.py +0 -285
- paddlex/ppdet/slim/__init__.py +0 -62
- paddlex/ppdet/slim/distill.py +0 -111
- paddlex/ppdet/slim/prune.py +0 -85
- paddlex/ppdet/slim/quant.py +0 -52
- paddlex/ppdet/utils/__init__.py +0 -13
- paddlex/ppdet/utils/check.py +0 -93
- paddlex/ppdet/utils/checkpoint.py +0 -216
- paddlex/ppdet/utils/cli.py +0 -151
- paddlex/ppdet/utils/colormap.py +0 -56
- paddlex/ppdet/utils/download.py +0 -477
- paddlex/ppdet/utils/logger.py +0 -71
- paddlex/ppdet/utils/stats.py +0 -95
- paddlex/ppdet/utils/visualizer.py +0 -292
- paddlex/ppdet/utils/voc_utils.py +0 -87
- paddlex/seg.py +0 -38
- paddlex/tools/__init__.py +0 -16
- paddlex/tools/convert.py +0 -52
- paddlex/tools/dataset_conversion/__init__.py +0 -24
- paddlex/tools/dataset_conversion/x2coco.py +0 -379
- paddlex/tools/dataset_conversion/x2imagenet.py +0 -82
- paddlex/tools/dataset_conversion/x2seg.py +0 -343
- paddlex/tools/dataset_conversion/x2voc.py +0 -230
- paddlex/tools/dataset_split/__init__.py +0 -23
- paddlex/tools/dataset_split/coco_split.py +0 -69
- paddlex/tools/dataset_split/imagenet_split.py +0 -75
- paddlex/tools/dataset_split/seg_split.py +0 -96
- paddlex/tools/dataset_split/utils.py +0 -75
- paddlex/tools/dataset_split/voc_split.py +0 -91
- paddlex/tools/split.py +0 -41
- paddlex/utils/checkpoint.py +0 -439
- paddlex/utils/shm.py +0 -67
- paddlex/utils/stats.py +0 -68
- paddlex/utils/utils.py +0 -140
- paddlex-2.0.0rc4.dist-info/LICENSE +0 -201
- paddlex-2.0.0rc4.dist-info/METADATA +0 -29
- paddlex-2.0.0rc4.dist-info/RECORD +0 -445
- paddlex-2.0.0rc4.dist-info/WHEEL +0 -5
- paddlex-2.0.0rc4.dist-info/entry_points.txt +0 -3
- paddlex-2.0.0rc4.dist-info/top_level.txt +0 -2
@@ -0,0 +1,1382 @@
|
|
1
|
+
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
from __future__ import annotations
|
15
|
+
|
16
|
+
import copy
|
17
|
+
import re
|
18
|
+
from typing import Any, Dict, List, Optional, Tuple, Union
|
19
|
+
|
20
|
+
import numpy as np
|
21
|
+
from PIL import Image
|
22
|
+
|
23
|
+
from ....utils import logging
|
24
|
+
from ....utils.deps import pipeline_requires_extra
|
25
|
+
from ...common.batch_sampler import ImageBatchSampler
|
26
|
+
from ...common.reader import ReadImage
|
27
|
+
from ...models.object_detection.result import DetResult
|
28
|
+
from ...utils.hpi import HPIConfig
|
29
|
+
from ...utils.pp_option import PaddlePredictorOption
|
30
|
+
from .._parallel import AutoParallelImageSimpleInferencePipeline
|
31
|
+
from ..base import BasePipeline
|
32
|
+
from ..ocr.result import OCRResult
|
33
|
+
from .result_v2 import LayoutParsingBlock, LayoutParsingRegion, LayoutParsingResultV2
|
34
|
+
from .setting import BLOCK_LABEL_MAP, BLOCK_SETTINGS, LINE_SETTINGS, REGION_SETTINGS
|
35
|
+
from .utils import (
|
36
|
+
caculate_bbox_area,
|
37
|
+
calculate_minimum_enclosing_bbox,
|
38
|
+
calculate_overlap_ratio,
|
39
|
+
convert_formula_res_to_ocr_format,
|
40
|
+
format_line,
|
41
|
+
gather_imgs,
|
42
|
+
get_bbox_intersection,
|
43
|
+
get_sub_regions_ocr_res,
|
44
|
+
group_boxes_into_lines,
|
45
|
+
remove_overlap_blocks,
|
46
|
+
shrink_supplement_region_bbox,
|
47
|
+
split_boxes_by_projection,
|
48
|
+
update_region_box,
|
49
|
+
)
|
50
|
+
|
51
|
+
|
52
|
+
class _LayoutParsingPipelineV2(BasePipeline):
|
53
|
+
"""Layout Parsing Pipeline V2"""
|
54
|
+
|
55
|
+
def __init__(
|
56
|
+
self,
|
57
|
+
config: dict,
|
58
|
+
device: str = None,
|
59
|
+
pp_option: PaddlePredictorOption = None,
|
60
|
+
use_hpip: bool = False,
|
61
|
+
hpi_config: Optional[Union[Dict[str, Any], HPIConfig]] = None,
|
62
|
+
) -> None:
|
63
|
+
"""Initializes the layout parsing pipeline.
|
64
|
+
|
65
|
+
Args:
|
66
|
+
config (Dict): Configuration dictionary containing various settings.
|
67
|
+
device (str, optional): Device to run the predictions on. Defaults to None.
|
68
|
+
pp_option (PaddlePredictorOption, optional): PaddlePredictor options. Defaults to None.
|
69
|
+
use_hpip (bool, optional): Whether to use the high-performance
|
70
|
+
inference plugin (HPIP) by default. Defaults to False.
|
71
|
+
hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional):
|
72
|
+
The default high-performance inference configuration dictionary.
|
73
|
+
Defaults to None.
|
74
|
+
"""
|
75
|
+
|
76
|
+
super().__init__(
|
77
|
+
device=device,
|
78
|
+
pp_option=pp_option,
|
79
|
+
use_hpip=use_hpip,
|
80
|
+
hpi_config=hpi_config,
|
81
|
+
)
|
82
|
+
|
83
|
+
self.inintial_predictor(config)
|
84
|
+
|
85
|
+
self.batch_sampler = ImageBatchSampler(batch_size=config.get("batch_size", 1))
|
86
|
+
self.img_reader = ReadImage(format="BGR")
|
87
|
+
|
88
|
+
def inintial_predictor(self, config: dict) -> None:
|
89
|
+
"""Initializes the predictor based on the provided configuration.
|
90
|
+
|
91
|
+
Args:
|
92
|
+
config (Dict): A dictionary containing the configuration for the predictor.
|
93
|
+
|
94
|
+
Returns:
|
95
|
+
None
|
96
|
+
"""
|
97
|
+
|
98
|
+
self.use_doc_preprocessor = config.get("use_doc_preprocessor", True)
|
99
|
+
self.use_table_recognition = config.get("use_table_recognition", True)
|
100
|
+
self.use_seal_recognition = config.get("use_seal_recognition", True)
|
101
|
+
self.use_region_detection = config.get(
|
102
|
+
"use_region_detection",
|
103
|
+
True,
|
104
|
+
)
|
105
|
+
self.use_formula_recognition = config.get(
|
106
|
+
"use_formula_recognition",
|
107
|
+
True,
|
108
|
+
)
|
109
|
+
self.use_chart_recognition = config.get(
|
110
|
+
"use_chart_recognition",
|
111
|
+
False,
|
112
|
+
)
|
113
|
+
|
114
|
+
if self.use_doc_preprocessor:
|
115
|
+
doc_preprocessor_config = config.get("SubPipelines", {}).get(
|
116
|
+
"DocPreprocessor",
|
117
|
+
{
|
118
|
+
"pipeline_config_error": "config error for doc_preprocessor_pipeline!",
|
119
|
+
},
|
120
|
+
)
|
121
|
+
self.doc_preprocessor_pipeline = self.create_pipeline(
|
122
|
+
doc_preprocessor_config,
|
123
|
+
)
|
124
|
+
if self.use_region_detection:
|
125
|
+
region_detection_config = config.get("SubModules", {}).get(
|
126
|
+
"RegionDetection",
|
127
|
+
{
|
128
|
+
"model_config_error": "config error for block_region_detection_model!"
|
129
|
+
},
|
130
|
+
)
|
131
|
+
self.region_detection_model = self.create_model(
|
132
|
+
region_detection_config,
|
133
|
+
)
|
134
|
+
|
135
|
+
layout_det_config = config.get("SubModules", {}).get(
|
136
|
+
"LayoutDetection",
|
137
|
+
{"model_config_error": "config error for layout_det_model!"},
|
138
|
+
)
|
139
|
+
layout_kwargs = {}
|
140
|
+
if (threshold := layout_det_config.get("threshold", None)) is not None:
|
141
|
+
layout_kwargs["threshold"] = threshold
|
142
|
+
if (layout_nms := layout_det_config.get("layout_nms", None)) is not None:
|
143
|
+
layout_kwargs["layout_nms"] = layout_nms
|
144
|
+
if (
|
145
|
+
layout_unclip_ratio := layout_det_config.get("layout_unclip_ratio", None)
|
146
|
+
) is not None:
|
147
|
+
layout_kwargs["layout_unclip_ratio"] = layout_unclip_ratio
|
148
|
+
if (
|
149
|
+
layout_merge_bboxes_mode := layout_det_config.get(
|
150
|
+
"layout_merge_bboxes_mode", None
|
151
|
+
)
|
152
|
+
) is not None:
|
153
|
+
layout_kwargs["layout_merge_bboxes_mode"] = layout_merge_bboxes_mode
|
154
|
+
self.layout_det_model = self.create_model(layout_det_config, **layout_kwargs)
|
155
|
+
|
156
|
+
general_ocr_config = config.get("SubPipelines", {}).get(
|
157
|
+
"GeneralOCR",
|
158
|
+
{"pipeline_config_error": "config error for general_ocr_pipeline!"},
|
159
|
+
)
|
160
|
+
self.general_ocr_pipeline = self.create_pipeline(
|
161
|
+
general_ocr_config,
|
162
|
+
)
|
163
|
+
|
164
|
+
if self.use_seal_recognition:
|
165
|
+
seal_recognition_config = config.get("SubPipelines", {}).get(
|
166
|
+
"SealRecognition",
|
167
|
+
{
|
168
|
+
"pipeline_config_error": "config error for seal_recognition_pipeline!",
|
169
|
+
},
|
170
|
+
)
|
171
|
+
self.seal_recognition_pipeline = self.create_pipeline(
|
172
|
+
seal_recognition_config,
|
173
|
+
)
|
174
|
+
|
175
|
+
if self.use_table_recognition:
|
176
|
+
table_recognition_config = config.get("SubPipelines", {}).get(
|
177
|
+
"TableRecognition",
|
178
|
+
{
|
179
|
+
"pipeline_config_error": "config error for table_recognition_pipeline!",
|
180
|
+
},
|
181
|
+
)
|
182
|
+
self.table_recognition_pipeline = self.create_pipeline(
|
183
|
+
table_recognition_config,
|
184
|
+
)
|
185
|
+
|
186
|
+
if self.use_formula_recognition:
|
187
|
+
formula_recognition_config = config.get("SubPipelines", {}).get(
|
188
|
+
"FormulaRecognition",
|
189
|
+
{
|
190
|
+
"pipeline_config_error": "config error for formula_recognition_pipeline!",
|
191
|
+
},
|
192
|
+
)
|
193
|
+
self.formula_recognition_pipeline = self.create_pipeline(
|
194
|
+
formula_recognition_config,
|
195
|
+
)
|
196
|
+
|
197
|
+
if self.use_chart_recognition:
|
198
|
+
chart_recognition_config = config.get("SubModules", {}).get(
|
199
|
+
"ChartRecognition",
|
200
|
+
{
|
201
|
+
"model_config_error": "config error for block_region_detection_model!"
|
202
|
+
},
|
203
|
+
)
|
204
|
+
self.chart_recognition_model = self.create_model(
|
205
|
+
chart_recognition_config,
|
206
|
+
)
|
207
|
+
|
208
|
+
return
|
209
|
+
|
210
|
+
def get_text_paragraphs_ocr_res(
|
211
|
+
self,
|
212
|
+
overall_ocr_res: OCRResult,
|
213
|
+
layout_det_res: DetResult,
|
214
|
+
) -> OCRResult:
|
215
|
+
"""
|
216
|
+
Retrieves the OCR results for text paragraphs, excluding those of formulas, tables, and seals.
|
217
|
+
|
218
|
+
Args:
|
219
|
+
overall_ocr_res (OCRResult): The overall OCR result containing text information.
|
220
|
+
layout_det_res (DetResult): The detection result containing the layout information of the document.
|
221
|
+
|
222
|
+
Returns:
|
223
|
+
OCRResult: The OCR result for text paragraphs after excluding formulas, tables, and seals.
|
224
|
+
"""
|
225
|
+
object_boxes = []
|
226
|
+
for box_info in layout_det_res["boxes"]:
|
227
|
+
if box_info["label"].lower() in ["formula", "table", "seal"]:
|
228
|
+
object_boxes.append(box_info["coordinate"])
|
229
|
+
object_boxes = np.array(object_boxes)
|
230
|
+
sub_regions_ocr_res = get_sub_regions_ocr_res(
|
231
|
+
overall_ocr_res, object_boxes, flag_within=False
|
232
|
+
)
|
233
|
+
return sub_regions_ocr_res
|
234
|
+
|
235
|
+
def check_model_settings_valid(self, input_params: dict) -> bool:
|
236
|
+
"""
|
237
|
+
Check if the input parameters are valid based on the initialized models.
|
238
|
+
|
239
|
+
Args:
|
240
|
+
input_params (Dict): A dictionary containing input parameters.
|
241
|
+
|
242
|
+
Returns:
|
243
|
+
bool: True if all required models are initialized according to input parameters, False otherwise.
|
244
|
+
"""
|
245
|
+
|
246
|
+
if input_params["use_doc_preprocessor"] and not self.use_doc_preprocessor:
|
247
|
+
logging.error(
|
248
|
+
"Set use_doc_preprocessor, but the models for doc preprocessor are not initialized.",
|
249
|
+
)
|
250
|
+
return False
|
251
|
+
|
252
|
+
if input_params["use_seal_recognition"] and not self.use_seal_recognition:
|
253
|
+
logging.error(
|
254
|
+
"Set use_seal_recognition, but the models for seal recognition are not initialized.",
|
255
|
+
)
|
256
|
+
return False
|
257
|
+
|
258
|
+
if input_params["use_table_recognition"] and not self.use_table_recognition:
|
259
|
+
logging.error(
|
260
|
+
"Set use_table_recognition, but the models for table recognition are not initialized.",
|
261
|
+
)
|
262
|
+
return False
|
263
|
+
|
264
|
+
return True
|
265
|
+
|
266
|
+
def standardized_data(
|
267
|
+
self,
|
268
|
+
image: list,
|
269
|
+
region_det_res: DetResult,
|
270
|
+
layout_det_res: DetResult,
|
271
|
+
overall_ocr_res: OCRResult,
|
272
|
+
formula_res_list: list,
|
273
|
+
text_rec_model: Any,
|
274
|
+
text_rec_score_thresh: Union[float, None] = None,
|
275
|
+
) -> list:
|
276
|
+
"""
|
277
|
+
Retrieves the layout parsing result based on the layout detection result, OCR result, and other recognition results.
|
278
|
+
Args:
|
279
|
+
image (list): The input image.
|
280
|
+
overall_ocr_res (OCRResult): An object containing the overall OCR results, including detected text boxes and recognized text. The structure is expected to have:
|
281
|
+
- "input_img": The image on which OCR was performed.
|
282
|
+
- "dt_boxes": A list of detected text box coordinates.
|
283
|
+
- "rec_texts": A list of recognized text corresponding to the detected boxes.
|
284
|
+
|
285
|
+
layout_det_res (DetResult): An object containing the layout detection results, including detected layout boxes and their labels. The structure is expected to have:
|
286
|
+
- "boxes": A list of dictionaries with keys "coordinate" for box coordinates and "block_label" for the type of content.
|
287
|
+
|
288
|
+
table_res_list (list): A list of table detection results, where each item is a dictionary containing:
|
289
|
+
- "block_bbox": The bounding box of the table layout.
|
290
|
+
- "pred_html": The predicted HTML representation of the table.
|
291
|
+
|
292
|
+
formula_res_list (list): A list of formula recognition results.
|
293
|
+
text_rec_model (Any): The text recognition model.
|
294
|
+
text_rec_score_thresh (Optional[float], optional): The score threshold for text recognition. Defaults to None.
|
295
|
+
Returns:
|
296
|
+
list: A list of dictionaries representing the layout parsing result.
|
297
|
+
"""
|
298
|
+
|
299
|
+
matched_ocr_dict = {}
|
300
|
+
region_to_block_map = {}
|
301
|
+
block_to_ocr_map = {}
|
302
|
+
object_boxes = []
|
303
|
+
footnote_list = []
|
304
|
+
paragraph_title_list = []
|
305
|
+
bottom_text_y_max = 0
|
306
|
+
max_block_area = 0.0
|
307
|
+
doc_title_num = 0
|
308
|
+
|
309
|
+
base_region_bbox = [65535, 65535, 0, 0]
|
310
|
+
layout_det_res = remove_overlap_blocks(
|
311
|
+
layout_det_res,
|
312
|
+
threshold=0.5,
|
313
|
+
smaller=True,
|
314
|
+
)
|
315
|
+
|
316
|
+
# convert formula_res_list to OCRResult format
|
317
|
+
convert_formula_res_to_ocr_format(formula_res_list, overall_ocr_res)
|
318
|
+
|
319
|
+
# match layout boxes and ocr boxes and get some information for layout_order_config
|
320
|
+
for box_idx, box_info in enumerate(layout_det_res["boxes"]):
|
321
|
+
box = box_info["coordinate"]
|
322
|
+
label = box_info["label"].lower()
|
323
|
+
object_boxes.append(box)
|
324
|
+
_, _, _, y2 = box
|
325
|
+
|
326
|
+
# update the region box and max_block_area according to the layout boxes
|
327
|
+
base_region_bbox = update_region_box(box, base_region_bbox)
|
328
|
+
max_block_area = max(max_block_area, caculate_bbox_area(box))
|
329
|
+
|
330
|
+
# update_layout_order_config_block_index(layout_order_config, label, box_idx)
|
331
|
+
|
332
|
+
# set the label of footnote to text, when it is above the text boxes
|
333
|
+
if label == "footnote":
|
334
|
+
footnote_list.append(box_idx)
|
335
|
+
elif label == "paragraph_title":
|
336
|
+
paragraph_title_list.append(box_idx)
|
337
|
+
if label == "text":
|
338
|
+
bottom_text_y_max = max(y2, bottom_text_y_max)
|
339
|
+
if label == "doc_title":
|
340
|
+
doc_title_num += 1
|
341
|
+
|
342
|
+
if label not in ["formula", "table", "seal"]:
|
343
|
+
_, matched_idxes = get_sub_regions_ocr_res(
|
344
|
+
overall_ocr_res, [box], return_match_idx=True
|
345
|
+
)
|
346
|
+
block_to_ocr_map[box_idx] = matched_idxes
|
347
|
+
for matched_idx in matched_idxes:
|
348
|
+
if matched_ocr_dict.get(matched_idx, None) is None:
|
349
|
+
matched_ocr_dict[matched_idx] = [box_idx]
|
350
|
+
else:
|
351
|
+
matched_ocr_dict[matched_idx].append(box_idx)
|
352
|
+
|
353
|
+
# fix the footnote label
|
354
|
+
for footnote_idx in footnote_list:
|
355
|
+
if (
|
356
|
+
layout_det_res["boxes"][footnote_idx]["coordinate"][3]
|
357
|
+
< bottom_text_y_max
|
358
|
+
):
|
359
|
+
layout_det_res["boxes"][footnote_idx]["label"] = "text"
|
360
|
+
|
361
|
+
# check if there is only one paragraph title and without doc_title
|
362
|
+
only_one_paragraph_title = len(paragraph_title_list) == 1 and doc_title_num == 0
|
363
|
+
if only_one_paragraph_title:
|
364
|
+
paragraph_title_block_area = caculate_bbox_area(
|
365
|
+
layout_det_res["boxes"][paragraph_title_list[0]]["coordinate"]
|
366
|
+
)
|
367
|
+
title_area_max_block_threshold = BLOCK_SETTINGS.get(
|
368
|
+
"title_conversion_area_ratio_threshold", 0.3
|
369
|
+
)
|
370
|
+
if (
|
371
|
+
paragraph_title_block_area
|
372
|
+
> max_block_area * title_area_max_block_threshold
|
373
|
+
):
|
374
|
+
layout_det_res["boxes"][paragraph_title_list[0]]["label"] = "doc_title"
|
375
|
+
|
376
|
+
# Replace the OCR information of the hurdles.
|
377
|
+
for overall_ocr_idx, layout_box_ids in matched_ocr_dict.items():
|
378
|
+
if len(layout_box_ids) > 1:
|
379
|
+
matched_no = 0
|
380
|
+
overall_ocr_box = copy.deepcopy(
|
381
|
+
overall_ocr_res["rec_boxes"][overall_ocr_idx]
|
382
|
+
)
|
383
|
+
overall_ocr_dt_poly = copy.deepcopy(
|
384
|
+
overall_ocr_res["dt_polys"][overall_ocr_idx]
|
385
|
+
)
|
386
|
+
for box_idx in layout_box_ids:
|
387
|
+
layout_box = layout_det_res["boxes"][box_idx]["coordinate"]
|
388
|
+
crop_box = get_bbox_intersection(overall_ocr_box, layout_box)
|
389
|
+
for ocr_idx in block_to_ocr_map[box_idx]:
|
390
|
+
ocr_box = overall_ocr_res["rec_boxes"][ocr_idx]
|
391
|
+
iou = calculate_overlap_ratio(ocr_box, crop_box, "small")
|
392
|
+
if iou > 0.8:
|
393
|
+
overall_ocr_res["rec_texts"][ocr_idx] = ""
|
394
|
+
x1, y1, x2, y2 = [int(i) for i in crop_box]
|
395
|
+
crop_img = np.array(image)[y1:y2, x1:x2]
|
396
|
+
crop_img_rec_res = list(text_rec_model([crop_img]))[0]
|
397
|
+
crop_img_dt_poly = get_bbox_intersection(
|
398
|
+
overall_ocr_dt_poly, layout_box, return_format="poly"
|
399
|
+
)
|
400
|
+
crop_img_rec_score = crop_img_rec_res["rec_score"]
|
401
|
+
crop_img_rec_text = crop_img_rec_res["rec_text"]
|
402
|
+
text_rec_score_thresh = (
|
403
|
+
text_rec_score_thresh
|
404
|
+
if text_rec_score_thresh is not None
|
405
|
+
else (self.general_ocr_pipeline.text_rec_score_thresh)
|
406
|
+
)
|
407
|
+
if crop_img_rec_score >= text_rec_score_thresh:
|
408
|
+
matched_no += 1
|
409
|
+
if matched_no == 1:
|
410
|
+
# the first matched ocr be replaced by the first matched layout box
|
411
|
+
overall_ocr_res["dt_polys"][
|
412
|
+
overall_ocr_idx
|
413
|
+
] = crop_img_dt_poly
|
414
|
+
overall_ocr_res["rec_boxes"][overall_ocr_idx] = crop_box
|
415
|
+
overall_ocr_res["rec_polys"][
|
416
|
+
overall_ocr_idx
|
417
|
+
] = crop_img_dt_poly
|
418
|
+
overall_ocr_res["rec_scores"][
|
419
|
+
overall_ocr_idx
|
420
|
+
] = crop_img_rec_score
|
421
|
+
overall_ocr_res["rec_texts"][
|
422
|
+
overall_ocr_idx
|
423
|
+
] = crop_img_rec_text
|
424
|
+
else:
|
425
|
+
# the other matched ocr be appended to the overall ocr result
|
426
|
+
overall_ocr_res["dt_polys"].append(crop_img_dt_poly)
|
427
|
+
overall_ocr_res["rec_boxes"] = np.vstack(
|
428
|
+
(overall_ocr_res["rec_boxes"], crop_box)
|
429
|
+
)
|
430
|
+
overall_ocr_res["rec_polys"].append(crop_img_dt_poly)
|
431
|
+
overall_ocr_res["rec_scores"].append(crop_img_rec_score)
|
432
|
+
overall_ocr_res["rec_texts"].append(crop_img_rec_text)
|
433
|
+
overall_ocr_res["rec_labels"].append("text")
|
434
|
+
block_to_ocr_map[box_idx].remove(overall_ocr_idx)
|
435
|
+
block_to_ocr_map[box_idx].append(
|
436
|
+
len(overall_ocr_res["rec_texts"]) - 1
|
437
|
+
)
|
438
|
+
|
439
|
+
# use layout bbox to do ocr recognition when there is no matched ocr
|
440
|
+
for layout_box_idx, overall_ocr_idxes in block_to_ocr_map.items():
|
441
|
+
has_text = False
|
442
|
+
for idx in overall_ocr_idxes:
|
443
|
+
if overall_ocr_res["rec_texts"][idx] != "":
|
444
|
+
has_text = True
|
445
|
+
break
|
446
|
+
if not has_text and layout_det_res["boxes"][layout_box_idx][
|
447
|
+
"label"
|
448
|
+
] not in BLOCK_LABEL_MAP.get("vision_labels", []):
|
449
|
+
crop_box = layout_det_res["boxes"][layout_box_idx]["coordinate"]
|
450
|
+
x1, y1, x2, y2 = [int(i) for i in crop_box]
|
451
|
+
crop_img = np.array(image)[y1:y2, x1:x2]
|
452
|
+
crop_img_rec_res = next(text_rec_model([crop_img]))
|
453
|
+
crop_img_dt_poly = get_bbox_intersection(
|
454
|
+
crop_box, crop_box, return_format="poly"
|
455
|
+
)
|
456
|
+
crop_img_rec_score = crop_img_rec_res["rec_score"]
|
457
|
+
crop_img_rec_text = crop_img_rec_res["rec_text"]
|
458
|
+
text_rec_score_thresh = (
|
459
|
+
text_rec_score_thresh
|
460
|
+
if text_rec_score_thresh is not None
|
461
|
+
else (self.general_ocr_pipeline.text_rec_score_thresh)
|
462
|
+
)
|
463
|
+
if crop_img_rec_score >= text_rec_score_thresh:
|
464
|
+
overall_ocr_res["rec_boxes"] = np.vstack(
|
465
|
+
(overall_ocr_res["rec_boxes"], crop_box)
|
466
|
+
)
|
467
|
+
overall_ocr_res["rec_polys"].append(crop_img_dt_poly)
|
468
|
+
overall_ocr_res["rec_scores"].append(crop_img_rec_score)
|
469
|
+
overall_ocr_res["rec_texts"].append(crop_img_rec_text)
|
470
|
+
overall_ocr_res["rec_labels"].append("text")
|
471
|
+
block_to_ocr_map[layout_box_idx].append(
|
472
|
+
len(overall_ocr_res["rec_texts"]) - 1
|
473
|
+
)
|
474
|
+
|
475
|
+
# when there is no layout detection result but there is ocr result, convert ocr detection result to layout detection result
|
476
|
+
if len(layout_det_res["boxes"]) == 0 and len(overall_ocr_res["rec_boxes"]) > 0:
|
477
|
+
for idx, ocr_rec_box in enumerate(overall_ocr_res["rec_boxes"]):
|
478
|
+
base_region_bbox = update_region_box(ocr_rec_box, base_region_bbox)
|
479
|
+
layout_det_res["boxes"].append(
|
480
|
+
{
|
481
|
+
"label": "text",
|
482
|
+
"coordinate": ocr_rec_box,
|
483
|
+
"score": overall_ocr_res["rec_scores"][idx],
|
484
|
+
}
|
485
|
+
)
|
486
|
+
block_to_ocr_map[idx] = [idx]
|
487
|
+
|
488
|
+
block_bboxes = [box["coordinate"] for box in layout_det_res["boxes"]]
|
489
|
+
region_det_res["boxes"] = sorted(
|
490
|
+
region_det_res["boxes"],
|
491
|
+
key=lambda item: caculate_bbox_area(item["coordinate"]),
|
492
|
+
)
|
493
|
+
if len(region_det_res["boxes"]) == 0:
|
494
|
+
region_det_res["boxes"] = [
|
495
|
+
{
|
496
|
+
"coordinate": base_region_bbox,
|
497
|
+
"label": "SupplementaryRegion",
|
498
|
+
"score": 1,
|
499
|
+
}
|
500
|
+
]
|
501
|
+
region_to_block_map[0] = range(len(block_bboxes))
|
502
|
+
else:
|
503
|
+
block_idxes_set = set(range(len(block_bboxes)))
|
504
|
+
# match block to region
|
505
|
+
for region_idx, region_info in enumerate(region_det_res["boxes"]):
|
506
|
+
matched_idxes = []
|
507
|
+
region_to_block_map[region_idx] = []
|
508
|
+
region_bbox = region_info["coordinate"]
|
509
|
+
for block_idx in block_idxes_set:
|
510
|
+
overlap_ratio = calculate_overlap_ratio(
|
511
|
+
region_bbox, block_bboxes[block_idx], mode="small"
|
512
|
+
)
|
513
|
+
if overlap_ratio > REGION_SETTINGS.get(
|
514
|
+
"match_block_overlap_ratio_threshold", 0.8
|
515
|
+
):
|
516
|
+
region_to_block_map[region_idx].append(block_idx)
|
517
|
+
matched_idxes.append(block_idx)
|
518
|
+
if len(matched_idxes) > 0:
|
519
|
+
for block_idx in matched_idxes:
|
520
|
+
block_idxes_set.remove(block_idx)
|
521
|
+
matched_bboxes = [block_bboxes[idx] for idx in matched_idxes]
|
522
|
+
new_region_bbox = calculate_minimum_enclosing_bbox(matched_bboxes)
|
523
|
+
region_det_res["boxes"][region_idx]["coordinate"] = new_region_bbox
|
524
|
+
# Supplement region when there is no matched block
|
525
|
+
if len(block_idxes_set) > 0:
|
526
|
+
while len(block_idxes_set) > 0:
|
527
|
+
matched_idxes = []
|
528
|
+
unmatched_bboxes = [block_bboxes[idx] for idx in block_idxes_set]
|
529
|
+
supplement_region_bbox = calculate_minimum_enclosing_bbox(
|
530
|
+
unmatched_bboxes
|
531
|
+
)
|
532
|
+
# check if the new region bbox is overlapped with other region bbox, if have, then shrink the new region bbox
|
533
|
+
for region_info in region_det_res["boxes"]:
|
534
|
+
region_bbox = region_info["coordinate"]
|
535
|
+
overlap_ratio = calculate_overlap_ratio(
|
536
|
+
supplement_region_bbox, region_bbox
|
537
|
+
)
|
538
|
+
if overlap_ratio > 0:
|
539
|
+
supplement_region_bbox, matched_idxes = (
|
540
|
+
shrink_supplement_region_bbox(
|
541
|
+
supplement_region_bbox,
|
542
|
+
region_bbox,
|
543
|
+
image.shape[1],
|
544
|
+
image.shape[0],
|
545
|
+
block_idxes_set,
|
546
|
+
block_bboxes,
|
547
|
+
)
|
548
|
+
)
|
549
|
+
if len(matched_idxes) == 0:
|
550
|
+
matched_idxes = list(block_idxes_set)
|
551
|
+
region_idx = len(region_det_res["boxes"])
|
552
|
+
region_to_block_map[region_idx] = list(matched_idxes)
|
553
|
+
for block_idx in matched_idxes:
|
554
|
+
block_idxes_set.remove(block_idx)
|
555
|
+
region_det_res["boxes"].append(
|
556
|
+
{
|
557
|
+
"coordinate": supplement_region_bbox,
|
558
|
+
"label": "SupplementaryRegion",
|
559
|
+
"score": 1,
|
560
|
+
}
|
561
|
+
)
|
562
|
+
|
563
|
+
region_block_ocr_idx_map = dict(
|
564
|
+
region_to_block_map=region_to_block_map,
|
565
|
+
block_to_ocr_map=block_to_ocr_map,
|
566
|
+
)
|
567
|
+
|
568
|
+
return region_block_ocr_idx_map, region_det_res, layout_det_res
|
569
|
+
|
570
|
+
def sort_line_by_projection(
|
571
|
+
self,
|
572
|
+
line: List[List[Union[List[int], str]]],
|
573
|
+
input_img: np.ndarray,
|
574
|
+
text_rec_model: Any,
|
575
|
+
text_rec_score_thresh: Union[float, None] = None,
|
576
|
+
direction: str = "vertical",
|
577
|
+
) -> None:
|
578
|
+
"""
|
579
|
+
Sort a line of text spans based on their vertical position within the layout bounding box.
|
580
|
+
|
581
|
+
Args:
|
582
|
+
line (list): A list of spans, where each span is a list containing a bounding box and text.
|
583
|
+
input_img (ndarray): The input image used for OCR.
|
584
|
+
general_ocr_pipeline (Any): The general OCR pipeline used for text recognition.
|
585
|
+
|
586
|
+
Returns:
|
587
|
+
list: The sorted line of text spans.
|
588
|
+
"""
|
589
|
+
sort_index = 0 if direction == "horizontal" else 1
|
590
|
+
splited_boxes = split_boxes_by_projection(line, direction)
|
591
|
+
splited_lines = []
|
592
|
+
if len(line) != len(splited_boxes):
|
593
|
+
splited_boxes.sort(key=lambda span: span[0][sort_index])
|
594
|
+
for span in splited_boxes:
|
595
|
+
bbox, text, label = span
|
596
|
+
if label == "text":
|
597
|
+
crop_img = input_img[
|
598
|
+
int(bbox[1]) : int(bbox[3]),
|
599
|
+
int(bbox[0]) : int(bbox[2]),
|
600
|
+
]
|
601
|
+
crop_img_rec_res = list(text_rec_model([crop_img]))[0]
|
602
|
+
crop_img_rec_score = crop_img_rec_res["rec_score"]
|
603
|
+
crop_img_rec_text = crop_img_rec_res["rec_text"]
|
604
|
+
text = (
|
605
|
+
crop_img_rec_text
|
606
|
+
if crop_img_rec_score >= text_rec_score_thresh
|
607
|
+
else ""
|
608
|
+
)
|
609
|
+
span[1] = text
|
610
|
+
|
611
|
+
splited_lines.append(span)
|
612
|
+
else:
|
613
|
+
splited_lines = line
|
614
|
+
|
615
|
+
return splited_lines
|
616
|
+
|
617
|
+
def get_block_rec_content(
|
618
|
+
self,
|
619
|
+
image: list,
|
620
|
+
ocr_rec_res: dict,
|
621
|
+
block: LayoutParsingBlock,
|
622
|
+
text_rec_model: Any,
|
623
|
+
text_rec_score_thresh: Union[float, None] = None,
|
624
|
+
) -> str:
|
625
|
+
|
626
|
+
if len(ocr_rec_res["rec_texts"]) == 0:
|
627
|
+
block.content = ""
|
628
|
+
return block
|
629
|
+
|
630
|
+
lines, text_direction, text_line_height = group_boxes_into_lines(
|
631
|
+
ocr_rec_res,
|
632
|
+
LINE_SETTINGS.get("line_height_iou_threshold", 0.8),
|
633
|
+
)
|
634
|
+
|
635
|
+
# format line
|
636
|
+
text_lines = []
|
637
|
+
need_new_line_num = 0
|
638
|
+
# words start coordinate and stop coordinate in the line
|
639
|
+
words_start_index = 0 if text_direction == "horizontal" else 1
|
640
|
+
words_stop_index = words_start_index + 2
|
641
|
+
lines_start_index = 1 if text_direction == "horizontal" else 3
|
642
|
+
line_width_list = []
|
643
|
+
|
644
|
+
if block.label == "reference":
|
645
|
+
rec_boxes = ocr_rec_res["boxes"]
|
646
|
+
block_start_coordinate = min([box[words_start_index] for box in rec_boxes])
|
647
|
+
block_stop_coordinate = max([box[words_stop_index] for box in rec_boxes])
|
648
|
+
else:
|
649
|
+
block_start_coordinate = block.bbox[words_start_index]
|
650
|
+
block_stop_coordinate = block.bbox[words_stop_index]
|
651
|
+
|
652
|
+
for idx, line in enumerate(lines):
|
653
|
+
line.sort(
|
654
|
+
key=lambda span: (
|
655
|
+
span[0][words_start_index] // 2,
|
656
|
+
(
|
657
|
+
span[0][lines_start_index]
|
658
|
+
if text_direction == "horizontal"
|
659
|
+
else -span[0][lines_start_index]
|
660
|
+
),
|
661
|
+
)
|
662
|
+
)
|
663
|
+
|
664
|
+
line_width = line[-1][0][words_stop_index] - line[0][0][words_start_index]
|
665
|
+
line_width_list.append(line_width)
|
666
|
+
# merge formula and text
|
667
|
+
ocr_labels = [span[2] for span in line]
|
668
|
+
if "formula" in ocr_labels:
|
669
|
+
line = self.sort_line_by_projection(
|
670
|
+
line, image, text_rec_model, text_rec_score_thresh, text_direction
|
671
|
+
)
|
672
|
+
|
673
|
+
line_text, need_new_line = format_line(
|
674
|
+
line,
|
675
|
+
text_direction,
|
676
|
+
np.max(line_width_list),
|
677
|
+
block_start_coordinate,
|
678
|
+
block_stop_coordinate,
|
679
|
+
line_gap_limit=text_line_height * 1.5,
|
680
|
+
block_label=block.label,
|
681
|
+
)
|
682
|
+
if need_new_line:
|
683
|
+
need_new_line_num += 1
|
684
|
+
if idx == 0:
|
685
|
+
line_start_coordinate = line[0][0][0]
|
686
|
+
block.seg_start_coordinate = line_start_coordinate
|
687
|
+
elif idx == len(lines) - 1:
|
688
|
+
line_end_coordinate = line[-1][0][2]
|
689
|
+
block.seg_end_coordinate = line_end_coordinate
|
690
|
+
text_lines.append(line_text)
|
691
|
+
|
692
|
+
delim = LINE_SETTINGS["delimiter_map"].get(block.label, "")
|
693
|
+
if need_new_line_num > len(text_lines) * 0.5 and delim == "":
|
694
|
+
text_lines = [text.replace("\n", "") for text in text_lines]
|
695
|
+
delim = "\n"
|
696
|
+
content = delim.join(text_lines)
|
697
|
+
block.content = content
|
698
|
+
block.num_of_lines = len(text_lines)
|
699
|
+
block.direction = text_direction
|
700
|
+
block.text_line_height = text_line_height
|
701
|
+
block.text_line_width = np.mean(line_width_list)
|
702
|
+
|
703
|
+
return block
|
704
|
+
|
705
|
+
def get_layout_parsing_blocks(
|
706
|
+
self,
|
707
|
+
image: list,
|
708
|
+
region_block_ocr_idx_map: dict,
|
709
|
+
region_det_res: DetResult,
|
710
|
+
overall_ocr_res: OCRResult,
|
711
|
+
layout_det_res: DetResult,
|
712
|
+
table_res_list: list,
|
713
|
+
seal_res_list: list,
|
714
|
+
chart_res_list: list,
|
715
|
+
text_rec_model: Any,
|
716
|
+
text_rec_score_thresh: Union[float, None] = None,
|
717
|
+
) -> list:
|
718
|
+
"""
|
719
|
+
Extract structured information from OCR and layout detection results.
|
720
|
+
|
721
|
+
Args:
|
722
|
+
image (list): The input image.
|
723
|
+
overall_ocr_res (OCRResult): An object containing the overall OCR results, including detected text boxes and recognized text. The structure is expected to have:
|
724
|
+
- "input_img": The image on which OCR was performed.
|
725
|
+
- "dt_boxes": A list of detected text box coordinates.
|
726
|
+
- "rec_texts": A list of recognized text corresponding to the detected boxes.
|
727
|
+
|
728
|
+
layout_det_res (DetResult): An object containing the layout detection results, including detected layout boxes and their labels. The structure is expected to have:
|
729
|
+
- "boxes": A list of dictionaries with keys "coordinate" for box coordinates and "block_label" for the type of content.
|
730
|
+
|
731
|
+
table_res_list (list): A list of table detection results, where each item is a dictionary containing:
|
732
|
+
- "block_bbox": The bounding box of the table layout.
|
733
|
+
- "pred_html": The predicted HTML representation of the table.
|
734
|
+
|
735
|
+
seal_res_list (List): A list of seal detection results. The details of each item depend on the specific application context.
|
736
|
+
text_rec_model (Any): A model for text recognition.
|
737
|
+
text_rec_score_thresh (Union[float, None]): The minimum score required for a recognized character to be considered valid. If None, use the default value specified during initialization. Default is None.
|
738
|
+
|
739
|
+
Returns:
|
740
|
+
list: A list of structured boxes where each item is a dictionary containing:
|
741
|
+
- "block_label": The label of the content (e.g., 'table', 'chart', 'image').
|
742
|
+
- The label as a key with either table HTML or image data and text.
|
743
|
+
- "block_bbox": The coordinates of the layout box.
|
744
|
+
"""
|
745
|
+
|
746
|
+
table_index = 0
|
747
|
+
seal_index = 0
|
748
|
+
chart_index = 0
|
749
|
+
layout_parsing_blocks: List[LayoutParsingBlock] = []
|
750
|
+
|
751
|
+
for box_idx, box_info in enumerate(layout_det_res["boxes"]):
|
752
|
+
|
753
|
+
label = box_info["label"]
|
754
|
+
block_bbox = box_info["coordinate"]
|
755
|
+
rec_res = {"boxes": [], "rec_texts": [], "rec_labels": []}
|
756
|
+
|
757
|
+
block = LayoutParsingBlock(label=label, bbox=block_bbox)
|
758
|
+
|
759
|
+
if label == "table" and len(table_res_list) > 0:
|
760
|
+
block.content = table_res_list[table_index]["pred_html"]
|
761
|
+
table_index += 1
|
762
|
+
elif label == "seal" and len(seal_res_list) > 0:
|
763
|
+
block.content = "\n".join(seal_res_list[seal_index]["rec_texts"])
|
764
|
+
seal_index += 1
|
765
|
+
elif label == "chart" and len(chart_res_list) > 0:
|
766
|
+
block.content = chart_res_list[chart_index]
|
767
|
+
chart_index += 1
|
768
|
+
else:
|
769
|
+
if label == "formula":
|
770
|
+
_, ocr_idx_list = get_sub_regions_ocr_res(
|
771
|
+
overall_ocr_res, [block_bbox], return_match_idx=True
|
772
|
+
)
|
773
|
+
region_block_ocr_idx_map["block_to_ocr_map"][box_idx] = ocr_idx_list
|
774
|
+
else:
|
775
|
+
ocr_idx_list = region_block_ocr_idx_map["block_to_ocr_map"].get(
|
776
|
+
box_idx, []
|
777
|
+
)
|
778
|
+
for box_no in ocr_idx_list:
|
779
|
+
rec_res["boxes"].append(overall_ocr_res["rec_boxes"][box_no])
|
780
|
+
rec_res["rec_texts"].append(
|
781
|
+
overall_ocr_res["rec_texts"][box_no],
|
782
|
+
)
|
783
|
+
rec_res["rec_labels"].append(
|
784
|
+
overall_ocr_res["rec_labels"][box_no],
|
785
|
+
)
|
786
|
+
block = self.get_block_rec_content(
|
787
|
+
image=image,
|
788
|
+
block=block,
|
789
|
+
ocr_rec_res=rec_res,
|
790
|
+
text_rec_model=text_rec_model,
|
791
|
+
text_rec_score_thresh=text_rec_score_thresh,
|
792
|
+
)
|
793
|
+
|
794
|
+
if (
|
795
|
+
label
|
796
|
+
in ["seal", "table", "formula", "chart"]
|
797
|
+
+ BLOCK_LABEL_MAP["image_labels"]
|
798
|
+
):
|
799
|
+
x_min, y_min, x_max, y_max = list(map(int, block_bbox))
|
800
|
+
img_path = (
|
801
|
+
f"imgs/img_in_{block.label}_box_{x_min}_{y_min}_{x_max}_{y_max}.jpg"
|
802
|
+
)
|
803
|
+
img = Image.fromarray(image[y_min:y_max, x_min:x_max, ::-1])
|
804
|
+
block.image = {"path": img_path, "img": img}
|
805
|
+
|
806
|
+
layout_parsing_blocks.append(block)
|
807
|
+
|
808
|
+
region_list: List[LayoutParsingRegion] = []
|
809
|
+
for region_idx, region_info in enumerate(region_det_res["boxes"]):
|
810
|
+
region_bbox = region_info["coordinate"]
|
811
|
+
region_blocks = [
|
812
|
+
layout_parsing_blocks[idx]
|
813
|
+
for idx in region_block_ocr_idx_map["region_to_block_map"][region_idx]
|
814
|
+
]
|
815
|
+
region = LayoutParsingRegion(
|
816
|
+
bbox=region_bbox,
|
817
|
+
blocks=region_blocks,
|
818
|
+
image_shape=image.shape[:2],
|
819
|
+
)
|
820
|
+
region_list.append(region)
|
821
|
+
|
822
|
+
region_list = sorted(
|
823
|
+
region_list,
|
824
|
+
key=lambda r: (r.weighted_distance),
|
825
|
+
)
|
826
|
+
|
827
|
+
return region_list
|
828
|
+
|
829
|
+
def get_layout_parsing_res(
|
830
|
+
self,
|
831
|
+
image: list,
|
832
|
+
region_det_res: DetResult,
|
833
|
+
layout_det_res: DetResult,
|
834
|
+
overall_ocr_res: OCRResult,
|
835
|
+
table_res_list: list,
|
836
|
+
seal_res_list: list,
|
837
|
+
chart_res_list: list,
|
838
|
+
formula_res_list: list,
|
839
|
+
text_rec_score_thresh: Union[float, None] = None,
|
840
|
+
) -> list:
|
841
|
+
"""
|
842
|
+
Retrieves the layout parsing result based on the layout detection result, OCR result, and other recognition results.
|
843
|
+
Args:
|
844
|
+
image (list): The input image.
|
845
|
+
layout_det_res (DetResult): The detection result containing the layout information of the document.
|
846
|
+
overall_ocr_res (OCRResult): The overall OCR result containing text information.
|
847
|
+
table_res_list (list): A list of table recognition results.
|
848
|
+
seal_res_list (list): A list of seal recognition results.
|
849
|
+
formula_res_list (list): A list of formula recognition results.
|
850
|
+
text_rec_score_thresh (Optional[float], optional): The score threshold for text recognition. Defaults to None.
|
851
|
+
Returns:
|
852
|
+
list: A list of dictionaries representing the layout parsing result.
|
853
|
+
"""
|
854
|
+
|
855
|
+
# Standardize data
|
856
|
+
region_block_ocr_idx_map, region_det_res, layout_det_res = (
|
857
|
+
self.standardized_data(
|
858
|
+
image=image,
|
859
|
+
region_det_res=region_det_res,
|
860
|
+
layout_det_res=layout_det_res,
|
861
|
+
overall_ocr_res=overall_ocr_res,
|
862
|
+
formula_res_list=formula_res_list,
|
863
|
+
text_rec_model=self.general_ocr_pipeline.text_rec_model,
|
864
|
+
text_rec_score_thresh=text_rec_score_thresh,
|
865
|
+
)
|
866
|
+
)
|
867
|
+
|
868
|
+
# Format layout parsing block
|
869
|
+
region_list = self.get_layout_parsing_blocks(
|
870
|
+
image=image,
|
871
|
+
region_block_ocr_idx_map=region_block_ocr_idx_map,
|
872
|
+
region_det_res=region_det_res,
|
873
|
+
overall_ocr_res=overall_ocr_res,
|
874
|
+
layout_det_res=layout_det_res,
|
875
|
+
table_res_list=table_res_list,
|
876
|
+
seal_res_list=seal_res_list,
|
877
|
+
chart_res_list=chart_res_list,
|
878
|
+
text_rec_model=self.general_ocr_pipeline.text_rec_model,
|
879
|
+
text_rec_score_thresh=self.general_ocr_pipeline.text_rec_score_thresh,
|
880
|
+
)
|
881
|
+
|
882
|
+
parsing_res_list = []
|
883
|
+
for region in region_list:
|
884
|
+
parsing_res_list.extend(region.sort())
|
885
|
+
|
886
|
+
index = 1
|
887
|
+
for block in parsing_res_list:
|
888
|
+
if block.label in BLOCK_LABEL_MAP["visualize_index_labels"]:
|
889
|
+
block.order_index = index
|
890
|
+
index += 1
|
891
|
+
|
892
|
+
return parsing_res_list
|
893
|
+
|
894
|
+
def get_model_settings(
|
895
|
+
self,
|
896
|
+
use_doc_orientation_classify: Union[bool, None],
|
897
|
+
use_doc_unwarping: Union[bool, None],
|
898
|
+
use_seal_recognition: Union[bool, None],
|
899
|
+
use_table_recognition: Union[bool, None],
|
900
|
+
use_formula_recognition: Union[bool, None],
|
901
|
+
use_chart_recognition: Union[bool, None],
|
902
|
+
use_region_detection: Union[bool, None],
|
903
|
+
) -> dict:
|
904
|
+
"""
|
905
|
+
Get the model settings based on the provided parameters or default values.
|
906
|
+
|
907
|
+
Args:
|
908
|
+
use_doc_orientation_classify (Union[bool, None]): Enables document orientation classification if True. Defaults to system setting if None.
|
909
|
+
use_doc_unwarping (Union[bool, None]): Enables document unwarping if True. Defaults to system setting if None.
|
910
|
+
use_seal_recognition (Union[bool, None]): Enables seal recognition if True. Defaults to system setting if None.
|
911
|
+
use_table_recognition (Union[bool, None]): Enables table recognition if True. Defaults to system setting if None.
|
912
|
+
use_formula_recognition (Union[bool, None]): Enables formula recognition if True. Defaults to system setting if None.
|
913
|
+
|
914
|
+
Returns:
|
915
|
+
dict: A dictionary containing the model settings.
|
916
|
+
|
917
|
+
"""
|
918
|
+
if use_doc_orientation_classify is None and use_doc_unwarping is None:
|
919
|
+
use_doc_preprocessor = self.use_doc_preprocessor
|
920
|
+
else:
|
921
|
+
if use_doc_orientation_classify is True or use_doc_unwarping is True:
|
922
|
+
use_doc_preprocessor = True
|
923
|
+
else:
|
924
|
+
use_doc_preprocessor = False
|
925
|
+
|
926
|
+
if use_seal_recognition is None:
|
927
|
+
use_seal_recognition = self.use_seal_recognition
|
928
|
+
|
929
|
+
if use_table_recognition is None:
|
930
|
+
use_table_recognition = self.use_table_recognition
|
931
|
+
|
932
|
+
if use_formula_recognition is None:
|
933
|
+
use_formula_recognition = self.use_formula_recognition
|
934
|
+
|
935
|
+
if use_region_detection is None:
|
936
|
+
use_region_detection = self.use_region_detection
|
937
|
+
|
938
|
+
if use_chart_recognition is None:
|
939
|
+
use_chart_recognition = self.use_chart_recognition
|
940
|
+
|
941
|
+
return dict(
|
942
|
+
use_doc_preprocessor=use_doc_preprocessor,
|
943
|
+
use_seal_recognition=use_seal_recognition,
|
944
|
+
use_table_recognition=use_table_recognition,
|
945
|
+
use_formula_recognition=use_formula_recognition,
|
946
|
+
use_chart_recognition=use_chart_recognition,
|
947
|
+
use_region_detection=use_region_detection,
|
948
|
+
)
|
949
|
+
|
950
|
+
def predict(
|
951
|
+
self,
|
952
|
+
input: Union[str, list[str], np.ndarray, list[np.ndarray]],
|
953
|
+
use_doc_orientation_classify: Union[bool, None] = False,
|
954
|
+
use_doc_unwarping: Union[bool, None] = False,
|
955
|
+
use_textline_orientation: Optional[bool] = None,
|
956
|
+
use_seal_recognition: Union[bool, None] = None,
|
957
|
+
use_table_recognition: Union[bool, None] = None,
|
958
|
+
use_formula_recognition: Union[bool, None] = None,
|
959
|
+
use_chart_recognition: Union[bool, None] = False,
|
960
|
+
use_region_detection: Union[bool, None] = None,
|
961
|
+
layout_threshold: Optional[Union[float, dict]] = None,
|
962
|
+
layout_nms: Optional[bool] = None,
|
963
|
+
layout_unclip_ratio: Optional[Union[float, Tuple[float, float], dict]] = None,
|
964
|
+
layout_merge_bboxes_mode: Optional[str] = None,
|
965
|
+
text_det_limit_side_len: Union[int, None] = None,
|
966
|
+
text_det_limit_type: Union[str, None] = None,
|
967
|
+
text_det_thresh: Union[float, None] = None,
|
968
|
+
text_det_box_thresh: Union[float, None] = None,
|
969
|
+
text_det_unclip_ratio: Union[float, None] = None,
|
970
|
+
text_rec_score_thresh: Union[float, None] = None,
|
971
|
+
seal_det_limit_side_len: Union[int, None] = None,
|
972
|
+
seal_det_limit_type: Union[str, None] = None,
|
973
|
+
seal_det_thresh: Union[float, None] = None,
|
974
|
+
seal_det_box_thresh: Union[float, None] = None,
|
975
|
+
seal_det_unclip_ratio: Union[float, None] = None,
|
976
|
+
seal_rec_score_thresh: Union[float, None] = None,
|
977
|
+
use_wired_table_cells_trans_to_html: bool = False,
|
978
|
+
use_wireless_table_cells_trans_to_html: bool = False,
|
979
|
+
use_table_orientation_classify: bool = True,
|
980
|
+
use_ocr_results_with_table_cells: bool = True,
|
981
|
+
use_e2e_wired_table_rec_model: bool = False,
|
982
|
+
use_e2e_wireless_table_rec_model: bool = True,
|
983
|
+
**kwargs,
|
984
|
+
) -> LayoutParsingResultV2:
|
985
|
+
"""
|
986
|
+
Predicts the layout parsing result for the given input.
|
987
|
+
|
988
|
+
Args:
|
989
|
+
use_doc_orientation_classify (Optional[bool]): Whether to use document orientation classification.
|
990
|
+
use_doc_unwarping (Optional[bool]): Whether to use document unwarping.
|
991
|
+
use_textline_orientation (Optional[bool]): Whether to use textline orientation prediction.
|
992
|
+
use_seal_recognition (Optional[bool]): Whether to use seal recognition.
|
993
|
+
use_table_recognition (Optional[bool]): Whether to use table recognition.
|
994
|
+
use_formula_recognition (Optional[bool]): Whether to use formula recognition.
|
995
|
+
use_region_detection (Optional[bool]): Whether to use region detection.
|
996
|
+
layout_threshold (Optional[float]): The threshold value to filter out low-confidence predictions. Default is None.
|
997
|
+
layout_nms (bool, optional): Whether to use layout-aware NMS. Defaults to False.
|
998
|
+
layout_unclip_ratio (Optional[Union[float, Tuple[float, float]]], optional): The ratio of unclipping the bounding box.
|
999
|
+
Defaults to None.
|
1000
|
+
If it's a single number, then both width and height are used.
|
1001
|
+
If it's a tuple of two numbers, then they are used separately for width and height respectively.
|
1002
|
+
If it's None, then no unclipping will be performed.
|
1003
|
+
layout_merge_bboxes_mode (Optional[str], optional): The mode for merging bounding boxes. Defaults to None.
|
1004
|
+
text_det_limit_side_len (Optional[int]): Maximum side length for text detection.
|
1005
|
+
text_det_limit_type (Optional[str]): Type of limit to apply for text detection.
|
1006
|
+
text_det_thresh (Optional[float]): Threshold for text detection.
|
1007
|
+
text_det_box_thresh (Optional[float]): Threshold for text detection boxes.
|
1008
|
+
text_det_unclip_ratio (Optional[float]): Ratio for unclipping text detection boxes.
|
1009
|
+
text_rec_score_thresh (Optional[float]): Score threshold for text recognition.
|
1010
|
+
seal_det_limit_side_len (Optional[int]): Maximum side length for seal detection.
|
1011
|
+
seal_det_limit_type (Optional[str]): Type of limit to apply for seal detection.
|
1012
|
+
seal_det_thresh (Optional[float]): Threshold for seal detection.
|
1013
|
+
seal_det_box_thresh (Optional[float]): Threshold for seal detection boxes.
|
1014
|
+
seal_det_unclip_ratio (Optional[float]): Ratio for unclipping seal detection boxes.
|
1015
|
+
seal_rec_score_thresh (Optional[float]): Score threshold for seal recognition.
|
1016
|
+
use_wired_table_cells_trans_to_html (bool): Whether to use wired table cells trans to HTML.
|
1017
|
+
use_wireless_table_cells_trans_to_html (bool): Whether to use wireless table cells trans to HTML.
|
1018
|
+
use_table_orientation_classify (bool): Whether to use table orientation classification.
|
1019
|
+
use_ocr_results_with_table_cells (bool): Whether to use OCR results processed by table cells.
|
1020
|
+
use_e2e_wired_table_rec_model (bool): Whether to use end-to-end wired table recognition model.
|
1021
|
+
use_e2e_wireless_table_rec_model (bool): Whether to use end-to-end wireless table recognition model.
|
1022
|
+
**kwargs (Any): Additional settings to extend functionality.
|
1023
|
+
|
1024
|
+
Returns:
|
1025
|
+
LayoutParsingResultV2: The predicted layout parsing result.
|
1026
|
+
"""
|
1027
|
+
|
1028
|
+
model_settings = self.get_model_settings(
|
1029
|
+
use_doc_orientation_classify,
|
1030
|
+
use_doc_unwarping,
|
1031
|
+
use_seal_recognition,
|
1032
|
+
use_table_recognition,
|
1033
|
+
use_formula_recognition,
|
1034
|
+
use_chart_recognition,
|
1035
|
+
use_region_detection,
|
1036
|
+
)
|
1037
|
+
|
1038
|
+
if not self.check_model_settings_valid(model_settings):
|
1039
|
+
yield {"error": "the input params for model settings are invalid!"}
|
1040
|
+
|
1041
|
+
for batch_data in self.batch_sampler(input):
|
1042
|
+
image_arrays = self.img_reader(batch_data.instances)
|
1043
|
+
|
1044
|
+
if model_settings["use_doc_preprocessor"]:
|
1045
|
+
doc_preprocessor_results = list(
|
1046
|
+
self.doc_preprocessor_pipeline(
|
1047
|
+
image_arrays,
|
1048
|
+
use_doc_orientation_classify=use_doc_orientation_classify,
|
1049
|
+
use_doc_unwarping=use_doc_unwarping,
|
1050
|
+
)
|
1051
|
+
)
|
1052
|
+
else:
|
1053
|
+
doc_preprocessor_results = [{"output_img": arr} for arr in image_arrays]
|
1054
|
+
|
1055
|
+
doc_preprocessor_images = [
|
1056
|
+
item["output_img"] for item in doc_preprocessor_results
|
1057
|
+
]
|
1058
|
+
|
1059
|
+
layout_det_results = list(
|
1060
|
+
self.layout_det_model(
|
1061
|
+
doc_preprocessor_images,
|
1062
|
+
threshold=layout_threshold,
|
1063
|
+
layout_nms=layout_nms,
|
1064
|
+
layout_unclip_ratio=layout_unclip_ratio,
|
1065
|
+
layout_merge_bboxes_mode=layout_merge_bboxes_mode,
|
1066
|
+
)
|
1067
|
+
)
|
1068
|
+
imgs_in_doc = [
|
1069
|
+
gather_imgs(img, res["boxes"])
|
1070
|
+
for img, res in zip(doc_preprocessor_images, layout_det_results)
|
1071
|
+
]
|
1072
|
+
|
1073
|
+
if model_settings["use_region_detection"]:
|
1074
|
+
region_det_results = list(
|
1075
|
+
self.region_detection_model(
|
1076
|
+
doc_preprocessor_images,
|
1077
|
+
layout_nms=True,
|
1078
|
+
layout_merge_bboxes_mode="small",
|
1079
|
+
),
|
1080
|
+
)
|
1081
|
+
else:
|
1082
|
+
region_det_results = [{"boxes": []} for _ in doc_preprocessor_images]
|
1083
|
+
|
1084
|
+
if model_settings["use_formula_recognition"]:
|
1085
|
+
formula_res_all = list(
|
1086
|
+
self.formula_recognition_pipeline(
|
1087
|
+
doc_preprocessor_images,
|
1088
|
+
use_layout_detection=False,
|
1089
|
+
use_doc_orientation_classify=False,
|
1090
|
+
use_doc_unwarping=False,
|
1091
|
+
layout_det_res=layout_det_results,
|
1092
|
+
),
|
1093
|
+
)
|
1094
|
+
formula_res_lists = [
|
1095
|
+
item["formula_res_list"] for item in formula_res_all
|
1096
|
+
]
|
1097
|
+
else:
|
1098
|
+
formula_res_lists = [[] for _ in doc_preprocessor_images]
|
1099
|
+
|
1100
|
+
for doc_preprocessor_image, formula_res_list in zip(
|
1101
|
+
doc_preprocessor_images, formula_res_lists
|
1102
|
+
):
|
1103
|
+
for formula_res in formula_res_list:
|
1104
|
+
x_min, y_min, x_max, y_max = list(map(int, formula_res["dt_polys"]))
|
1105
|
+
doc_preprocessor_image[y_min:y_max, x_min:x_max, :] = 255.0
|
1106
|
+
|
1107
|
+
overall_ocr_results = list(
|
1108
|
+
self.general_ocr_pipeline(
|
1109
|
+
doc_preprocessor_images,
|
1110
|
+
use_textline_orientation=use_textline_orientation,
|
1111
|
+
text_det_limit_side_len=text_det_limit_side_len,
|
1112
|
+
text_det_limit_type=text_det_limit_type,
|
1113
|
+
text_det_thresh=text_det_thresh,
|
1114
|
+
text_det_box_thresh=text_det_box_thresh,
|
1115
|
+
text_det_unclip_ratio=text_det_unclip_ratio,
|
1116
|
+
text_rec_score_thresh=text_rec_score_thresh,
|
1117
|
+
),
|
1118
|
+
)
|
1119
|
+
|
1120
|
+
for overall_ocr_res in overall_ocr_results:
|
1121
|
+
overall_ocr_res["rec_labels"] = ["text"] * len(
|
1122
|
+
overall_ocr_res["rec_texts"]
|
1123
|
+
)
|
1124
|
+
|
1125
|
+
if model_settings["use_table_recognition"]:
|
1126
|
+
table_res_lists = []
|
1127
|
+
for (
|
1128
|
+
layout_det_res,
|
1129
|
+
doc_preprocessor_image,
|
1130
|
+
overall_ocr_res,
|
1131
|
+
formula_res_list,
|
1132
|
+
imgs_in_doc_for_img,
|
1133
|
+
) in zip(
|
1134
|
+
layout_det_results,
|
1135
|
+
doc_preprocessor_images,
|
1136
|
+
overall_ocr_results,
|
1137
|
+
formula_res_lists,
|
1138
|
+
imgs_in_doc,
|
1139
|
+
):
|
1140
|
+
table_contents_for_img = copy.deepcopy(overall_ocr_res)
|
1141
|
+
for formula_res in formula_res_list:
|
1142
|
+
x_min, y_min, x_max, y_max = list(
|
1143
|
+
map(int, formula_res["dt_polys"])
|
1144
|
+
)
|
1145
|
+
poly_points = [
|
1146
|
+
(x_min, y_min),
|
1147
|
+
(x_max, y_min),
|
1148
|
+
(x_max, y_max),
|
1149
|
+
(x_min, y_max),
|
1150
|
+
]
|
1151
|
+
table_contents_for_img["dt_polys"].append(poly_points)
|
1152
|
+
rec_formula = formula_res["rec_formula"]
|
1153
|
+
if not rec_formula.startswith("$") or not rec_formula.endswith(
|
1154
|
+
"$"
|
1155
|
+
):
|
1156
|
+
rec_formula = f"${rec_formula}$"
|
1157
|
+
table_contents_for_img["rec_texts"].append(f"{rec_formula}")
|
1158
|
+
if table_contents_for_img["rec_boxes"].size == 0:
|
1159
|
+
table_contents_for_img["rec_boxes"] = np.array(
|
1160
|
+
[formula_res["dt_polys"]]
|
1161
|
+
)
|
1162
|
+
else:
|
1163
|
+
table_contents_for_img["rec_boxes"] = np.vstack(
|
1164
|
+
(
|
1165
|
+
table_contents_for_img["rec_boxes"],
|
1166
|
+
[formula_res["dt_polys"]],
|
1167
|
+
)
|
1168
|
+
)
|
1169
|
+
table_contents_for_img["rec_polys"].append(poly_points)
|
1170
|
+
table_contents_for_img["rec_scores"].append(1)
|
1171
|
+
|
1172
|
+
for img in imgs_in_doc_for_img:
|
1173
|
+
img_path = img["path"]
|
1174
|
+
x_min, y_min, x_max, y_max = img["coordinate"]
|
1175
|
+
poly_points = [
|
1176
|
+
(x_min, y_min),
|
1177
|
+
(x_max, y_min),
|
1178
|
+
(x_max, y_max),
|
1179
|
+
(x_min, y_max),
|
1180
|
+
]
|
1181
|
+
table_contents_for_img["dt_polys"].append(poly_points)
|
1182
|
+
table_contents_for_img["rec_texts"].append(
|
1183
|
+
f'<div style="text-align: center;"><img src="{img_path}" alt="Image" /></div>'
|
1184
|
+
)
|
1185
|
+
if table_contents_for_img["rec_boxes"].size == 0:
|
1186
|
+
table_contents_for_img["rec_boxes"] = np.array(
|
1187
|
+
[img["coordinate"]]
|
1188
|
+
)
|
1189
|
+
else:
|
1190
|
+
table_contents_for_img["rec_boxes"] = np.vstack(
|
1191
|
+
(table_contents_for_img["rec_boxes"], img["coordinate"])
|
1192
|
+
)
|
1193
|
+
table_contents_for_img["rec_polys"].append(poly_points)
|
1194
|
+
table_contents_for_img["rec_scores"].append(img["score"])
|
1195
|
+
|
1196
|
+
table_res_all = list(
|
1197
|
+
self.table_recognition_pipeline(
|
1198
|
+
doc_preprocessor_image,
|
1199
|
+
use_doc_orientation_classify=False,
|
1200
|
+
use_doc_unwarping=False,
|
1201
|
+
use_layout_detection=False,
|
1202
|
+
use_ocr_model=False,
|
1203
|
+
overall_ocr_res=table_contents_for_img,
|
1204
|
+
layout_det_res=layout_det_res,
|
1205
|
+
cell_sort_by_y_projection=True,
|
1206
|
+
use_wired_table_cells_trans_to_html=use_wired_table_cells_trans_to_html,
|
1207
|
+
use_wireless_table_cells_trans_to_html=use_wireless_table_cells_trans_to_html,
|
1208
|
+
use_table_orientation_classify=use_table_orientation_classify,
|
1209
|
+
use_ocr_results_with_table_cells=use_ocr_results_with_table_cells,
|
1210
|
+
use_e2e_wired_table_rec_model=use_e2e_wired_table_rec_model,
|
1211
|
+
use_e2e_wireless_table_rec_model=use_e2e_wireless_table_rec_model,
|
1212
|
+
),
|
1213
|
+
)
|
1214
|
+
single_table_res_lists = [
|
1215
|
+
item["table_res_list"] for item in table_res_all
|
1216
|
+
]
|
1217
|
+
table_res_lists.extend(single_table_res_lists)
|
1218
|
+
else:
|
1219
|
+
table_res_lists = [[] for _ in doc_preprocessor_images]
|
1220
|
+
|
1221
|
+
if model_settings["use_seal_recognition"]:
|
1222
|
+
seal_res_all = list(
|
1223
|
+
self.seal_recognition_pipeline(
|
1224
|
+
doc_preprocessor_images,
|
1225
|
+
use_doc_orientation_classify=False,
|
1226
|
+
use_doc_unwarping=False,
|
1227
|
+
use_layout_detection=False,
|
1228
|
+
layout_det_res=layout_det_results,
|
1229
|
+
seal_det_limit_side_len=seal_det_limit_side_len,
|
1230
|
+
seal_det_limit_type=seal_det_limit_type,
|
1231
|
+
seal_det_thresh=seal_det_thresh,
|
1232
|
+
seal_det_box_thresh=seal_det_box_thresh,
|
1233
|
+
seal_det_unclip_ratio=seal_det_unclip_ratio,
|
1234
|
+
seal_rec_score_thresh=seal_rec_score_thresh,
|
1235
|
+
),
|
1236
|
+
)
|
1237
|
+
seal_res_lists = [item["seal_res_list"] for item in seal_res_all]
|
1238
|
+
else:
|
1239
|
+
seal_res_lists = [[] for _ in doc_preprocessor_images]
|
1240
|
+
|
1241
|
+
for (
|
1242
|
+
input_path,
|
1243
|
+
page_index,
|
1244
|
+
doc_preprocessor_image,
|
1245
|
+
doc_preprocessor_res,
|
1246
|
+
layout_det_res,
|
1247
|
+
region_det_res,
|
1248
|
+
overall_ocr_res,
|
1249
|
+
table_res_list,
|
1250
|
+
seal_res_list,
|
1251
|
+
formula_res_list,
|
1252
|
+
imgs_in_doc_for_img,
|
1253
|
+
) in zip(
|
1254
|
+
batch_data.input_paths,
|
1255
|
+
batch_data.page_indexes,
|
1256
|
+
doc_preprocessor_images,
|
1257
|
+
doc_preprocessor_results,
|
1258
|
+
layout_det_results,
|
1259
|
+
region_det_results,
|
1260
|
+
overall_ocr_results,
|
1261
|
+
table_res_lists,
|
1262
|
+
seal_res_lists,
|
1263
|
+
formula_res_lists,
|
1264
|
+
imgs_in_doc,
|
1265
|
+
):
|
1266
|
+
chart_res_list = []
|
1267
|
+
if model_settings["use_chart_recognition"]:
|
1268
|
+
chart_imgs_list = []
|
1269
|
+
for bbox in layout_det_res["boxes"]:
|
1270
|
+
if bbox["label"] == "chart":
|
1271
|
+
x_min, y_min, x_max, y_max = bbox["coordinate"]
|
1272
|
+
chart_img = doc_preprocessor_image[
|
1273
|
+
int(y_min) : int(y_max), int(x_min) : int(x_max), :
|
1274
|
+
]
|
1275
|
+
chart_imgs_list.append({"image": chart_img})
|
1276
|
+
|
1277
|
+
for chart_res_batch in self.chart_recognition_model(
|
1278
|
+
input=chart_imgs_list
|
1279
|
+
):
|
1280
|
+
chart_res_list.append(chart_res_batch["result"])
|
1281
|
+
|
1282
|
+
parsing_res_list = self.get_layout_parsing_res(
|
1283
|
+
doc_preprocessor_image,
|
1284
|
+
region_det_res=region_det_res,
|
1285
|
+
layout_det_res=layout_det_res,
|
1286
|
+
overall_ocr_res=overall_ocr_res,
|
1287
|
+
table_res_list=table_res_list,
|
1288
|
+
seal_res_list=seal_res_list,
|
1289
|
+
chart_res_list=chart_res_list,
|
1290
|
+
formula_res_list=formula_res_list,
|
1291
|
+
text_rec_score_thresh=text_rec_score_thresh,
|
1292
|
+
)
|
1293
|
+
|
1294
|
+
for formula_res in formula_res_list:
|
1295
|
+
x_min, y_min, x_max, y_max = list(map(int, formula_res["dt_polys"]))
|
1296
|
+
doc_preprocessor_image[y_min:y_max, x_min:x_max, :] = formula_res[
|
1297
|
+
"input_img"
|
1298
|
+
]
|
1299
|
+
|
1300
|
+
single_img_res = {
|
1301
|
+
"input_path": input_path,
|
1302
|
+
"page_index": page_index,
|
1303
|
+
"doc_preprocessor_res": doc_preprocessor_res,
|
1304
|
+
"layout_det_res": layout_det_res,
|
1305
|
+
"region_det_res": region_det_res,
|
1306
|
+
"overall_ocr_res": overall_ocr_res,
|
1307
|
+
"table_res_list": table_res_list,
|
1308
|
+
"seal_res_list": seal_res_list,
|
1309
|
+
"chart_res_list": chart_res_list,
|
1310
|
+
"formula_res_list": formula_res_list,
|
1311
|
+
"parsing_res_list": parsing_res_list,
|
1312
|
+
"imgs_in_doc": imgs_in_doc_for_img,
|
1313
|
+
"model_settings": model_settings,
|
1314
|
+
}
|
1315
|
+
yield LayoutParsingResultV2(single_img_res)
|
1316
|
+
|
1317
|
+
def concatenate_markdown_pages(self, markdown_list: list) -> tuple:
|
1318
|
+
"""
|
1319
|
+
Concatenate Markdown content from multiple pages into a single document.
|
1320
|
+
|
1321
|
+
Args:
|
1322
|
+
markdown_list (list): A list containing Markdown data for each page.
|
1323
|
+
|
1324
|
+
Returns:
|
1325
|
+
tuple: A tuple containing the processed Markdown text.
|
1326
|
+
"""
|
1327
|
+
markdown_texts = ""
|
1328
|
+
previous_page_last_element_paragraph_end_flag = True
|
1329
|
+
|
1330
|
+
for res in markdown_list:
|
1331
|
+
# Get the paragraph flags for the current page
|
1332
|
+
page_first_element_paragraph_start_flag: bool = res[
|
1333
|
+
"page_continuation_flags"
|
1334
|
+
][0]
|
1335
|
+
page_last_element_paragraph_end_flag: bool = res["page_continuation_flags"][
|
1336
|
+
1
|
1337
|
+
]
|
1338
|
+
|
1339
|
+
# Determine whether to add a space or a newline
|
1340
|
+
if (
|
1341
|
+
not page_first_element_paragraph_start_flag
|
1342
|
+
and not previous_page_last_element_paragraph_end_flag
|
1343
|
+
):
|
1344
|
+
last_char_of_markdown = markdown_texts[-1] if markdown_texts else ""
|
1345
|
+
first_char_of_handler = (
|
1346
|
+
res["markdown_texts"][0] if res["markdown_texts"] else ""
|
1347
|
+
)
|
1348
|
+
|
1349
|
+
# Check if the last character and the first character are Chinese characters
|
1350
|
+
last_is_chinese_char = (
|
1351
|
+
re.match(r"[\u4e00-\u9fff]", last_char_of_markdown)
|
1352
|
+
if last_char_of_markdown
|
1353
|
+
else False
|
1354
|
+
)
|
1355
|
+
first_is_chinese_char = (
|
1356
|
+
re.match(r"[\u4e00-\u9fff]", first_char_of_handler)
|
1357
|
+
if first_char_of_handler
|
1358
|
+
else False
|
1359
|
+
)
|
1360
|
+
if not (last_is_chinese_char or first_is_chinese_char):
|
1361
|
+
markdown_texts += " " + res["markdown_texts"]
|
1362
|
+
else:
|
1363
|
+
markdown_texts += res["markdown_texts"]
|
1364
|
+
else:
|
1365
|
+
markdown_texts += "\n\n" + res["markdown_texts"]
|
1366
|
+
previous_page_last_element_paragraph_end_flag = (
|
1367
|
+
page_last_element_paragraph_end_flag
|
1368
|
+
)
|
1369
|
+
|
1370
|
+
return markdown_texts
|
1371
|
+
|
1372
|
+
|
1373
|
+
@pipeline_requires_extra("ocr")
|
1374
|
+
class LayoutParsingPipelineV2(AutoParallelImageSimpleInferencePipeline):
|
1375
|
+
entities = ["PP-StructureV3"]
|
1376
|
+
|
1377
|
+
@property
|
1378
|
+
def _pipeline_cls(self):
|
1379
|
+
return _LayoutParsingPipelineV2
|
1380
|
+
|
1381
|
+
def _get_batch_size(self, config):
|
1382
|
+
return config.get("batch_size", 1)
|