paddlex 2.1.0__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 -19
- 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 +32 -45
- 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/deploy/__init__.py +0 -1
- PaddleClas/deploy/paddleserving/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/classification_web_service.py +0 -74
- PaddleClas/deploy/paddleserving/cpu_utilization.py +0 -4
- PaddleClas/deploy/paddleserving/pipeline_http_client.py +0 -20
- PaddleClas/deploy/paddleserving/pipeline_rpc_client.py +0 -33
- PaddleClas/deploy/paddleserving/recognition/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/recognition/pipeline_http_client.py +0 -21
- PaddleClas/deploy/paddleserving/recognition/pipeline_rpc_client.py +0 -34
- PaddleClas/deploy/paddleserving/recognition/recognition_web_service.py +0 -209
- PaddleClas/deploy/python/__init__.py +0 -0
- PaddleClas/deploy/python/build_gallery.py +0 -214
- PaddleClas/deploy/python/det_preprocess.py +0 -205
- PaddleClas/deploy/python/postprocess.py +0 -161
- PaddleClas/deploy/python/predict_cls.py +0 -142
- PaddleClas/deploy/python/predict_det.py +0 -158
- PaddleClas/deploy/python/predict_rec.py +0 -138
- PaddleClas/deploy/python/predict_system.py +0 -144
- PaddleClas/deploy/python/preprocess.py +0 -337
- PaddleClas/deploy/utils/__init__.py +0 -5
- PaddleClas/deploy/utils/config.py +0 -197
- PaddleClas/deploy/utils/draw_bbox.py +0 -61
- PaddleClas/deploy/utils/encode_decode.py +0 -31
- PaddleClas/deploy/utils/get_image_list.py +0 -49
- PaddleClas/deploy/utils/logger.py +0 -120
- PaddleClas/deploy/utils/predictor.py +0 -71
- PaddleClas/deploy/vector_search/__init__.py +0 -1
- PaddleClas/deploy/vector_search/interface.py +0 -272
- PaddleClas/deploy/vector_search/test.py +0 -34
- PaddleClas/hubconf.py +0 -788
- PaddleClas/paddleclas.py +0 -552
- PaddleClas/ppcls/__init__.py +0 -20
- PaddleClas/ppcls/arch/__init__.py +0 -127
- PaddleClas/ppcls/arch/backbone/__init__.py +0 -80
- PaddleClas/ppcls/arch/backbone/base/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/base/theseus_layer.py +0 -126
- PaddleClas/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- PaddleClas/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- PaddleClas/ppcls/arch/backbone/legendary_models/hrnet.py +0 -744
- PaddleClas/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- PaddleClas/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- PaddleClas/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- PaddleClas/ppcls/arch/backbone/legendary_models/vgg.py +0 -231
- PaddleClas/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- PaddleClas/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- PaddleClas/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- PaddleClas/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- PaddleClas/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- PaddleClas/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- PaddleClas/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- PaddleClas/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- PaddleClas/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- PaddleClas/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- PaddleClas/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- PaddleClas/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- PaddleClas/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- PaddleClas/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- PaddleClas/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- PaddleClas/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- PaddleClas/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- PaddleClas/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- PaddleClas/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- PaddleClas/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- PaddleClas/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- PaddleClas/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- PaddleClas/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- PaddleClas/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- PaddleClas/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- PaddleClas/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- PaddleClas/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- PaddleClas/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- PaddleClas/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- PaddleClas/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- PaddleClas/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- PaddleClas/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- PaddleClas/ppcls/arch/gears/__init__.py +0 -32
- PaddleClas/ppcls/arch/gears/arcmargin.py +0 -72
- PaddleClas/ppcls/arch/gears/circlemargin.py +0 -59
- PaddleClas/ppcls/arch/gears/cosmargin.py +0 -55
- PaddleClas/ppcls/arch/gears/fc.py +0 -35
- PaddleClas/ppcls/arch/gears/identity_head.py +0 -9
- PaddleClas/ppcls/arch/gears/vehicle_neck.py +0 -52
- PaddleClas/ppcls/arch/utils.py +0 -53
- PaddleClas/ppcls/data/__init__.py +0 -144
- PaddleClas/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- PaddleClas/ppcls/data/dataloader/__init__.py +0 -9
- PaddleClas/ppcls/data/dataloader/common_dataset.py +0 -84
- PaddleClas/ppcls/data/dataloader/dali.py +0 -319
- PaddleClas/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- PaddleClas/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- PaddleClas/ppcls/data/dataloader/logo_dataset.py +0 -46
- PaddleClas/ppcls/data/dataloader/mix_dataset.py +0 -49
- PaddleClas/ppcls/data/dataloader/mix_sampler.py +0 -79
- PaddleClas/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- PaddleClas/ppcls/data/dataloader/pk_sampler.py +0 -105
- PaddleClas/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- PaddleClas/ppcls/data/postprocess/__init__.py +0 -41
- PaddleClas/ppcls/data/postprocess/topk.py +0 -85
- PaddleClas/ppcls/data/preprocess/__init__.py +0 -100
- PaddleClas/ppcls/data/preprocess/batch_ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- PaddleClas/ppcls/data/preprocess/ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- PaddleClas/ppcls/data/preprocess/ops/cutout.py +0 -41
- PaddleClas/ppcls/data/preprocess/ops/fmix.py +0 -217
- PaddleClas/ppcls/data/preprocess/ops/functional.py +0 -138
- PaddleClas/ppcls/data/preprocess/ops/grid.py +0 -89
- PaddleClas/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- PaddleClas/ppcls/data/preprocess/ops/operators.py +0 -384
- PaddleClas/ppcls/data/preprocess/ops/randaugment.py +0 -106
- PaddleClas/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- PaddleClas/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- PaddleClas/ppcls/data/utils/__init__.py +0 -13
- PaddleClas/ppcls/data/utils/get_image_list.py +0 -49
- PaddleClas/ppcls/engine/__init__.py +0 -0
- PaddleClas/ppcls/engine/engine.py +0 -436
- PaddleClas/ppcls/engine/evaluation/__init__.py +0 -16
- PaddleClas/ppcls/engine/evaluation/classification.py +0 -143
- PaddleClas/ppcls/engine/evaluation/retrieval.py +0 -169
- PaddleClas/ppcls/engine/slim/__init__.py +0 -16
- PaddleClas/ppcls/engine/slim/prune.py +0 -66
- PaddleClas/ppcls/engine/slim/quant.py +0 -55
- PaddleClas/ppcls/engine/train/__init__.py +0 -14
- PaddleClas/ppcls/engine/train/train.py +0 -79
- PaddleClas/ppcls/engine/train/utils.py +0 -72
- PaddleClas/ppcls/loss/__init__.py +0 -65
- PaddleClas/ppcls/loss/celoss.py +0 -67
- PaddleClas/ppcls/loss/centerloss.py +0 -54
- PaddleClas/ppcls/loss/comfunc.py +0 -45
- PaddleClas/ppcls/loss/deephashloss.py +0 -92
- PaddleClas/ppcls/loss/distanceloss.py +0 -43
- PaddleClas/ppcls/loss/distillationloss.py +0 -141
- PaddleClas/ppcls/loss/dmlloss.py +0 -46
- PaddleClas/ppcls/loss/emlloss.py +0 -97
- PaddleClas/ppcls/loss/googlenetloss.py +0 -41
- PaddleClas/ppcls/loss/msmloss.py +0 -78
- PaddleClas/ppcls/loss/multilabelloss.py +0 -43
- PaddleClas/ppcls/loss/npairsloss.py +0 -38
- PaddleClas/ppcls/loss/pairwisecosface.py +0 -55
- PaddleClas/ppcls/loss/supconloss.py +0 -108
- PaddleClas/ppcls/loss/trihardloss.py +0 -82
- PaddleClas/ppcls/loss/triplet.py +0 -137
- PaddleClas/ppcls/metric/__init__.py +0 -51
- PaddleClas/ppcls/metric/metrics.py +0 -308
- PaddleClas/ppcls/optimizer/__init__.py +0 -72
- PaddleClas/ppcls/optimizer/learning_rate.py +0 -326
- PaddleClas/ppcls/optimizer/optimizer.py +0 -207
- PaddleClas/ppcls/utils/__init__.py +0 -27
- PaddleClas/ppcls/utils/check.py +0 -151
- PaddleClas/ppcls/utils/config.py +0 -210
- PaddleClas/ppcls/utils/download.py +0 -319
- PaddleClas/ppcls/utils/ema.py +0 -63
- PaddleClas/ppcls/utils/logger.py +0 -137
- PaddleClas/ppcls/utils/metrics.py +0 -107
- PaddleClas/ppcls/utils/misc.py +0 -63
- PaddleClas/ppcls/utils/model_zoo.py +0 -213
- PaddleClas/ppcls/utils/profiler.py +0 -111
- PaddleClas/ppcls/utils/save_load.py +0 -136
- PaddleClas/setup.py +0 -58
- PaddleClas/tools/__init__.py +0 -15
- PaddleClas/tools/eval.py +0 -31
- PaddleClas/tools/export_model.py +0 -34
- PaddleClas/tools/infer.py +0 -31
- PaddleClas/tools/train.py +0 -32
- paddlex/cls.py +0 -82
- paddlex/command.py +0 -215
- paddlex/cv/__init__.py +0 -17
- paddlex/cv/datasets/__init__.py +0 -18
- paddlex/cv/datasets/coco.py +0 -208
- paddlex/cv/datasets/imagenet.py +0 -88
- paddlex/cv/datasets/seg_dataset.py +0 -91
- paddlex/cv/datasets/voc.py +0 -445
- paddlex/cv/models/__init__.py +0 -18
- paddlex/cv/models/base.py +0 -631
- paddlex/cv/models/classifier.py +0 -989
- paddlex/cv/models/detector.py +0 -2292
- paddlex/cv/models/load_model.py +0 -148
- paddlex/cv/models/segmenter.py +0 -768
- 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 -476
- paddlex/cv/models/utils/det_metrics/metrics.py +0 -220
- paddlex/cv/models/utils/infer_nets.py +0 -45
- paddlex/cv/models/utils/seg_metrics.py +0 -62
- paddlex/cv/models/utils/visualize.py +0 -399
- 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/deploy.py +0 -268
- paddlex/det.py +0 -49
- 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/arch/__init__.py +0 -127
- paddlex/ppcls/arch/backbone/__init__.py +0 -80
- paddlex/ppcls/arch/backbone/base/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/base/theseus_layer.py +0 -130
- paddlex/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- paddlex/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- paddlex/ppcls/arch/backbone/legendary_models/hrnet.py +0 -748
- paddlex/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- paddlex/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- paddlex/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- paddlex/ppcls/arch/backbone/legendary_models/vgg.py +0 -235
- paddlex/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- paddlex/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- paddlex/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- paddlex/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- paddlex/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- paddlex/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- paddlex/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- paddlex/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- paddlex/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- paddlex/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- paddlex/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- paddlex/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- paddlex/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- paddlex/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- paddlex/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- paddlex/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- paddlex/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- paddlex/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- paddlex/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- paddlex/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- paddlex/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- paddlex/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- paddlex/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- paddlex/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- paddlex/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- paddlex/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- paddlex/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- paddlex/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- paddlex/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- paddlex/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- paddlex/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- paddlex/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- paddlex/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- paddlex/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- paddlex/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- paddlex/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- paddlex/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- paddlex/ppcls/arch/gears/__init__.py +0 -32
- paddlex/ppcls/arch/gears/arcmargin.py +0 -72
- paddlex/ppcls/arch/gears/circlemargin.py +0 -59
- paddlex/ppcls/arch/gears/cosmargin.py +0 -55
- paddlex/ppcls/arch/gears/fc.py +0 -35
- paddlex/ppcls/arch/gears/identity_head.py +0 -9
- paddlex/ppcls/arch/gears/vehicle_neck.py +0 -52
- paddlex/ppcls/arch/utils.py +0 -53
- paddlex/ppcls/data/__init__.py +0 -144
- paddlex/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- paddlex/ppcls/data/dataloader/__init__.py +0 -9
- paddlex/ppcls/data/dataloader/common_dataset.py +0 -84
- paddlex/ppcls/data/dataloader/dali.py +0 -319
- paddlex/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- paddlex/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- paddlex/ppcls/data/dataloader/logo_dataset.py +0 -46
- paddlex/ppcls/data/dataloader/mix_dataset.py +0 -49
- paddlex/ppcls/data/dataloader/mix_sampler.py +0 -79
- paddlex/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- paddlex/ppcls/data/dataloader/pk_sampler.py +0 -105
- paddlex/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- paddlex/ppcls/data/postprocess/__init__.py +0 -41
- paddlex/ppcls/data/postprocess/topk.py +0 -85
- paddlex/ppcls/data/preprocess/__init__.py +0 -100
- paddlex/ppcls/data/preprocess/batch_ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- paddlex/ppcls/data/preprocess/ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- paddlex/ppcls/data/preprocess/ops/cutout.py +0 -41
- paddlex/ppcls/data/preprocess/ops/fmix.py +0 -217
- paddlex/ppcls/data/preprocess/ops/functional.py +0 -141
- paddlex/ppcls/data/preprocess/ops/grid.py +0 -89
- paddlex/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- paddlex/ppcls/data/preprocess/ops/operators.py +0 -384
- paddlex/ppcls/data/preprocess/ops/randaugment.py +0 -106
- paddlex/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- paddlex/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- paddlex/ppcls/data/utils/__init__.py +0 -13
- paddlex/ppcls/data/utils/get_image_list.py +0 -49
- paddlex/ppcls/engine/__init__.py +0 -0
- paddlex/ppcls/engine/engine.py +0 -436
- paddlex/ppcls/engine/evaluation/__init__.py +0 -16
- paddlex/ppcls/engine/evaluation/classification.py +0 -143
- paddlex/ppcls/engine/evaluation/retrieval.py +0 -169
- paddlex/ppcls/engine/slim/__init__.py +0 -16
- paddlex/ppcls/engine/slim/prune.py +0 -66
- paddlex/ppcls/engine/slim/quant.py +0 -55
- paddlex/ppcls/engine/train/__init__.py +0 -14
- paddlex/ppcls/engine/train/train.py +0 -79
- paddlex/ppcls/engine/train/utils.py +0 -72
- paddlex/ppcls/loss/__init__.py +0 -65
- paddlex/ppcls/loss/celoss.py +0 -67
- paddlex/ppcls/loss/centerloss.py +0 -54
- paddlex/ppcls/loss/comfunc.py +0 -45
- paddlex/ppcls/loss/deephashloss.py +0 -96
- paddlex/ppcls/loss/distanceloss.py +0 -43
- paddlex/ppcls/loss/distillationloss.py +0 -141
- paddlex/ppcls/loss/dmlloss.py +0 -46
- paddlex/ppcls/loss/emlloss.py +0 -97
- paddlex/ppcls/loss/googlenetloss.py +0 -42
- paddlex/ppcls/loss/msmloss.py +0 -78
- paddlex/ppcls/loss/multilabelloss.py +0 -43
- paddlex/ppcls/loss/npairsloss.py +0 -38
- paddlex/ppcls/loss/pairwisecosface.py +0 -59
- paddlex/ppcls/loss/supconloss.py +0 -108
- paddlex/ppcls/loss/trihardloss.py +0 -82
- paddlex/ppcls/loss/triplet.py +0 -137
- paddlex/ppcls/metric/__init__.py +0 -51
- paddlex/ppcls/metric/metrics.py +0 -308
- paddlex/ppcls/optimizer/__init__.py +0 -72
- paddlex/ppcls/optimizer/learning_rate.py +0 -326
- paddlex/ppcls/optimizer/optimizer.py +0 -208
- paddlex/ppcls/utils/__init__.py +0 -27
- paddlex/ppcls/utils/check.py +0 -151
- paddlex/ppcls/utils/config.py +0 -210
- paddlex/ppcls/utils/download.py +0 -319
- paddlex/ppcls/utils/ema.py +0 -63
- paddlex/ppcls/utils/logger.py +0 -137
- paddlex/ppcls/utils/metrics.py +0 -112
- paddlex/ppcls/utils/misc.py +0 -63
- paddlex/ppcls/utils/model_zoo.py +0 -213
- paddlex/ppcls/utils/profiler.py +0 -111
- paddlex/ppcls/utils/save_load.py +0 -136
- 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 -278
- paddlex/ppdet/data/__init__.py +0 -21
- paddlex/ppdet/data/crop_utils/__init__.py +0 -13
- paddlex/ppdet/data/crop_utils/annotation_cropper.py +0 -585
- paddlex/ppdet/data/crop_utils/chip_box_utils.py +0 -170
- paddlex/ppdet/data/reader.py +0 -302
- paddlex/ppdet/data/shm_utils.py +0 -67
- paddlex/ppdet/data/source/__init__.py +0 -29
- paddlex/ppdet/data/source/category.py +0 -904
- paddlex/ppdet/data/source/coco.py +0 -251
- paddlex/ppdet/data/source/dataset.py +0 -197
- paddlex/ppdet/data/source/keypoint_coco.py +0 -669
- paddlex/ppdet/data/source/mot.py +0 -636
- paddlex/ppdet/data/source/sniper_coco.py +0 -191
- paddlex/ppdet/data/source/voc.py +0 -231
- paddlex/ppdet/data/source/widerface.py +0 -180
- paddlex/ppdet/data/transform/__init__.py +0 -28
- paddlex/ppdet/data/transform/atss_assigner.py +0 -270
- paddlex/ppdet/data/transform/autoaugment_utils.py +0 -1591
- paddlex/ppdet/data/transform/batch_operators.py +0 -1080
- paddlex/ppdet/data/transform/gridmask_utils.py +0 -86
- paddlex/ppdet/data/transform/keypoint_operators.py +0 -868
- paddlex/ppdet/data/transform/mot_operators.py +0 -628
- paddlex/ppdet/data/transform/op_helper.py +0 -498
- paddlex/ppdet/data/transform/operators.py +0 -3025
- paddlex/ppdet/engine/__init__.py +0 -30
- paddlex/ppdet/engine/callbacks.py +0 -340
- paddlex/ppdet/engine/env.py +0 -50
- paddlex/ppdet/engine/export_utils.py +0 -177
- paddlex/ppdet/engine/tracker.py +0 -538
- paddlex/ppdet/engine/trainer.py +0 -723
- paddlex/ppdet/metrics/__init__.py +0 -29
- paddlex/ppdet/metrics/coco_utils.py +0 -184
- paddlex/ppdet/metrics/json_results.py +0 -149
- paddlex/ppdet/metrics/keypoint_metrics.py +0 -401
- paddlex/ppdet/metrics/map_utils.py +0 -444
- paddlex/ppdet/metrics/mcmot_metrics.py +0 -470
- paddlex/ppdet/metrics/metrics.py +0 -434
- paddlex/ppdet/metrics/mot_metrics.py +0 -1236
- paddlex/ppdet/metrics/munkres.py +0 -428
- paddlex/ppdet/metrics/widerface_utils.py +0 -393
- paddlex/ppdet/model_zoo/__init__.py +0 -18
- paddlex/ppdet/model_zoo/model_zoo.py +0 -84
- paddlex/ppdet/modeling/__init__.py +0 -45
- paddlex/ppdet/modeling/architectures/__init__.py +0 -51
- paddlex/ppdet/modeling/architectures/blazeface.py +0 -91
- paddlex/ppdet/modeling/architectures/cascade_rcnn.py +0 -144
- paddlex/ppdet/modeling/architectures/centernet.py +0 -108
- paddlex/ppdet/modeling/architectures/deepsort.py +0 -69
- paddlex/ppdet/modeling/architectures/detr.py +0 -93
- paddlex/ppdet/modeling/architectures/fairmot.py +0 -100
- paddlex/ppdet/modeling/architectures/faster_rcnn.py +0 -106
- paddlex/ppdet/modeling/architectures/fcos.py +0 -105
- paddlex/ppdet/modeling/architectures/gfl.py +0 -87
- paddlex/ppdet/modeling/architectures/jde.py +0 -111
- paddlex/ppdet/modeling/architectures/keypoint_hrhrnet.py +0 -287
- paddlex/ppdet/modeling/architectures/keypoint_hrnet.py +0 -267
- paddlex/ppdet/modeling/architectures/mask_rcnn.py +0 -135
- paddlex/ppdet/modeling/architectures/meta_arch.py +0 -128
- paddlex/ppdet/modeling/architectures/picodet.py +0 -91
- paddlex/ppdet/modeling/architectures/s2anet.py +0 -102
- paddlex/ppdet/modeling/architectures/solov2.py +0 -110
- paddlex/ppdet/modeling/architectures/sparse_rcnn.py +0 -99
- paddlex/ppdet/modeling/architectures/ssd.py +0 -93
- paddlex/ppdet/modeling/architectures/tood.py +0 -78
- paddlex/ppdet/modeling/architectures/ttfnet.py +0 -98
- paddlex/ppdet/modeling/architectures/yolo.py +0 -124
- paddlex/ppdet/modeling/assigners/__init__.py +0 -23
- paddlex/ppdet/modeling/assigners/atss_assigner.py +0 -211
- paddlex/ppdet/modeling/assigners/simota_assigner.py +0 -262
- paddlex/ppdet/modeling/assigners/task_aligned_assigner.py +0 -158
- paddlex/ppdet/modeling/assigners/utils.py +0 -195
- paddlex/ppdet/modeling/backbones/__init__.py +0 -49
- paddlex/ppdet/modeling/backbones/blazenet.py +0 -323
- paddlex/ppdet/modeling/backbones/darknet.py +0 -340
- paddlex/ppdet/modeling/backbones/dla.py +0 -244
- paddlex/ppdet/modeling/backbones/esnet.py +0 -290
- paddlex/ppdet/modeling/backbones/ghostnet.py +0 -470
- paddlex/ppdet/modeling/backbones/hardnet.py +0 -224
- paddlex/ppdet/modeling/backbones/hrnet.py +0 -727
- paddlex/ppdet/modeling/backbones/lcnet.py +0 -259
- paddlex/ppdet/modeling/backbones/lite_hrnet.py +0 -886
- paddlex/ppdet/modeling/backbones/mobilenet_v1.py +0 -418
- paddlex/ppdet/modeling/backbones/mobilenet_v3.py +0 -483
- paddlex/ppdet/modeling/backbones/name_adapter.py +0 -69
- paddlex/ppdet/modeling/backbones/res2net.py +0 -358
- paddlex/ppdet/modeling/backbones/resnet.py +0 -613
- paddlex/ppdet/modeling/backbones/senet.py +0 -139
- paddlex/ppdet/modeling/backbones/shufflenet_v2.py +0 -246
- paddlex/ppdet/modeling/backbones/swin_transformer.py +0 -743
- paddlex/ppdet/modeling/backbones/vgg.py +0 -210
- paddlex/ppdet/modeling/bbox_utils.py +0 -778
- paddlex/ppdet/modeling/heads/__init__.py +0 -53
- paddlex/ppdet/modeling/heads/bbox_head.py +0 -377
- paddlex/ppdet/modeling/heads/cascade_head.py +0 -284
- paddlex/ppdet/modeling/heads/centernet_head.py +0 -292
- paddlex/ppdet/modeling/heads/detr_head.py +0 -368
- paddlex/ppdet/modeling/heads/face_head.py +0 -110
- paddlex/ppdet/modeling/heads/fcos_head.py +0 -259
- paddlex/ppdet/modeling/heads/gfl_head.py +0 -487
- paddlex/ppdet/modeling/heads/keypoint_hrhrnet_head.py +0 -108
- paddlex/ppdet/modeling/heads/mask_head.py +0 -250
- paddlex/ppdet/modeling/heads/pico_head.py +0 -278
- paddlex/ppdet/modeling/heads/roi_extractor.py +0 -111
- paddlex/ppdet/modeling/heads/s2anet_head.py +0 -1056
- paddlex/ppdet/modeling/heads/simota_head.py +0 -506
- paddlex/ppdet/modeling/heads/solov2_head.py +0 -560
- paddlex/ppdet/modeling/heads/sparsercnn_head.py +0 -375
- paddlex/ppdet/modeling/heads/ssd_head.py +0 -215
- paddlex/ppdet/modeling/heads/tood_head.py +0 -366
- paddlex/ppdet/modeling/heads/ttf_head.py +0 -316
- paddlex/ppdet/modeling/heads/yolo_head.py +0 -124
- paddlex/ppdet/modeling/initializer.py +0 -317
- paddlex/ppdet/modeling/keypoint_utils.py +0 -342
- paddlex/ppdet/modeling/layers.py +0 -1430
- paddlex/ppdet/modeling/losses/__init__.py +0 -43
- paddlex/ppdet/modeling/losses/ctfocal_loss.py +0 -68
- paddlex/ppdet/modeling/losses/detr_loss.py +0 -233
- paddlex/ppdet/modeling/losses/fairmot_loss.py +0 -41
- paddlex/ppdet/modeling/losses/fcos_loss.py +0 -225
- paddlex/ppdet/modeling/losses/gfocal_loss.py +0 -217
- paddlex/ppdet/modeling/losses/iou_aware_loss.py +0 -47
- paddlex/ppdet/modeling/losses/iou_loss.py +0 -210
- paddlex/ppdet/modeling/losses/jde_loss.py +0 -193
- paddlex/ppdet/modeling/losses/keypoint_loss.py +0 -229
- paddlex/ppdet/modeling/losses/solov2_loss.py +0 -101
- paddlex/ppdet/modeling/losses/sparsercnn_loss.py +0 -425
- paddlex/ppdet/modeling/losses/ssd_loss.py +0 -170
- paddlex/ppdet/modeling/losses/varifocal_loss.py +0 -152
- 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 -144
- 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 -297
- paddlex/ppdet/modeling/mot/tracker/base_sde_tracker.py +0 -156
- paddlex/ppdet/modeling/mot/tracker/deepsort_tracker.py +0 -188
- paddlex/ppdet/modeling/mot/tracker/jde_tracker.py +0 -277
- paddlex/ppdet/modeling/mot/utils.py +0 -263
- paddlex/ppdet/modeling/mot/visualization.py +0 -150
- paddlex/ppdet/modeling/necks/__init__.py +0 -30
- paddlex/ppdet/modeling/necks/bifpn.py +0 -302
- paddlex/ppdet/modeling/necks/blazeface_fpn.py +0 -216
- paddlex/ppdet/modeling/necks/centernet_fpn.py +0 -426
- paddlex/ppdet/modeling/necks/csp_pan.py +0 -364
- paddlex/ppdet/modeling/necks/fpn.py +0 -231
- paddlex/ppdet/modeling/necks/hrfpn.py +0 -126
- paddlex/ppdet/modeling/necks/ttf_fpn.py +0 -242
- paddlex/ppdet/modeling/necks/yolo_fpn.py +0 -988
- paddlex/ppdet/modeling/ops.py +0 -1611
- paddlex/ppdet/modeling/post_process.py +0 -731
- 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 -77
- paddlex/ppdet/modeling/proposal_generator/rpn_head.py +0 -260
- paddlex/ppdet/modeling/proposal_generator/target.py +0 -681
- paddlex/ppdet/modeling/proposal_generator/target_layer.py +0 -491
- paddlex/ppdet/modeling/reid/__init__.py +0 -25
- paddlex/ppdet/modeling/reid/fairmot_embedding_head.py +0 -225
- paddlex/ppdet/modeling/reid/jde_embedding_head.py +0 -214
- paddlex/ppdet/modeling/reid/pplcnet_embedding.py +0 -282
- paddlex/ppdet/modeling/reid/pyramidal_embedding.py +0 -144
- paddlex/ppdet/modeling/reid/resnet.py +0 -310
- paddlex/ppdet/modeling/shape_spec.py +0 -25
- paddlex/ppdet/modeling/transformers/__init__.py +0 -25
- paddlex/ppdet/modeling/transformers/deformable_transformer.py +0 -517
- paddlex/ppdet/modeling/transformers/detr_transformer.py +0 -353
- paddlex/ppdet/modeling/transformers/matchers.py +0 -127
- paddlex/ppdet/modeling/transformers/position_encoding.py +0 -108
- paddlex/ppdet/modeling/transformers/utils.py +0 -110
- paddlex/ppdet/optimizer.py +0 -335
- paddlex/ppdet/slim/__init__.py +0 -82
- paddlex/ppdet/slim/distill.py +0 -110
- paddlex/ppdet/slim/prune.py +0 -85
- paddlex/ppdet/slim/quant.py +0 -84
- paddlex/ppdet/slim/unstructured_prune.py +0 -66
- paddlex/ppdet/utils/__init__.py +0 -13
- paddlex/ppdet/utils/check.py +0 -112
- paddlex/ppdet/utils/checkpoint.py +0 -226
- paddlex/ppdet/utils/cli.py +0 -151
- paddlex/ppdet/utils/colormap.py +0 -58
- paddlex/ppdet/utils/download.py +0 -558
- paddlex/ppdet/utils/logger.py +0 -70
- paddlex/ppdet/utils/profiler.py +0 -111
- paddlex/ppdet/utils/stats.py +0 -94
- paddlex/ppdet/utils/visualizer.py +0 -321
- paddlex/ppdet/utils/voc_utils.py +0 -86
- paddlex/seg.py +0 -41
- paddlex/tools/__init__.py +0 -17
- paddlex/tools/anchor_clustering/__init__.py +0 -15
- paddlex/tools/anchor_clustering/yolo_cluster.py +0 -178
- 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 -492
- paddlex/utils/shm.py +0 -67
- paddlex/utils/stats.py +0 -68
- paddlex/utils/utils.py +0 -229
- paddlex-2.1.0.data/data/paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex-2.1.0.dist-info/LICENSE +0 -201
- paddlex-2.1.0.dist-info/METADATA +0 -32
- paddlex-2.1.0.dist-info/RECORD +0 -698
- paddlex-2.1.0.dist-info/WHEEL +0 -5
- paddlex-2.1.0.dist-info/entry_points.txt +0 -4
- paddlex-2.1.0.dist-info/top_level.txt +0 -3
- paddlex_restful/__init__.py +0 -15
- paddlex_restful/command.py +0 -63
- paddlex_restful/restful/__init__.py +0 -15
- paddlex_restful/restful/app.py +0 -969
- paddlex_restful/restful/dataset/__init__.py +0 -13
- paddlex_restful/restful/dataset/cls_dataset.py +0 -159
- paddlex_restful/restful/dataset/dataset.py +0 -266
- paddlex_restful/restful/dataset/datasetbase.py +0 -86
- paddlex_restful/restful/dataset/det_dataset.py +0 -190
- paddlex_restful/restful/dataset/ins_seg_dataset.py +0 -312
- paddlex_restful/restful/dataset/operate.py +0 -155
- paddlex_restful/restful/dataset/seg_dataset.py +0 -222
- paddlex_restful/restful/dataset/utils.py +0 -267
- paddlex_restful/restful/demo.py +0 -202
- paddlex_restful/restful/dir.py +0 -45
- paddlex_restful/restful/model.py +0 -312
- paddlex_restful/restful/project/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/classification.py +0 -126
- paddlex_restful/restful/project/evaluate/detection.py +0 -789
- paddlex_restful/restful/project/evaluate/draw_pred_result.py +0 -181
- paddlex_restful/restful/project/evaluate/segmentation.py +0 -122
- paddlex_restful/restful/project/operate.py +0 -931
- paddlex_restful/restful/project/project.py +0 -143
- paddlex_restful/restful/project/prune/__init__.py +0 -13
- paddlex_restful/restful/project/prune/classification.py +0 -32
- paddlex_restful/restful/project/prune/detection.py +0 -48
- paddlex_restful/restful/project/prune/segmentation.py +0 -34
- paddlex_restful/restful/project/task.py +0 -884
- paddlex_restful/restful/project/train/__init__.py +0 -13
- paddlex_restful/restful/project/train/classification.py +0 -141
- paddlex_restful/restful/project/train/detection.py +0 -263
- paddlex_restful/restful/project/train/params.py +0 -432
- paddlex_restful/restful/project/train/params_v2.py +0 -326
- paddlex_restful/restful/project/train/segmentation.py +0 -191
- paddlex_restful/restful/project/visualize.py +0 -244
- paddlex_restful/restful/system.py +0 -102
- paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex_restful/restful/utils.py +0 -841
- paddlex_restful/restful/workspace.py +0 -343
- paddlex_restful/restful/workspace_pb2.py +0 -1411
|
@@ -0,0 +1,745 @@
|
|
|
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 math
|
|
18
|
+
import re
|
|
19
|
+
from functools import partial
|
|
20
|
+
from typing import List
|
|
21
|
+
|
|
22
|
+
import numpy as np
|
|
23
|
+
from PIL import Image, ImageDraw, ImageFont
|
|
24
|
+
|
|
25
|
+
from ....utils.fonts import PINGFANG_FONT_FILE_PATH
|
|
26
|
+
from ...common.result import (
|
|
27
|
+
BaseCVResult,
|
|
28
|
+
HtmlMixin,
|
|
29
|
+
JsonMixin,
|
|
30
|
+
MarkdownMixin,
|
|
31
|
+
XlsxMixin,
|
|
32
|
+
)
|
|
33
|
+
from .setting import BLOCK_LABEL_MAP
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
def compile_title_pattern():
|
|
37
|
+
# Precompiled regex pattern for matching numbering at the beginning of the title
|
|
38
|
+
numbering_pattern = (
|
|
39
|
+
r"(?:" + r"[1-9][0-9]*(?:\.[1-9][0-9]*)*[\.、]?|" + r"[\(\（](?:[1-9][0-9]*|["
|
|
40
|
+
r"一二三四五六七八九十百千万亿零壹贰叁肆伍陆柒捌玖拾]+)[\)\）]|" + r"["
|
|
41
|
+
r"一二三四五六七八九十百千万亿零壹贰叁肆伍陆柒捌玖拾]+"
|
|
42
|
+
r"[、\.]?|" + r"(?:I|II|III|IV|V|VI|VII|VIII|IX|X)\.?" + r")"
|
|
43
|
+
)
|
|
44
|
+
return re.compile(r"^\s*(" + numbering_pattern + r")(\s*)(.*)$")
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
TITLE_RE_PATTERN = compile_title_pattern()
|
|
48
|
+
|
|
49
|
+
|
|
50
|
+
def format_title_func(block):
|
|
51
|
+
"""
|
|
52
|
+
Normalize chapter title.
|
|
53
|
+
Add the '#' to indicate the level of the title.
|
|
54
|
+
If numbering exists, ensure there's exactly one space between it and the title content.
|
|
55
|
+
If numbering does not exist, return the original title unchanged.
|
|
56
|
+
|
|
57
|
+
:param title: Original chapter title string.
|
|
58
|
+
:return: Normalized chapter title string.
|
|
59
|
+
"""
|
|
60
|
+
title = block.content
|
|
61
|
+
match = TITLE_RE_PATTERN.match(title)
|
|
62
|
+
if match:
|
|
63
|
+
numbering = match.group(1).strip()
|
|
64
|
+
title_content = match.group(3).lstrip()
|
|
65
|
+
# Return numbering and title content separated by one space
|
|
66
|
+
title = numbering + " " + title_content
|
|
67
|
+
|
|
68
|
+
title = title.rstrip(".")
|
|
69
|
+
level = (
|
|
70
|
+
title.count(
|
|
71
|
+
".",
|
|
72
|
+
)
|
|
73
|
+
+ 1
|
|
74
|
+
if "." in title
|
|
75
|
+
else 1
|
|
76
|
+
)
|
|
77
|
+
return f"#{'#' * level} {title}".replace("-\n", "").replace(
|
|
78
|
+
"\n",
|
|
79
|
+
" ",
|
|
80
|
+
)
|
|
81
|
+
|
|
82
|
+
|
|
83
|
+
def format_centered_by_html(string):
|
|
84
|
+
return (
|
|
85
|
+
f'<div style="text-align: center;">{string}</div>'.replace(
|
|
86
|
+
"-\n",
|
|
87
|
+
"",
|
|
88
|
+
).replace("\n", " ")
|
|
89
|
+
+ "\n"
|
|
90
|
+
)
|
|
91
|
+
|
|
92
|
+
|
|
93
|
+
def format_text_plain_func(block):
|
|
94
|
+
return block.content
|
|
95
|
+
|
|
96
|
+
|
|
97
|
+
def format_image_scaled_by_html_func(block, original_image_width):
|
|
98
|
+
img_tags = []
|
|
99
|
+
image_path = block.image["path"]
|
|
100
|
+
image_width = block.image["img"].width
|
|
101
|
+
scale = int(image_width / original_image_width * 100)
|
|
102
|
+
img_tags.append(
|
|
103
|
+
'<img src="{}" alt="Image" width="{}%" />'.format(
|
|
104
|
+
image_path.replace("-\n", "").replace("\n", " "), scale
|
|
105
|
+
),
|
|
106
|
+
)
|
|
107
|
+
return "\n".join(img_tags)
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
def format_image_plain_func(block):
|
|
111
|
+
img_tags = []
|
|
112
|
+
image_path = block.image["path"]
|
|
113
|
+
img_tags.append("".format(image_path.replace("-\n", "").replace("\n", " ")))
|
|
114
|
+
return "\n".join(img_tags)
|
|
115
|
+
|
|
116
|
+
|
|
117
|
+
def format_chart2table_func(block):
|
|
118
|
+
lines_list = block.content.split("\n")
|
|
119
|
+
column_num = len(lines_list[0].split("|"))
|
|
120
|
+
lines_list.insert(1, "|".join(["---"] * column_num))
|
|
121
|
+
lines_list = [f"|{line}|" for line in lines_list]
|
|
122
|
+
return "\n".join(lines_list)
|
|
123
|
+
|
|
124
|
+
|
|
125
|
+
def simplify_table_func(table_code):
|
|
126
|
+
return "\n" + table_code.replace("<html>", "").replace("</html>", "").replace(
|
|
127
|
+
"<body>", ""
|
|
128
|
+
).replace("</body>", "")
|
|
129
|
+
|
|
130
|
+
|
|
131
|
+
def format_first_line_func(block, templates, format_func, spliter):
|
|
132
|
+
lines = block.content.split(spliter)
|
|
133
|
+
for idx in range(len(lines)):
|
|
134
|
+
line = lines[idx]
|
|
135
|
+
if line.strip() == "":
|
|
136
|
+
continue
|
|
137
|
+
if line.lower() in templates:
|
|
138
|
+
lines[idx] = format_func(line)
|
|
139
|
+
break
|
|
140
|
+
return spliter.join(lines)
|
|
141
|
+
|
|
142
|
+
|
|
143
|
+
def get_seg_flag(block: LayoutParsingBlock, prev_block: LayoutParsingBlock):
|
|
144
|
+
|
|
145
|
+
seg_start_flag = True
|
|
146
|
+
seg_end_flag = True
|
|
147
|
+
|
|
148
|
+
block_box = block.bbox
|
|
149
|
+
context_left_coordinate = block_box[0]
|
|
150
|
+
context_right_coordinate = block_box[2]
|
|
151
|
+
seg_start_coordinate = block.seg_start_coordinate
|
|
152
|
+
seg_end_coordinate = block.seg_end_coordinate
|
|
153
|
+
|
|
154
|
+
if prev_block is not None:
|
|
155
|
+
prev_block_bbox = prev_block.bbox
|
|
156
|
+
num_of_prev_lines = prev_block.num_of_lines
|
|
157
|
+
pre_block_seg_end_coordinate = prev_block.seg_end_coordinate
|
|
158
|
+
prev_end_space_small = (
|
|
159
|
+
abs(prev_block_bbox[2] - pre_block_seg_end_coordinate) < 10
|
|
160
|
+
)
|
|
161
|
+
prev_lines_more_than_one = num_of_prev_lines > 1
|
|
162
|
+
|
|
163
|
+
overlap_blocks = context_left_coordinate < prev_block_bbox[2]
|
|
164
|
+
|
|
165
|
+
# update context_left_coordinate and context_right_coordinate
|
|
166
|
+
if overlap_blocks:
|
|
167
|
+
context_left_coordinate = min(prev_block_bbox[0], context_left_coordinate)
|
|
168
|
+
context_right_coordinate = max(prev_block_bbox[2], context_right_coordinate)
|
|
169
|
+
prev_end_space_small = (
|
|
170
|
+
abs(context_right_coordinate - pre_block_seg_end_coordinate) < 10
|
|
171
|
+
)
|
|
172
|
+
edge_distance = 0
|
|
173
|
+
else:
|
|
174
|
+
edge_distance = abs(block_box[0] - prev_block_bbox[2])
|
|
175
|
+
|
|
176
|
+
current_start_space_small = seg_start_coordinate - context_left_coordinate < 10
|
|
177
|
+
|
|
178
|
+
if (
|
|
179
|
+
prev_end_space_small
|
|
180
|
+
and current_start_space_small
|
|
181
|
+
and prev_lines_more_than_one
|
|
182
|
+
and edge_distance < max(prev_block.width, block.width)
|
|
183
|
+
):
|
|
184
|
+
seg_start_flag = False
|
|
185
|
+
else:
|
|
186
|
+
if seg_start_coordinate - context_left_coordinate < 10:
|
|
187
|
+
seg_start_flag = False
|
|
188
|
+
|
|
189
|
+
if context_right_coordinate - seg_end_coordinate < 10:
|
|
190
|
+
seg_end_flag = False
|
|
191
|
+
|
|
192
|
+
return seg_start_flag, seg_end_flag
|
|
193
|
+
|
|
194
|
+
|
|
195
|
+
class LayoutParsingResultV2(BaseCVResult, HtmlMixin, XlsxMixin, MarkdownMixin):
|
|
196
|
+
"""Layout Parsing Result V2"""
|
|
197
|
+
|
|
198
|
+
def __init__(self, data) -> None:
|
|
199
|
+
"""Initializes a new instance of the class with the specified data."""
|
|
200
|
+
super().__init__(data)
|
|
201
|
+
HtmlMixin.__init__(self)
|
|
202
|
+
XlsxMixin.__init__(self)
|
|
203
|
+
MarkdownMixin.__init__(self)
|
|
204
|
+
JsonMixin.__init__(self)
|
|
205
|
+
|
|
206
|
+
def _to_img(self) -> dict[str, np.ndarray]:
|
|
207
|
+
from .utils import get_show_color
|
|
208
|
+
|
|
209
|
+
res_img_dict = {}
|
|
210
|
+
model_settings = self["model_settings"]
|
|
211
|
+
if model_settings["use_doc_preprocessor"]:
|
|
212
|
+
for key, value in self["doc_preprocessor_res"].img.items():
|
|
213
|
+
res_img_dict[key] = value
|
|
214
|
+
res_img_dict["layout_det_res"] = self["layout_det_res"].img["res"]
|
|
215
|
+
|
|
216
|
+
if model_settings["use_region_detection"]:
|
|
217
|
+
res_img_dict["region_det_res"] = self["region_det_res"].img["res"]
|
|
218
|
+
|
|
219
|
+
res_img_dict["overall_ocr_res"] = self["overall_ocr_res"].img["ocr_res_img"]
|
|
220
|
+
|
|
221
|
+
if model_settings["use_table_recognition"] and len(self["table_res_list"]) > 0:
|
|
222
|
+
table_cell_img = Image.fromarray(
|
|
223
|
+
copy.deepcopy(self["doc_preprocessor_res"]["output_img"])
|
|
224
|
+
)
|
|
225
|
+
table_draw = ImageDraw.Draw(table_cell_img)
|
|
226
|
+
rectangle_color = (255, 0, 0)
|
|
227
|
+
for sno in range(len(self["table_res_list"])):
|
|
228
|
+
table_res = self["table_res_list"][sno]
|
|
229
|
+
cell_box_list = table_res["cell_box_list"]
|
|
230
|
+
for box in cell_box_list:
|
|
231
|
+
x1, y1, x2, y2 = [int(pos) for pos in box]
|
|
232
|
+
table_draw.rectangle(
|
|
233
|
+
[x1, y1, x2, y2], outline=rectangle_color, width=2
|
|
234
|
+
)
|
|
235
|
+
res_img_dict["table_cell_img"] = table_cell_img
|
|
236
|
+
|
|
237
|
+
if model_settings["use_seal_recognition"] and len(self["seal_res_list"]) > 0:
|
|
238
|
+
for sno in range(len(self["seal_res_list"])):
|
|
239
|
+
seal_res = self["seal_res_list"][sno]
|
|
240
|
+
seal_region_id = seal_res["seal_region_id"]
|
|
241
|
+
sub_seal_res_dict = seal_res.img
|
|
242
|
+
key = f"seal_res_region{seal_region_id}"
|
|
243
|
+
res_img_dict[key] = sub_seal_res_dict["ocr_res_img"]
|
|
244
|
+
|
|
245
|
+
# for layout ordering image
|
|
246
|
+
image = Image.fromarray(self["doc_preprocessor_res"]["output_img"][:, :, ::-1])
|
|
247
|
+
draw = ImageDraw.Draw(image, "RGBA")
|
|
248
|
+
font_size = int(0.018 * int(image.width)) + 2
|
|
249
|
+
font = ImageFont.truetype(PINGFANG_FONT_FILE_PATH, font_size, encoding="utf-8")
|
|
250
|
+
parsing_result: List[LayoutParsingBlock] = self["parsing_res_list"]
|
|
251
|
+
for block in parsing_result:
|
|
252
|
+
bbox = block.bbox
|
|
253
|
+
index = block.order_index
|
|
254
|
+
label = block.label
|
|
255
|
+
fill_color = get_show_color(label, False)
|
|
256
|
+
draw.rectangle(bbox, fill=fill_color)
|
|
257
|
+
if index is not None:
|
|
258
|
+
text_position = (bbox[2] + 2, bbox[1] - font_size // 2)
|
|
259
|
+
if int(image.width) - bbox[2] < font_size:
|
|
260
|
+
text_position = (
|
|
261
|
+
int(bbox[2] - font_size * 1.1),
|
|
262
|
+
bbox[1] - font_size // 2,
|
|
263
|
+
)
|
|
264
|
+
draw.text(text_position, str(index), font=font, fill="red")
|
|
265
|
+
|
|
266
|
+
res_img_dict["layout_order_res"] = image
|
|
267
|
+
|
|
268
|
+
return res_img_dict
|
|
269
|
+
|
|
270
|
+
def _to_str(self, *args, **kwargs) -> dict[str, str]:
|
|
271
|
+
"""Converts the instance's attributes to a dictionary and then to a string.
|
|
272
|
+
|
|
273
|
+
Args:
|
|
274
|
+
*args: Additional positional arguments passed to the base class method.
|
|
275
|
+
**kwargs: Additional keyword arguments passed to the base class method.
|
|
276
|
+
|
|
277
|
+
Returns:
|
|
278
|
+
Dict[str, str]: A dictionary with the instance's attributes converted to strings.
|
|
279
|
+
"""
|
|
280
|
+
data = {}
|
|
281
|
+
data["input_path"] = self["input_path"]
|
|
282
|
+
data["page_index"] = self["page_index"]
|
|
283
|
+
model_settings = self["model_settings"]
|
|
284
|
+
data["model_settings"] = model_settings
|
|
285
|
+
if self["model_settings"]["use_doc_preprocessor"]:
|
|
286
|
+
data["doc_preprocessor_res"] = self["doc_preprocessor_res"].str["res"]
|
|
287
|
+
data["layout_det_res"] = self["layout_det_res"].str["res"]
|
|
288
|
+
data["overall_ocr_res"] = self["overall_ocr_res"].str["res"]
|
|
289
|
+
if model_settings["use_table_recognition"] and len(self["table_res_list"]) > 0:
|
|
290
|
+
data["table_res_list"] = []
|
|
291
|
+
for sno in range(len(self["table_res_list"])):
|
|
292
|
+
table_res = self["table_res_list"][sno]
|
|
293
|
+
data["table_res_list"].append(table_res.str["res"])
|
|
294
|
+
if model_settings["use_seal_recognition"] and len(self["seal_res_list"]) > 0:
|
|
295
|
+
data["seal_res_list"] = []
|
|
296
|
+
for sno in range(len(self["seal_res_list"])):
|
|
297
|
+
seal_res = self["seal_res_list"][sno]
|
|
298
|
+
data["seal_res_list"].append(seal_res.str["res"])
|
|
299
|
+
if (
|
|
300
|
+
model_settings["use_formula_recognition"]
|
|
301
|
+
and len(self["formula_res_list"]) > 0
|
|
302
|
+
):
|
|
303
|
+
data["formula_res_list"] = []
|
|
304
|
+
for sno in range(len(self["formula_res_list"])):
|
|
305
|
+
formula_res = self["formula_res_list"][sno]
|
|
306
|
+
data["formula_res_list"].append(formula_res.str["res"])
|
|
307
|
+
|
|
308
|
+
return JsonMixin._to_str(data, *args, **kwargs)
|
|
309
|
+
|
|
310
|
+
def _to_json(self, *args, **kwargs) -> dict[str, str]:
|
|
311
|
+
"""
|
|
312
|
+
Converts the object's data to a JSON dictionary.
|
|
313
|
+
|
|
314
|
+
Args:
|
|
315
|
+
*args: Positional arguments passed to the JsonMixin._to_json method.
|
|
316
|
+
**kwargs: Keyword arguments passed to the JsonMixin._to_json method.
|
|
317
|
+
|
|
318
|
+
Returns:
|
|
319
|
+
Dict[str, str]: A dictionary containing the object's data in JSON format.
|
|
320
|
+
"""
|
|
321
|
+
data = {}
|
|
322
|
+
data["input_path"] = self["input_path"]
|
|
323
|
+
data["page_index"] = self["page_index"]
|
|
324
|
+
model_settings = self["model_settings"]
|
|
325
|
+
data["model_settings"] = model_settings
|
|
326
|
+
parsing_res_list = self["parsing_res_list"]
|
|
327
|
+
parsing_res_list = [
|
|
328
|
+
{
|
|
329
|
+
"block_label": parsing_res.label,
|
|
330
|
+
"block_content": parsing_res.content,
|
|
331
|
+
"block_bbox": parsing_res.bbox,
|
|
332
|
+
}
|
|
333
|
+
for parsing_res in parsing_res_list
|
|
334
|
+
]
|
|
335
|
+
data["parsing_res_list"] = parsing_res_list
|
|
336
|
+
if self["model_settings"]["use_doc_preprocessor"]:
|
|
337
|
+
data["doc_preprocessor_res"] = self["doc_preprocessor_res"].json["res"]
|
|
338
|
+
data["layout_det_res"] = self["layout_det_res"].json["res"]
|
|
339
|
+
data["overall_ocr_res"] = self["overall_ocr_res"].json["res"]
|
|
340
|
+
if model_settings["use_table_recognition"] and len(self["table_res_list"]) > 0:
|
|
341
|
+
data["table_res_list"] = []
|
|
342
|
+
for sno in range(len(self["table_res_list"])):
|
|
343
|
+
table_res = self["table_res_list"][sno]
|
|
344
|
+
data["table_res_list"].append(table_res.json["res"])
|
|
345
|
+
if model_settings["use_seal_recognition"] and len(self["seal_res_list"]) > 0:
|
|
346
|
+
data["seal_res_list"] = []
|
|
347
|
+
for sno in range(len(self["seal_res_list"])):
|
|
348
|
+
seal_res = self["seal_res_list"][sno]
|
|
349
|
+
data["seal_res_list"].append(seal_res.json["res"])
|
|
350
|
+
if (
|
|
351
|
+
model_settings["use_formula_recognition"]
|
|
352
|
+
and len(self["formula_res_list"]) > 0
|
|
353
|
+
):
|
|
354
|
+
data["formula_res_list"] = []
|
|
355
|
+
for sno in range(len(self["formula_res_list"])):
|
|
356
|
+
formula_res = self["formula_res_list"][sno]
|
|
357
|
+
data["formula_res_list"].append(formula_res.json["res"])
|
|
358
|
+
return JsonMixin._to_json(data, *args, **kwargs)
|
|
359
|
+
|
|
360
|
+
def _to_html(self) -> dict[str, str]:
|
|
361
|
+
"""Converts the prediction to its corresponding HTML representation.
|
|
362
|
+
|
|
363
|
+
Returns:
|
|
364
|
+
Dict[str, str]: The str type HTML representation result.
|
|
365
|
+
"""
|
|
366
|
+
model_settings = self["model_settings"]
|
|
367
|
+
res_html_dict = {}
|
|
368
|
+
if model_settings["use_table_recognition"] and len(self["table_res_list"]) > 0:
|
|
369
|
+
for sno in range(len(self["table_res_list"])):
|
|
370
|
+
table_res = self["table_res_list"][sno]
|
|
371
|
+
table_region_id = table_res["table_region_id"]
|
|
372
|
+
key = f"table_{table_region_id}"
|
|
373
|
+
res_html_dict[key] = table_res.html["pred"]
|
|
374
|
+
return res_html_dict
|
|
375
|
+
|
|
376
|
+
def _to_xlsx(self) -> dict[str, str]:
|
|
377
|
+
"""Converts the prediction HTML to an XLSX file path.
|
|
378
|
+
|
|
379
|
+
Returns:
|
|
380
|
+
Dict[str, str]: The str type XLSX representation result.
|
|
381
|
+
"""
|
|
382
|
+
model_settings = self["model_settings"]
|
|
383
|
+
res_xlsx_dict = {}
|
|
384
|
+
if model_settings["use_table_recognition"] and len(self["table_res_list"]) > 0:
|
|
385
|
+
for sno in range(len(self["table_res_list"])):
|
|
386
|
+
table_res = self["table_res_list"][sno]
|
|
387
|
+
table_region_id = table_res["table_region_id"]
|
|
388
|
+
key = f"table_{table_region_id}"
|
|
389
|
+
res_xlsx_dict[key] = table_res.xlsx["pred"]
|
|
390
|
+
return res_xlsx_dict
|
|
391
|
+
|
|
392
|
+
def _to_markdown(self, pretty=True) -> dict:
|
|
393
|
+
"""
|
|
394
|
+
Save the parsing result to a Markdown file.
|
|
395
|
+
|
|
396
|
+
Args:
|
|
397
|
+
pretty (Optional[bool]): whether to pretty markdown by HTML, default by True.
|
|
398
|
+
|
|
399
|
+
Returns:
|
|
400
|
+
Dict
|
|
401
|
+
"""
|
|
402
|
+
original_image_width = self["doc_preprocessor_res"]["output_img"].shape[1]
|
|
403
|
+
|
|
404
|
+
if pretty:
|
|
405
|
+
format_text_func = lambda block: format_centered_by_html(
|
|
406
|
+
format_text_plain_func(block)
|
|
407
|
+
)
|
|
408
|
+
format_image_func = lambda block: format_centered_by_html(
|
|
409
|
+
format_image_scaled_by_html_func(
|
|
410
|
+
block,
|
|
411
|
+
original_image_width=original_image_width,
|
|
412
|
+
)
|
|
413
|
+
)
|
|
414
|
+
else:
|
|
415
|
+
format_text_func = lambda block: block.content
|
|
416
|
+
format_image_func = format_image_plain_func
|
|
417
|
+
|
|
418
|
+
if self["model_settings"].get("use_chart_recognition", False):
|
|
419
|
+
format_chart_func = format_chart2table_func
|
|
420
|
+
else:
|
|
421
|
+
format_chart_func = format_image_func
|
|
422
|
+
|
|
423
|
+
if self["model_settings"].get("use_seal_recognition", False):
|
|
424
|
+
format_seal_func = lambda block: "\n".join(
|
|
425
|
+
[format_image_func(block), format_text_func(block)]
|
|
426
|
+
)
|
|
427
|
+
else:
|
|
428
|
+
format_seal_func = format_image_func
|
|
429
|
+
|
|
430
|
+
if self["model_settings"].get("use_table_recognition", False):
|
|
431
|
+
if pretty:
|
|
432
|
+
format_table_func = lambda block: "\n" + format_text_func(
|
|
433
|
+
block
|
|
434
|
+
).replace("<table>", '<table border="1">')
|
|
435
|
+
else:
|
|
436
|
+
format_table_func = lambda block: simplify_table_func(
|
|
437
|
+
"\n" + block.content
|
|
438
|
+
)
|
|
439
|
+
else:
|
|
440
|
+
format_table_func = format_image_func
|
|
441
|
+
|
|
442
|
+
if self["model_settings"].get("use_formula_recognition", False):
|
|
443
|
+
format_formula_func = lambda block: f"$${block.content}$$"
|
|
444
|
+
else:
|
|
445
|
+
format_formula_func = format_image_func
|
|
446
|
+
|
|
447
|
+
handle_funcs_dict = {
|
|
448
|
+
"paragraph_title": format_title_func,
|
|
449
|
+
"abstract_title": format_title_func,
|
|
450
|
+
"reference_title": format_title_func,
|
|
451
|
+
"content_title": format_title_func,
|
|
452
|
+
"doc_title": lambda block: f"# {block.content}".replace(
|
|
453
|
+
"-\n",
|
|
454
|
+
"",
|
|
455
|
+
).replace("\n", " "),
|
|
456
|
+
"table_title": format_text_func,
|
|
457
|
+
"figure_title": format_text_func,
|
|
458
|
+
"chart_title": format_text_func,
|
|
459
|
+
"text": lambda block: block.content.replace("\n\n", "\n").replace(
|
|
460
|
+
"\n", "\n\n"
|
|
461
|
+
),
|
|
462
|
+
"abstract": partial(
|
|
463
|
+
format_first_line_func,
|
|
464
|
+
templates=["摘要", "abstract"],
|
|
465
|
+
format_func=lambda l: f"## {l}\n",
|
|
466
|
+
spliter=" ",
|
|
467
|
+
),
|
|
468
|
+
"content": lambda block: block.content.replace("-\n", " \n").replace(
|
|
469
|
+
"\n", " \n"
|
|
470
|
+
),
|
|
471
|
+
"image": format_image_func,
|
|
472
|
+
"chart": format_chart_func,
|
|
473
|
+
"formula": format_formula_func,
|
|
474
|
+
"table": format_table_func,
|
|
475
|
+
"reference": partial(
|
|
476
|
+
format_first_line_func,
|
|
477
|
+
templates=["参考文献", "references"],
|
|
478
|
+
format_func=lambda l: f"## {l}",
|
|
479
|
+
spliter="\n",
|
|
480
|
+
),
|
|
481
|
+
"algorithm": lambda block: block.content.strip("\n"),
|
|
482
|
+
"seal": format_seal_func,
|
|
483
|
+
}
|
|
484
|
+
|
|
485
|
+
markdown_content = ""
|
|
486
|
+
last_label = None
|
|
487
|
+
seg_start_flag = None
|
|
488
|
+
seg_end_flag = None
|
|
489
|
+
prev_block = None
|
|
490
|
+
page_first_element_seg_start_flag = None
|
|
491
|
+
page_last_element_seg_end_flag = None
|
|
492
|
+
markdown_info = {}
|
|
493
|
+
markdown_info["markdown_images"] = {}
|
|
494
|
+
for block in self["parsing_res_list"]:
|
|
495
|
+
seg_start_flag, seg_end_flag = get_seg_flag(block, prev_block)
|
|
496
|
+
|
|
497
|
+
label = block.label
|
|
498
|
+
if block.image is not None:
|
|
499
|
+
markdown_info["markdown_images"][block.image["path"]] = block.image[
|
|
500
|
+
"img"
|
|
501
|
+
]
|
|
502
|
+
page_first_element_seg_start_flag = (
|
|
503
|
+
seg_start_flag
|
|
504
|
+
if (page_first_element_seg_start_flag is None)
|
|
505
|
+
else page_first_element_seg_start_flag
|
|
506
|
+
)
|
|
507
|
+
|
|
508
|
+
handle_func = handle_funcs_dict.get(label, None)
|
|
509
|
+
if handle_func:
|
|
510
|
+
prev_block = block
|
|
511
|
+
if label == last_label == "text" and seg_start_flag == False:
|
|
512
|
+
markdown_content += handle_func(block)
|
|
513
|
+
else:
|
|
514
|
+
markdown_content += (
|
|
515
|
+
"\n\n" + handle_func(block)
|
|
516
|
+
if markdown_content
|
|
517
|
+
else handle_func(block)
|
|
518
|
+
)
|
|
519
|
+
last_label = label
|
|
520
|
+
page_last_element_seg_end_flag = seg_end_flag
|
|
521
|
+
|
|
522
|
+
markdown_info["markdown_texts"] = markdown_content
|
|
523
|
+
markdown_info["page_continuation_flags"] = (
|
|
524
|
+
page_first_element_seg_start_flag,
|
|
525
|
+
page_last_element_seg_end_flag,
|
|
526
|
+
)
|
|
527
|
+
for img in self["imgs_in_doc"]:
|
|
528
|
+
markdown_info["markdown_images"][img["path"]] = img["img"]
|
|
529
|
+
|
|
530
|
+
return markdown_info
|
|
531
|
+
|
|
532
|
+
|
|
533
|
+
class LayoutParsingBlock:
|
|
534
|
+
|
|
535
|
+
def __init__(self, label, bbox, content="") -> None:
|
|
536
|
+
self.label = label
|
|
537
|
+
self.order_label = None
|
|
538
|
+
self.bbox = list(map(int, bbox))
|
|
539
|
+
self.content = content
|
|
540
|
+
self.seg_start_coordinate = float("inf")
|
|
541
|
+
self.seg_end_coordinate = float("-inf")
|
|
542
|
+
self.width = bbox[2] - bbox[0]
|
|
543
|
+
self.height = bbox[3] - bbox[1]
|
|
544
|
+
self.area = self.width * self.height
|
|
545
|
+
self.num_of_lines = 1
|
|
546
|
+
self.image = None
|
|
547
|
+
self.index = None
|
|
548
|
+
self.order_index = None
|
|
549
|
+
self.text_line_width = 1
|
|
550
|
+
self.text_line_height = 1
|
|
551
|
+
self.direction = self.get_bbox_direction()
|
|
552
|
+
self.child_blocks = []
|
|
553
|
+
self.update_direction_info()
|
|
554
|
+
|
|
555
|
+
def __str__(self) -> str:
|
|
556
|
+
return f"{self.__dict__}"
|
|
557
|
+
|
|
558
|
+
def __repr__(self) -> str:
|
|
559
|
+
_str = f"\n\n#################\nindex:\t{self.index}\nlabel:\t{self.label}\nregion_label:\t{self.order_label}\nbbox:\t{self.bbox}\ncontent:\t{self.content}\n#################"
|
|
560
|
+
return _str
|
|
561
|
+
|
|
562
|
+
def to_dict(self) -> dict:
|
|
563
|
+
return self.__dict__
|
|
564
|
+
|
|
565
|
+
def update_direction_info(self) -> None:
|
|
566
|
+
if self.direction == "horizontal":
|
|
567
|
+
self.secondary_direction = "vertical"
|
|
568
|
+
self.short_side_length = self.height
|
|
569
|
+
self.long_side_length = self.width
|
|
570
|
+
self.start_coordinate = self.bbox[0]
|
|
571
|
+
self.end_coordinate = self.bbox[2]
|
|
572
|
+
self.secondary_direction_start_coordinate = self.bbox[1]
|
|
573
|
+
self.secondary_direction_end_coordinate = self.bbox[3]
|
|
574
|
+
else:
|
|
575
|
+
self.secondary_direction = "horizontal"
|
|
576
|
+
self.short_side_length = self.width
|
|
577
|
+
self.long_side_length = self.height
|
|
578
|
+
self.start_coordinate = self.bbox[1]
|
|
579
|
+
self.end_coordinate = self.bbox[3]
|
|
580
|
+
self.secondary_direction_start_coordinate = self.bbox[0]
|
|
581
|
+
self.secondary_direction_end_coordinate = self.bbox[2]
|
|
582
|
+
|
|
583
|
+
def append_child_block(self, child_block: LayoutParsingBlock) -> None:
|
|
584
|
+
if not self.child_blocks:
|
|
585
|
+
self.ori_bbox = self.bbox.copy()
|
|
586
|
+
x1, y1, x2, y2 = self.bbox
|
|
587
|
+
x1_child, y1_child, x2_child, y2_child = child_block.bbox
|
|
588
|
+
union_bbox = (
|
|
589
|
+
min(x1, x1_child),
|
|
590
|
+
min(y1, y1_child),
|
|
591
|
+
max(x2, x2_child),
|
|
592
|
+
max(y2, y2_child),
|
|
593
|
+
)
|
|
594
|
+
self.bbox = union_bbox
|
|
595
|
+
self.update_direction_info()
|
|
596
|
+
child_blocks = [child_block]
|
|
597
|
+
if child_block.child_blocks:
|
|
598
|
+
child_blocks.extend(child_block.get_child_blocks())
|
|
599
|
+
self.child_blocks.extend(child_blocks)
|
|
600
|
+
|
|
601
|
+
def get_child_blocks(self) -> list:
|
|
602
|
+
self.bbox = self.ori_bbox
|
|
603
|
+
child_blocks = self.child_blocks.copy()
|
|
604
|
+
self.child_blocks = []
|
|
605
|
+
return child_blocks
|
|
606
|
+
|
|
607
|
+
def get_centroid(self) -> tuple:
|
|
608
|
+
x1, y1, x2, y2 = self.bbox
|
|
609
|
+
centroid = ((x1 + x2) / 2, (y1 + y2) / 2)
|
|
610
|
+
return centroid
|
|
611
|
+
|
|
612
|
+
def get_bbox_direction(self, direction_ratio: float = 1.0) -> bool:
|
|
613
|
+
"""
|
|
614
|
+
Determine if a bounding box is horizontal or vertical.
|
|
615
|
+
|
|
616
|
+
Args:
|
|
617
|
+
bbox (List[float]): Bounding box [x_min, y_min, x_max, y_max].
|
|
618
|
+
direction_ratio (float): Ratio for determining direction. Default is 1.0.
|
|
619
|
+
|
|
620
|
+
Returns:
|
|
621
|
+
str: "horizontal" or "vertical".
|
|
622
|
+
"""
|
|
623
|
+
return (
|
|
624
|
+
"horizontal" if self.width * direction_ratio >= self.height else "vertical"
|
|
625
|
+
)
|
|
626
|
+
|
|
627
|
+
|
|
628
|
+
class LayoutParsingRegion:
|
|
629
|
+
|
|
630
|
+
def __init__(
|
|
631
|
+
self, bbox, blocks: List[LayoutParsingBlock] = [], image_shape=None
|
|
632
|
+
) -> None:
|
|
633
|
+
self.bbox = bbox
|
|
634
|
+
self.block_map = {}
|
|
635
|
+
self.direction = "horizontal"
|
|
636
|
+
self.calculate_bbox_metrics(image_shape)
|
|
637
|
+
self.doc_title_block_idxes = []
|
|
638
|
+
self.paragraph_title_block_idxes = []
|
|
639
|
+
self.vision_block_idxes = []
|
|
640
|
+
self.unordered_block_idxes = []
|
|
641
|
+
self.vision_title_block_idxes = []
|
|
642
|
+
self.normal_text_block_idxes = []
|
|
643
|
+
self.header_block_idxes = []
|
|
644
|
+
self.footer_block_idxes = []
|
|
645
|
+
self.text_line_width = 20
|
|
646
|
+
self.text_line_height = 10
|
|
647
|
+
self.init_region_info_from_layout(blocks)
|
|
648
|
+
self.init_direction_info()
|
|
649
|
+
|
|
650
|
+
def init_region_info_from_layout(self, blocks: List[LayoutParsingBlock]):
|
|
651
|
+
horizontal_normal_text_block_num = 0
|
|
652
|
+
text_line_height_list = []
|
|
653
|
+
text_line_width_list = []
|
|
654
|
+
for idx, block in enumerate(blocks):
|
|
655
|
+
self.block_map[idx] = block
|
|
656
|
+
block.index = idx
|
|
657
|
+
if block.label in BLOCK_LABEL_MAP["header_labels"]:
|
|
658
|
+
self.header_block_idxes.append(idx)
|
|
659
|
+
elif block.label in BLOCK_LABEL_MAP["doc_title_labels"]:
|
|
660
|
+
self.doc_title_block_idxes.append(idx)
|
|
661
|
+
elif block.label in BLOCK_LABEL_MAP["paragraph_title_labels"]:
|
|
662
|
+
self.paragraph_title_block_idxes.append(idx)
|
|
663
|
+
elif block.label in BLOCK_LABEL_MAP["vision_labels"]:
|
|
664
|
+
self.vision_block_idxes.append(idx)
|
|
665
|
+
elif block.label in BLOCK_LABEL_MAP["vision_title_labels"]:
|
|
666
|
+
self.vision_title_block_idxes.append(idx)
|
|
667
|
+
elif block.label in BLOCK_LABEL_MAP["footer_labels"]:
|
|
668
|
+
self.footer_block_idxes.append(idx)
|
|
669
|
+
elif block.label in BLOCK_LABEL_MAP["unordered_labels"]:
|
|
670
|
+
self.unordered_block_idxes.append(idx)
|
|
671
|
+
else:
|
|
672
|
+
self.normal_text_block_idxes.append(idx)
|
|
673
|
+
text_line_height_list.append(block.text_line_height)
|
|
674
|
+
text_line_width_list.append(block.text_line_width)
|
|
675
|
+
if block.direction == "horizontal":
|
|
676
|
+
horizontal_normal_text_block_num += 1
|
|
677
|
+
self.direction = (
|
|
678
|
+
"horizontal"
|
|
679
|
+
if horizontal_normal_text_block_num
|
|
680
|
+
>= len(self.normal_text_block_idxes) * 0.5
|
|
681
|
+
else "vertical"
|
|
682
|
+
)
|
|
683
|
+
self.text_line_width = (
|
|
684
|
+
np.mean(text_line_width_list) if text_line_width_list else 20
|
|
685
|
+
)
|
|
686
|
+
self.text_line_height = (
|
|
687
|
+
np.mean(text_line_height_list) if text_line_height_list else 10
|
|
688
|
+
)
|
|
689
|
+
|
|
690
|
+
def init_direction_info(self):
|
|
691
|
+
if self.direction == "horizontal":
|
|
692
|
+
self.direction_start_index = 0
|
|
693
|
+
self.direction_end_index = 2
|
|
694
|
+
self.secondary_direction_start_index = 1
|
|
695
|
+
self.secondary_direction_end_index = 3
|
|
696
|
+
self.secondary_direction = "vertical"
|
|
697
|
+
else:
|
|
698
|
+
self.direction_start_index = 1
|
|
699
|
+
self.direction_end_index = 3
|
|
700
|
+
self.secondary_direction_start_index = 0
|
|
701
|
+
self.secondary_direction_end_index = 2
|
|
702
|
+
self.secondary_direction = "horizontal"
|
|
703
|
+
|
|
704
|
+
self.direction_center_coordinate = (
|
|
705
|
+
self.bbox[self.direction_start_index] + self.bbox[self.direction_end_index]
|
|
706
|
+
) / 2
|
|
707
|
+
self.secondary_direction_center_coordinate = (
|
|
708
|
+
self.bbox[self.secondary_direction_start_index]
|
|
709
|
+
+ self.bbox[self.secondary_direction_end_index]
|
|
710
|
+
) / 2
|
|
711
|
+
|
|
712
|
+
def calculate_bbox_metrics(self, image_shape):
|
|
713
|
+
x1, y1, x2, y2 = self.bbox
|
|
714
|
+
image_height, image_width = image_shape
|
|
715
|
+
width = x2 - x1
|
|
716
|
+
x_center, y_center = (x1 + x2) / 2, (y1 + y2) / 2
|
|
717
|
+
self.euclidean_distance = math.sqrt(((x1) ** 2 + (y1) ** 2))
|
|
718
|
+
self.center_euclidean_distance = math.sqrt(((x_center) ** 2 + (y_center) ** 2))
|
|
719
|
+
self.angle_rad = math.atan2(y_center, x_center)
|
|
720
|
+
self.weighted_distance = (
|
|
721
|
+
y2 + width + (x1 // (image_width // 10)) * (image_width // 10) * 1.5
|
|
722
|
+
)
|
|
723
|
+
|
|
724
|
+
def sort_normal_blocks(self, blocks):
|
|
725
|
+
if self.direction == "horizontal":
|
|
726
|
+
blocks.sort(
|
|
727
|
+
key=lambda x: (
|
|
728
|
+
x.bbox[1] // self.text_line_height,
|
|
729
|
+
x.bbox[0] // self.text_line_width,
|
|
730
|
+
x.bbox[1] ** 2 + x.bbox[0] ** 2,
|
|
731
|
+
),
|
|
732
|
+
)
|
|
733
|
+
else:
|
|
734
|
+
blocks.sort(
|
|
735
|
+
key=lambda x: (
|
|
736
|
+
-x.bbox[0] // self.text_line_width,
|
|
737
|
+
x.bbox[1] // self.text_line_height,
|
|
738
|
+
-(x.bbox[2] ** 2 + x.bbox[1] ** 2),
|
|
739
|
+
),
|
|
740
|
+
)
|
|
741
|
+
|
|
742
|
+
def sort(self):
|
|
743
|
+
from .xycut_enhanced import xycut_enhanced
|
|
744
|
+
|
|
745
|
+
return xycut_enhanced(self)
|