paddlex 3.0.0b2__py3-none-any.whl → 3.0.0rc0__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 -1
- paddlex/__init__.py +1 -0
- paddlex/__main__.py +3 -4
- paddlex/configs/modules/3d_bev_detection/BEVFusion.yaml +38 -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/UniMERNet.yaml +40 -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_large_384.yaml +41 -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-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/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/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/YOLOX-X.yaml +40 -0
- paddlex/configs/modules/open_vocabulary_detection/GroundingDINO-T.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/rotated_object_detection/PP-YOLOE-R-L.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/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/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/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_recognition/PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec_doc.yaml +39 -0
- paddlex/configs/modules/text_recognition/arabic_PP-OCRv3_mobile_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/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 +44 -0
- paddlex/configs/pipelines/PP-ChatOCRv3-doc.yaml +149 -0
- paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml +184 -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/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 +101 -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 +51 -0
- paddlex/configs/pipelines/semantic_segmentation.yaml +10 -0
- paddlex/configs/pipelines/small_object_detection.yaml +10 -0
- paddlex/configs/pipelines/table_recognition.yaml +56 -0
- paddlex/configs/pipelines/table_recognition_v2.yaml +76 -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/engine.py +1 -1
- paddlex/hpip_links.html +19 -0
- paddlex/inference/__init__.py +3 -1
- paddlex/inference/common/batch_sampler/__init__.py +20 -0
- paddlex/inference/common/batch_sampler/audio_batch_sampler.py +84 -0
- paddlex/inference/common/batch_sampler/base_batch_sampler.py +90 -0
- paddlex/inference/common/batch_sampler/det_3d_batch_sampler.py +147 -0
- paddlex/inference/common/batch_sampler/image_batch_sampler.py +136 -0
- paddlex/inference/common/batch_sampler/ts_batch_sampler.py +110 -0
- paddlex/inference/common/batch_sampler/video_batch_sampler.py +94 -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 +239 -0
- paddlex/inference/common/reader/image_reader.py +69 -0
- paddlex/inference/common/reader/ts_reader.py +45 -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 +31 -0
- paddlex/inference/common/result/base_result.py +70 -0
- paddlex/inference/common/result/base_ts_result.py +42 -0
- paddlex/inference/common/result/base_video_result.py +36 -0
- paddlex/inference/common/result/mixin.py +703 -0
- paddlex/inference/models/3d_bev_detection/__init__.py +15 -0
- paddlex/inference/models/3d_bev_detection/predictor.py +314 -0
- paddlex/inference/models/3d_bev_detection/processors.py +978 -0
- paddlex/inference/models/3d_bev_detection/result.py +65 -0
- paddlex/inference/models/3d_bev_detection/visualizer_3d.py +131 -0
- paddlex/inference/models/__init__.py +37 -13
- paddlex/inference/models/anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/anomaly_detection/predictor.py +145 -0
- paddlex/inference/models/anomaly_detection/processors.py +46 -0
- paddlex/inference/models/anomaly_detection/result.py +70 -0
- paddlex/inference/models/base/__init__.py +1 -2
- paddlex/inference/models/base/predictor/__init__.py +16 -0
- paddlex/inference/models/base/predictor/base_predictor.py +175 -0
- paddlex/inference/models/base/predictor/basic_predictor.py +139 -0
- paddlex/inference/models/common/__init__.py +35 -0
- paddlex/inference/models/common/static_infer.py +329 -0
- paddlex/inference/models/common/tokenizer/__init__.py +17 -0
- paddlex/inference/models/common/tokenizer/bert_tokenizer.py +655 -0
- paddlex/inference/models/common/tokenizer/gpt_tokenizer.py +451 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils.py +2141 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils_base.py +3504 -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 +533 -0
- paddlex/inference/models/common/ts/processors.py +313 -0
- paddlex/inference/models/common/vision/__init__.py +23 -0
- paddlex/inference/models/common/vision/funcs.py +93 -0
- paddlex/inference/models/common/vision/processors.py +270 -0
- paddlex/inference/models/face_feature/__init__.py +15 -0
- paddlex/inference/models/face_feature/predictor.py +65 -0
- paddlex/inference/models/formula_recognition/__init__.py +15 -0
- paddlex/inference/models/formula_recognition/predictor.py +203 -0
- paddlex/inference/models/formula_recognition/processors.py +986 -0
- paddlex/inference/models/formula_recognition/result.py +403 -0
- paddlex/inference/models/image_classification/__init__.py +15 -0
- paddlex/inference/models/image_classification/predictor.py +182 -0
- paddlex/inference/models/image_classification/processors.py +87 -0
- paddlex/inference/models/image_classification/result.py +92 -0
- paddlex/inference/models/image_feature/__init__.py +15 -0
- paddlex/inference/models/image_feature/predictor.py +156 -0
- paddlex/inference/models/image_feature/processors.py +29 -0
- paddlex/inference/models/image_feature/result.py +33 -0
- paddlex/inference/models/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/models/image_multilabel_classification/predictor.py +94 -0
- paddlex/inference/models/image_multilabel_classification/processors.py +85 -0
- paddlex/inference/models/image_multilabel_classification/result.py +95 -0
- paddlex/inference/models/image_unwarping/__init__.py +15 -0
- paddlex/inference/models/image_unwarping/predictor.py +105 -0
- paddlex/inference/models/image_unwarping/processors.py +88 -0
- paddlex/inference/models/image_unwarping/result.py +45 -0
- paddlex/inference/models/instance_segmentation/__init__.py +15 -0
- paddlex/inference/models/instance_segmentation/predictor.py +210 -0
- paddlex/inference/models/instance_segmentation/processors.py +105 -0
- paddlex/inference/models/instance_segmentation/result.py +161 -0
- paddlex/inference/models/keypoint_detection/__init__.py +15 -0
- paddlex/inference/models/keypoint_detection/predictor.py +188 -0
- paddlex/inference/models/keypoint_detection/processors.py +359 -0
- paddlex/inference/models/keypoint_detection/result.py +192 -0
- paddlex/inference/models/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/models/multilingual_speech_recognition/predictor.py +141 -0
- paddlex/inference/models/multilingual_speech_recognition/processors.py +1941 -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 +348 -0
- paddlex/inference/models/object_detection/processors.py +855 -0
- paddlex/inference/models/object_detection/result.py +113 -0
- paddlex/inference/models/object_detection/utils.py +68 -0
- paddlex/inference/models/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_detection/predictor.py +155 -0
- paddlex/inference/models/open_vocabulary_detection/processors/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_detection/processors/groundingdino_processors.py +485 -0
- paddlex/inference/models/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/predictor.py +120 -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 +147 -0
- paddlex/inference/models/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/models/semantic_segmentation/predictor.py +167 -0
- paddlex/inference/models/semantic_segmentation/processors.py +114 -0
- paddlex/inference/models/semantic_segmentation/result.py +72 -0
- paddlex/inference/models/table_structure_recognition/__init__.py +15 -0
- paddlex/inference/models/table_structure_recognition/predictor.py +171 -0
- paddlex/inference/models/table_structure_recognition/processors.py +235 -0
- paddlex/inference/models/table_structure_recognition/result.py +70 -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 +466 -0
- paddlex/inference/models/text_detection/result.py +51 -0
- paddlex/inference/models/text_recognition/__init__.py +15 -0
- paddlex/inference/models/text_recognition/predictor.py +106 -0
- paddlex/inference/models/text_recognition/processors.py +231 -0
- paddlex/inference/models/text_recognition/result.py +75 -0
- paddlex/inference/models/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/ts_anomaly_detection/predictor.py +146 -0
- paddlex/inference/models/ts_anomaly_detection/processors.py +94 -0
- paddlex/inference/models/ts_anomaly_detection/result.py +72 -0
- paddlex/inference/models/ts_classification/__init__.py +15 -0
- paddlex/inference/models/ts_classification/predictor.py +135 -0
- paddlex/inference/models/ts_classification/processors.py +117 -0
- paddlex/inference/models/ts_classification/result.py +78 -0
- paddlex/inference/models/ts_forecasting/__init__.py +15 -0
- paddlex/inference/models/ts_forecasting/predictor.py +159 -0
- paddlex/inference/models/ts_forecasting/processors.py +149 -0
- paddlex/inference/models/ts_forecasting/result.py +83 -0
- paddlex/inference/models/video_classification/__init__.py +15 -0
- paddlex/inference/models/video_classification/predictor.py +147 -0
- paddlex/inference/models/video_classification/processors.py +409 -0
- paddlex/inference/models/video_classification/result.py +92 -0
- paddlex/inference/models/video_detection/__init__.py +15 -0
- paddlex/inference/models/video_detection/predictor.py +136 -0
- paddlex/inference/models/video_detection/processors.py +450 -0
- paddlex/inference/models/video_detection/result.py +104 -0
- paddlex/inference/pipelines/3d_bev_detection/__init__.py +15 -0
- paddlex/inference/pipelines/3d_bev_detection/pipeline.py +67 -0
- paddlex/inference/pipelines/__init__.py +174 -73
- paddlex/inference/pipelines/anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/anomaly_detection/pipeline.py +62 -0
- paddlex/inference/pipelines/attribute_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/attribute_recognition/pipeline.py +105 -0
- paddlex/inference/pipelines/attribute_recognition/result.py +100 -0
- paddlex/inference/pipelines/base.py +103 -57
- paddlex/inference/pipelines/components/__init__.py +23 -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 +18 -0
- paddlex/inference/pipelines/components/common/base_operator.py +36 -0
- paddlex/inference/pipelines/components/common/base_result.py +65 -0
- paddlex/inference/pipelines/components/common/convert_points_and_boxes.py +46 -0
- paddlex/inference/pipelines/components/common/crop_image_regions.py +550 -0
- paddlex/inference/pipelines/components/common/seal_det_warp.py +941 -0
- paddlex/inference/pipelines/components/common/sort_boxes.py +83 -0
- paddlex/inference/pipelines/components/faisser.py +352 -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 +127 -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 +226 -0
- paddlex/inference/pipelines/components/retriever/openai_bot_retriever.py +70 -0
- paddlex/inference/pipelines/components/retriever/qianfan_bot_retriever.py +163 -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 +190 -0
- paddlex/inference/pipelines/doc_preprocessor/result.py +103 -0
- paddlex/inference/pipelines/face_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/face_recognition/pipeline.py +61 -0
- paddlex/inference/pipelines/face_recognition/result.py +43 -0
- paddlex/inference/pipelines/formula_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/formula_recognition/pipeline.py +303 -0
- paddlex/inference/pipelines/formula_recognition/result.py +291 -0
- paddlex/inference/pipelines/image_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_classification/pipeline.py +71 -0
- paddlex/inference/pipelines/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_multilabel_classification/pipeline.py +78 -0
- paddlex/inference/pipelines/instance_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/instance_segmentation/pipeline.py +70 -0
- paddlex/inference/pipelines/keypoint_detection/__init__.py +15 -0
- paddlex/inference/pipelines/keypoint_detection/pipeline.py +137 -0
- paddlex/inference/pipelines/layout_parsing/__init__.py +2 -1
- paddlex/inference/pipelines/layout_parsing/pipeline.py +570 -0
- paddlex/inference/pipelines/layout_parsing/pipeline_v2.py +739 -0
- paddlex/inference/pipelines/layout_parsing/result.py +203 -0
- paddlex/inference/pipelines/layout_parsing/result_v2.py +470 -0
- paddlex/inference/pipelines/layout_parsing/utils.py +2385 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/pipeline.py +67 -0
- paddlex/inference/pipelines/object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/ocr/__init__.py +15 -0
- paddlex/inference/pipelines/ocr/pipeline.py +389 -0
- paddlex/inference/pipelines/ocr/result.py +248 -0
- paddlex/inference/pipelines/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_detection/pipeline.py +75 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/pipeline.py +89 -0
- paddlex/inference/pipelines/pp_chatocr/__init__.py +16 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_base.py +102 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v3.py +773 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v4.py +977 -0
- paddlex/inference/pipelines/pp_shitu_v2/__init__.py +15 -0
- paddlex/inference/pipelines/pp_shitu_v2/pipeline.py +152 -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 +74 -0
- paddlex/inference/pipelines/seal_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/seal_recognition/pipeline.py +271 -0
- paddlex/inference/pipelines/seal_recognition/result.py +87 -0
- paddlex/inference/pipelines/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/semantic_segmentation/pipeline.py +74 -0
- paddlex/inference/pipelines/small_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/small_object_detection/pipeline.py +74 -0
- paddlex/inference/pipelines/table_recognition/__init__.py +2 -1
- paddlex/inference/pipelines/table_recognition/pipeline.py +462 -0
- paddlex/inference/pipelines/table_recognition/pipeline_v2.py +792 -0
- paddlex/inference/pipelines/table_recognition/result.py +216 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing.py +362 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing_v2.py +470 -0
- paddlex/inference/pipelines/table_recognition/utils.py +23 -436
- paddlex/inference/pipelines/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/ts_anomaly_detection/pipeline.py +62 -0
- paddlex/inference/pipelines/ts_classification/__init__.py +15 -0
- paddlex/inference/pipelines/ts_classification/pipeline.py +62 -0
- paddlex/inference/pipelines/ts_forecasting/__init__.py +15 -0
- paddlex/inference/pipelines/ts_forecasting/pipeline.py +62 -0
- paddlex/inference/pipelines/video_classification/__init__.py +15 -0
- paddlex/inference/pipelines/video_classification/pipeline.py +68 -0
- paddlex/inference/pipelines/video_detection/__init__.py +15 -0
- paddlex/inference/pipelines/video_detection/pipeline.py +73 -0
- paddlex/inference/serving/__init__.py +13 -0
- paddlex/inference/serving/basic_serving/__init__.py +18 -0
- paddlex/inference/serving/basic_serving/_app.py +209 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/__init__.py +41 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/__init__.py +13 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/common.py +96 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/image_recognition.py +36 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/ocr.py +90 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/anomaly_detection.py +64 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_preprocessor.py +97 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/face_recognition.py +223 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/formula_recognition.py +97 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/human_keypoint_detection.py +78 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_classification.py +66 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_multilabel_classification.py +70 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/instance_segmentation.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/layout_parsing.py +115 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/m_3d_bev_detection.py +76 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/multilingual_speech_recognition.py +89 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/object_detection.py +74 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ocr.py +99 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_detection.py +78 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_segmentation.py +85 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pedestrian_attribute_recognition.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv3_doc.py +191 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv4_doc.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_shituv2.py +218 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_structurev3.py +136 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/rotated_object_detection.py +78 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/seal_recognition.py +103 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/semantic_segmentation.py +64 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/small_object_detection.py +69 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition.py +105 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition_v2.py +107 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_anomaly_detection.py +62 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_classification.py +61 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_forecast.py +62 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/vehicle_attribute_recognition.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_classification.py +73 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_detection.py +89 -0
- paddlex/inference/serving/basic_serving/_server.py +35 -0
- paddlex/inference/serving/infra/__init__.py +13 -0
- paddlex/inference/serving/infra/config.py +36 -0
- paddlex/inference/serving/infra/models.py +72 -0
- paddlex/inference/serving/infra/storage.py +175 -0
- paddlex/inference/serving/infra/utils.py +259 -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/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 +72 -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 +134 -0
- paddlex/inference/serving/schemas/pp_chatocrv4_doc.py +151 -0
- paddlex/inference/serving/schemas/pp_shituv2.py +124 -0
- paddlex/inference/serving/schemas/pp_structurev3.py +84 -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 +66 -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/benchmark.py +23 -11
- paddlex/inference/utils/get_pipeline_path.py +2 -1
- paddlex/inference/utils/io/__init__.py +3 -0
- paddlex/inference/utils/io/readers.py +164 -17
- paddlex/inference/utils/io/writers.py +85 -2
- paddlex/inference/utils/new_ir_blacklist.py +6 -0
- paddlex/inference/utils/official_models.py +277 -211
- paddlex/inference/utils/pp_option.py +24 -4
- paddlex/model.py +12 -5
- paddlex/modules/3d_bev_detection/__init__.py +18 -0
- paddlex/modules/3d_bev_detection/dataset_checker/__init__.py +95 -0
- paddlex/modules/3d_bev_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/3d_bev_detection/dataset_checker/dataset_src/analyse_dataset.py +106 -0
- paddlex/modules/3d_bev_detection/dataset_checker/dataset_src/check_dataset.py +102 -0
- paddlex/modules/3d_bev_detection/evaluator.py +46 -0
- paddlex/modules/3d_bev_detection/exportor.py +22 -0
- paddlex/modules/3d_bev_detection/model_list.py +18 -0
- paddlex/modules/3d_bev_detection/trainer.py +70 -0
- paddlex/modules/__init__.py +34 -1
- paddlex/modules/base/build_model.py +1 -1
- paddlex/modules/base/dataset_checker/dataset_checker.py +6 -1
- paddlex/modules/base/evaluator.py +20 -4
- paddlex/modules/base/exportor.py +30 -5
- paddlex/modules/base/trainer.py +29 -6
- paddlex/modules/face_recognition/trainer.py +1 -23
- paddlex/modules/formula_recognition/__init__.py +5 -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 +157 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/check_dataset.py +80 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/convert_dataset.py +94 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/formula_recognition/evaluator.py +77 -0
- paddlex/modules/formula_recognition/exportor.py +22 -0
- paddlex/modules/formula_recognition/model_list.py +3 -0
- paddlex/modules/formula_recognition/trainer.py +121 -0
- paddlex/modules/image_classification/model_list.py +2 -0
- paddlex/modules/instance_segmentation/dataset_checker/__init__.py +15 -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 +86 -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 +119 -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/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 +40 -0
- paddlex/modules/object_detection/evaluator.py +12 -1
- paddlex/modules/object_detection/model_list.py +10 -0
- paddlex/modules/object_detection/trainer.py +15 -1
- 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 +18 -0
- paddlex/modules/open_vocabulary_detection/trainer.py +42 -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 +42 -0
- paddlex/modules/semantic_segmentation/dataset_checker/__init__.py +15 -0
- paddlex/modules/semantic_segmentation/exportor.py +9 -0
- paddlex/modules/semantic_segmentation/model_list.py +2 -0
- paddlex/modules/semantic_segmentation/trainer.py +2 -0
- paddlex/modules/table_recognition/dataset_checker/__init__.py +16 -1
- paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py +13 -14
- paddlex/modules/table_recognition/model_list.py +2 -0
- paddlex/modules/text_detection/dataset_checker/__init__.py +16 -1
- paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py +13 -3
- paddlex/modules/text_detection/model_list.py +2 -0
- paddlex/modules/text_recognition/dataset_checker/__init__.py +16 -4
- paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py +13 -3
- paddlex/modules/text_recognition/evaluator.py +4 -3
- paddlex/modules/text_recognition/exportor.py +0 -3
- paddlex/modules/text_recognition/model_list.py +14 -0
- paddlex/modules/text_recognition/trainer.py +4 -3
- paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py +15 -0
- paddlex/modules/ts_anomaly_detection/trainer.py +17 -1
- paddlex/modules/ts_classification/dataset_checker/__init__.py +15 -0
- paddlex/modules/ts_classification/trainer.py +17 -1
- paddlex/modules/ts_forecast/dataset_checker/__init__.py +15 -0
- paddlex/modules/ts_forecast/trainer.py +17 -1
- 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 +121 -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 +101 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/check_dataset.py +134 -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 +149 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.cpp +264 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.h +27 -0
- paddlex/ops/iou3d_nms/iou3d_nms.cpp +204 -0
- paddlex/ops/iou3d_nms/iou3d_nms.h +33 -0
- paddlex/ops/iou3d_nms/iou3d_nms_api.cpp +108 -0
- paddlex/ops/iou3d_nms/iou3d_nms_kernel.cu +482 -0
- paddlex/ops/setup.py +37 -0
- paddlex/ops/voxel/voxelize_op.cc +191 -0
- paddlex/ops/voxel/voxelize_op.cu +346 -0
- paddlex/paddle2onnx_requirements.txt +1 -0
- paddlex/paddlex_cli.py +339 -72
- 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 +144 -0
- paddlex/repo_apis/PaddleClas_api/cls/model.py +6 -0
- paddlex/repo_apis/PaddleClas_api/cls/register.py +20 -2
- paddlex/repo_apis/PaddleDetection_api/instance_seg/config.py +8 -4
- paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py +6 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/config.py +27 -5
- paddlex/repo_apis/PaddleDetection_api/object_det/model.py +6 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py +81 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/register.py +182 -3
- paddlex/repo_apis/PaddleOCR_api/__init__.py +1 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/config.py +570 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py +402 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/register.py +73 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/runner.py +240 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/register.py +18 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/register.py +18 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/config.py +21 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/model.py +6 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/register.py +126 -7
- paddlex/repo_apis/PaddleSeg_api/seg/config.py +9 -0
- paddlex/repo_apis/PaddleSeg_api/seg/model.py +10 -0
- paddlex/repo_apis/PaddleSeg_api/seg/register.py +20 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/config.py +24 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/model.py +11 -7
- 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 +547 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/model.py +346 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/register.py +71 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/runner.py +205 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/config.py +548 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/model.py +298 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/register.py +45 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/runner.py +200 -0
- paddlex/repo_apis/base/runner.py +2 -1
- paddlex/repo_manager/meta.py +29 -2
- paddlex/repo_manager/repo.py +24 -5
- paddlex/repo_manager/requirements.txt +10 -7
- paddlex/repo_manager/utils.py +62 -1
- paddlex/serving_requirements.txt +9 -0
- paddlex/utils/config.py +4 -3
- paddlex/utils/custom_device_whitelist.py +457 -0
- paddlex/utils/device.py +74 -26
- paddlex/utils/env.py +28 -0
- paddlex/utils/flags.py +4 -0
- paddlex/utils/fonts/__init__.py +48 -5
- paddlex/utils/lazy_loader.py +2 -0
- paddlex/utils/logging.py +1 -2
- paddlex/utils/pipeline_arguments.py +711 -0
- paddlex-3.0.0rc0.dist-info/METADATA +1035 -0
- paddlex-3.0.0rc0.dist-info/RECORD +1015 -0
- paddlex-3.0.0rc0.dist-info/WHEEL +5 -0
- paddlex/configs/face_recognition/MobileFaceNet.yaml +0 -44
- paddlex/configs/face_recognition/ResNet50_face.yaml +0 -44
- paddlex/configs/formula_recognition/LaTeX_OCR_rec.yaml +0 -40
- paddlex/configs/image_classification/CLIP_vit_base_patch16_224.yaml +0 -41
- paddlex/configs/image_classification/CLIP_vit_large_patch14_224.yaml +0 -41
- paddlex/configs/image_classification/ConvNeXt_large_384.yaml +0 -41
- paddlex/configs/object_detection/YOLOX-X.yaml +0 -40
- paddlex/configs/semantic_segmentation/SeaFormer_base.yaml +0 -40
- paddlex/configs/semantic_segmentation/SeaFormer_large.yaml +0 -40
- paddlex/configs/semantic_segmentation/SeaFormer_small.yaml +0 -40
- paddlex/configs/semantic_segmentation/SeaFormer_tiny.yaml +0 -40
- paddlex/inference/components/__init__.py +0 -18
- paddlex/inference/components/base.py +0 -292
- paddlex/inference/components/llm/__init__.py +0 -25
- paddlex/inference/components/llm/base.py +0 -65
- paddlex/inference/components/llm/erniebot.py +0 -212
- paddlex/inference/components/paddle_predictor/__init__.py +0 -20
- paddlex/inference/components/paddle_predictor/predictor.py +0 -332
- paddlex/inference/components/retrieval/__init__.py +0 -15
- paddlex/inference/components/retrieval/faiss.py +0 -359
- paddlex/inference/components/task_related/__init__.py +0 -33
- paddlex/inference/components/task_related/clas.py +0 -124
- paddlex/inference/components/task_related/det.py +0 -284
- paddlex/inference/components/task_related/instance_seg.py +0 -89
- paddlex/inference/components/task_related/seal_det_warp.py +0 -940
- paddlex/inference/components/task_related/seg.py +0 -40
- paddlex/inference/components/task_related/table_rec.py +0 -191
- paddlex/inference/components/task_related/text_det.py +0 -895
- paddlex/inference/components/task_related/text_rec.py +0 -353
- paddlex/inference/components/task_related/warp.py +0 -43
- paddlex/inference/components/transforms/__init__.py +0 -16
- paddlex/inference/components/transforms/image/__init__.py +0 -15
- paddlex/inference/components/transforms/image/common.py +0 -598
- paddlex/inference/components/transforms/image/funcs.py +0 -58
- paddlex/inference/components/transforms/read_data.py +0 -67
- paddlex/inference/components/transforms/ts/__init__.py +0 -15
- paddlex/inference/components/transforms/ts/common.py +0 -393
- paddlex/inference/components/transforms/ts/funcs.py +0 -424
- paddlex/inference/models/anomaly_detection.py +0 -87
- paddlex/inference/models/base/base_predictor.py +0 -76
- paddlex/inference/models/base/basic_predictor.py +0 -122
- paddlex/inference/models/face_recognition.py +0 -21
- paddlex/inference/models/formula_recognition.py +0 -55
- paddlex/inference/models/general_recognition.py +0 -99
- paddlex/inference/models/image_classification.py +0 -101
- paddlex/inference/models/image_unwarping.py +0 -43
- paddlex/inference/models/instance_segmentation.py +0 -66
- paddlex/inference/models/multilabel_classification.py +0 -33
- paddlex/inference/models/object_detection.py +0 -129
- paddlex/inference/models/semantic_segmentation.py +0 -86
- paddlex/inference/models/table_recognition.py +0 -106
- paddlex/inference/models/text_detection.py +0 -105
- paddlex/inference/models/text_recognition.py +0 -78
- paddlex/inference/models/ts_ad.py +0 -68
- paddlex/inference/models/ts_cls.py +0 -57
- paddlex/inference/models/ts_fc.py +0 -73
- paddlex/inference/pipelines/attribute_recognition.py +0 -92
- paddlex/inference/pipelines/face_recognition.py +0 -49
- paddlex/inference/pipelines/formula_recognition.py +0 -102
- paddlex/inference/pipelines/layout_parsing/layout_parsing.py +0 -362
- paddlex/inference/pipelines/ocr.py +0 -80
- paddlex/inference/pipelines/pp_shitu_v2.py +0 -152
- paddlex/inference/pipelines/ppchatocrv3/__init__.py +0 -15
- paddlex/inference/pipelines/ppchatocrv3/ch_prompt.yaml +0 -14
- paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py +0 -717
- paddlex/inference/pipelines/ppchatocrv3/utils.py +0 -168
- paddlex/inference/pipelines/seal_recognition.py +0 -152
- paddlex/inference/pipelines/serving/__init__.py +0 -17
- paddlex/inference/pipelines/serving/_pipeline_apps/__init__.py +0 -205
- paddlex/inference/pipelines/serving/_pipeline_apps/anomaly_detection.py +0 -80
- paddlex/inference/pipelines/serving/_pipeline_apps/face_recognition.py +0 -317
- paddlex/inference/pipelines/serving/_pipeline_apps/formula_recognition.py +0 -119
- paddlex/inference/pipelines/serving/_pipeline_apps/image_classification.py +0 -101
- paddlex/inference/pipelines/serving/_pipeline_apps/instance_segmentation.py +0 -112
- paddlex/inference/pipelines/serving/_pipeline_apps/layout_parsing.py +0 -205
- paddlex/inference/pipelines/serving/_pipeline_apps/multi_label_image_classification.py +0 -90
- paddlex/inference/pipelines/serving/_pipeline_apps/object_detection.py +0 -90
- paddlex/inference/pipelines/serving/_pipeline_apps/ocr.py +0 -98
- paddlex/inference/pipelines/serving/_pipeline_apps/pedestrian_attribute_recognition.py +0 -102
- paddlex/inference/pipelines/serving/_pipeline_apps/pp_shitu_v2.py +0 -319
- paddlex/inference/pipelines/serving/_pipeline_apps/ppchatocrv3.py +0 -445
- paddlex/inference/pipelines/serving/_pipeline_apps/seal_recognition.py +0 -110
- paddlex/inference/pipelines/serving/_pipeline_apps/semantic_segmentation.py +0 -82
- paddlex/inference/pipelines/serving/_pipeline_apps/small_object_detection.py +0 -92
- paddlex/inference/pipelines/serving/_pipeline_apps/table_recognition.py +0 -110
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_ad.py +0 -68
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_cls.py +0 -68
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_fc.py +0 -68
- paddlex/inference/pipelines/serving/_pipeline_apps/vehicle_attribute_recognition.py +0 -102
- paddlex/inference/pipelines/serving/app.py +0 -164
- paddlex/inference/pipelines/serving/models.py +0 -30
- paddlex/inference/pipelines/serving/server.py +0 -25
- paddlex/inference/pipelines/serving/storage.py +0 -161
- paddlex/inference/pipelines/serving/utils.py +0 -190
- paddlex/inference/pipelines/single_model_pipeline.py +0 -76
- paddlex/inference/pipelines/table_recognition/table_recognition.py +0 -193
- paddlex/inference/results/__init__.py +0 -31
- paddlex/inference/results/attribute_rec.py +0 -89
- paddlex/inference/results/base.py +0 -43
- paddlex/inference/results/chat_ocr.py +0 -158
- paddlex/inference/results/clas.py +0 -133
- paddlex/inference/results/det.py +0 -86
- paddlex/inference/results/face_rec.py +0 -34
- paddlex/inference/results/formula_rec.py +0 -363
- paddlex/inference/results/instance_seg.py +0 -152
- paddlex/inference/results/ocr.py +0 -157
- paddlex/inference/results/seal_rec.py +0 -50
- paddlex/inference/results/seg.py +0 -72
- paddlex/inference/results/shitu.py +0 -35
- paddlex/inference/results/table_rec.py +0 -109
- paddlex/inference/results/text_det.py +0 -33
- paddlex/inference/results/text_rec.py +0 -66
- paddlex/inference/results/ts.py +0 -37
- paddlex/inference/results/utils/mixin.py +0 -204
- paddlex/inference/results/warp.py +0 -31
- paddlex/inference/utils/process_hook.py +0 -54
- paddlex/pipelines/OCR.yaml +0 -8
- paddlex/pipelines/PP-ChatOCRv3-doc.yaml +0 -27
- paddlex/pipelines/PP-ShiTuV2.yaml +0 -13
- paddlex/pipelines/anomaly_detection.yaml +0 -7
- paddlex/pipelines/face_recognition.yaml +0 -13
- paddlex/pipelines/formula_recognition.yaml +0 -8
- paddlex/pipelines/image_classification.yaml +0 -7
- paddlex/pipelines/instance_segmentation.yaml +0 -7
- paddlex/pipelines/layout_parsing.yaml +0 -14
- paddlex/pipelines/multi_label_image_classification.yaml +0 -7
- paddlex/pipelines/object_detection.yaml +0 -7
- paddlex/pipelines/pedestrian_attribute_recognition.yaml +0 -7
- paddlex/pipelines/seal_recognition.yaml +0 -10
- paddlex/pipelines/semantic_segmentation.yaml +0 -7
- paddlex/pipelines/small_object_detection.yaml +0 -7
- paddlex/pipelines/table_recognition.yaml +0 -12
- paddlex/pipelines/ts_ad.yaml +0 -7
- paddlex/pipelines/ts_cls.yaml +0 -7
- paddlex/pipelines/ts_fc.yaml +0 -7
- paddlex/pipelines/vehicle_attribute_recognition.yaml +0 -7
- paddlex/utils/fonts/PingFang-SC-Regular.ttf +0 -0
- paddlex-3.0.0b2.dist-info/METADATA +0 -760
- paddlex-3.0.0b2.dist-info/RECORD +0 -646
- paddlex-3.0.0b2.dist-info/WHEEL +0 -5
- /paddlex/configs/{doc_text_orientation → modules/doc_text_orientation}/PP-LCNet_x1_0_doc_ori.yaml +0 -0
- /paddlex/configs/{face_detection → modules/face_detection}/BlazeFace-FPN-SSH.yaml +0 -0
- /paddlex/configs/{face_detection → modules/face_detection}/BlazeFace.yaml +0 -0
- /paddlex/configs/{face_detection → modules/face_detection}/PP-YOLOE_plus-S_face.yaml +0 -0
- /paddlex/configs/{face_detection → modules/face_detection}/PicoDet_LCNet_x2_5_face.yaml +0 -0
- /paddlex/configs/{human_detection → modules/human_detection}/PP-YOLOE-L_human.yaml +0 -0
- /paddlex/configs/{human_detection → modules/human_detection}/PP-YOLOE-S_human.yaml +0 -0
- /paddlex/configs/{anomaly_detection → modules/image_anomaly_detection}/STFPM.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ConvNeXt_base_224.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ConvNeXt_base_384.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ConvNeXt_large_224.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ConvNeXt_small.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ConvNeXt_tiny.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-L.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-M.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-S.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-T0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-T1.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/FasterNet-T2.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV1_x0_25.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV1_x0_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV1_x0_75.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV1_x1_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV2_x0_25.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV2_x0_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV2_x1_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV2_x1_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV2_x2_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_large_x0_35.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_large_x0_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_large_x0_75.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_large_x1_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_large_x1_25.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_small_x0_35.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_small_x0_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_small_x0_75.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_small_x1_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV3_small_x1_25.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV4_conv_large.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV4_conv_medium.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV4_conv_small.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV4_hybrid_large.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/MobileNetV4_hybrid_medium.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B1.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B2.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B3.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B4.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNetV2-B6.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNet_base.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNet_small.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-HGNet_tiny.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNetV2_base.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNetV2_large.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNetV2_small.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x0_25.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x0_35.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x0_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x0_75.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x1_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x1_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x2_0.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/PP-LCNet_x2_5.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet101.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet101_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet152.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet152_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet18.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet18_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet200_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet34.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet34_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet50.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/ResNet50_vd.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/StarNet-S1.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/StarNet-S2.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/StarNet-S3.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/StarNet-S4.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_base_patch4_window12_384.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_base_patch4_window7_224.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_large_patch4_window12_384.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_large_patch4_window7_224.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_small_patch4_window7_224.yaml +0 -0
- /paddlex/configs/{image_classification → modules/image_classification}/SwinTransformer_tiny_patch4_window7_224.yaml +0 -0
- /paddlex/configs/{general_recognition → modules/image_feature}/PP-ShiTuV2_rec.yaml +0 -0
- /paddlex/configs/{general_recognition → modules/image_feature}/PP-ShiTuV2_rec_CLIP_vit_base.yaml +0 -0
- /paddlex/configs/{general_recognition → modules/image_feature}/PP-ShiTuV2_rec_CLIP_vit_large.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/CLIP_vit_base_patch16_448_ML.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/PP-HGNetV2-B0_ML.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/PP-HGNetV2-B4_ML.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/PP-HGNetV2-B6_ML.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/PP-LCNet_x1_0_ML.yaml +0 -0
- /paddlex/configs/{multilabel_classification → modules/image_multilabel_classification}/ResNet50_ML.yaml +0 -0
- /paddlex/configs/{image_unwarping → modules/image_unwarping}/UVDoc.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Cascade-MaskRCNN-ResNet50-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Mask-RT-DETR-H.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Mask-RT-DETR-L.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Mask-RT-DETR-M.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Mask-RT-DETR-S.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/Mask-RT-DETR-X.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNeXt101-vd-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNet101-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNet101-vd-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNet50-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNet50-vd-FPN.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/MaskRCNN-ResNet50.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/PP-YOLOE_seg-S.yaml +0 -0
- /paddlex/configs/{instance_segmentation → modules/instance_segmentation}/SOLOv2.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet-L_layout_17cls.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet-L_layout_3cls.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet-S_layout_17cls.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet-S_layout_3cls.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet_layout_1x.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/PicoDet_layout_1x_table.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/RT-DETR-H_layout_17cls.yaml +0 -0
- /paddlex/configs/{structure_analysis → modules/layout_detection}/RT-DETR-H_layout_3cls.yaml +0 -0
- /paddlex/configs/{mainbody_detection → modules/mainbody_detection}/PP-ShiTuV2_det.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/Cascade-FasterRCNN-ResNet50-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/CenterNet-DLA-34.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/CenterNet-ResNet50.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/DETR-R50.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FCOS-ResNet50.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNeXt101-vd-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet101-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet101.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet34-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet50-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet50-vd-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-ResNet50.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/FasterRCNN-Swin-Tiny-FPN.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PP-YOLOE_plus-L.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PP-YOLOE_plus-M.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PP-YOLOE_plus-S.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PP-YOLOE_plus-X.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PicoDet-L.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PicoDet-M.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PicoDet-S.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/PicoDet-XS.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/RT-DETR-H.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/RT-DETR-L.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/RT-DETR-R18.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/RT-DETR-R50.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/RT-DETR-X.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOX-L.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOX-M.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOX-N.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOX-S.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOX-T.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOv3-DarkNet53.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOv3-MobileNetV3.yaml +0 -0
- /paddlex/configs/{object_detection → modules/object_detection}/YOLOv3-ResNet50_vd_DCN.yaml +0 -0
- /paddlex/configs/{pedestrian_attribute → modules/pedestrian_attribute_recognition}/PP-LCNet_x1_0_pedestrian_attribute.yaml +0 -0
- /paddlex/configs/{text_detection_seal → modules/seal_text_detection}/PP-OCRv4_mobile_seal_det.yaml +0 -0
- /paddlex/configs/{text_detection_seal → modules/seal_text_detection}/PP-OCRv4_server_seal_det.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/Deeplabv3-R101.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/Deeplabv3-R50.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/Deeplabv3_Plus-R101.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/Deeplabv3_Plus-R50.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/OCRNet_HRNet-W18.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/OCRNet_HRNet-W48.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/PP-LiteSeg-B.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/PP-LiteSeg-T.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B0.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B1.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B2.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B3.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B4.yaml +0 -0
- /paddlex/configs/{semantic_segmentation → modules/semantic_segmentation}/SegFormer-B5.yaml +0 -0
- /paddlex/configs/{small_object_detection → modules/small_object_detection}/PP-YOLOE_plus_SOD-L.yaml +0 -0
- /paddlex/configs/{small_object_detection → modules/small_object_detection}/PP-YOLOE_plus_SOD-S.yaml +0 -0
- /paddlex/configs/{small_object_detection → modules/small_object_detection}/PP-YOLOE_plus_SOD-largesize-L.yaml +0 -0
- /paddlex/configs/{table_recognition → modules/table_structure_recognition}/SLANet.yaml +0 -0
- /paddlex/configs/{table_recognition → modules/table_structure_recognition}/SLANet_plus.yaml +0 -0
- /paddlex/configs/{text_detection → modules/text_detection}/PP-OCRv4_mobile_det.yaml +0 -0
- /paddlex/configs/{text_detection → modules/text_detection}/PP-OCRv4_server_det.yaml +0 -0
- /paddlex/configs/{text_recognition → modules/text_recognition}/PP-OCRv4_mobile_rec.yaml +0 -0
- /paddlex/configs/{text_recognition → modules/text_recognition}/PP-OCRv4_server_rec.yaml +0 -0
- /paddlex/configs/{text_recognition → modules/text_recognition}/ch_RepSVTR_rec.yaml +0 -0
- /paddlex/configs/{text_recognition → modules/text_recognition}/ch_SVTRv2_rec.yaml +0 -0
- /paddlex/configs/{ts_anomaly_detection → modules/ts_anomaly_detection}/AutoEncoder_ad.yaml +0 -0
- /paddlex/configs/{ts_anomaly_detection → modules/ts_anomaly_detection}/DLinear_ad.yaml +0 -0
- /paddlex/configs/{ts_anomaly_detection → modules/ts_anomaly_detection}/Nonstationary_ad.yaml +0 -0
- /paddlex/configs/{ts_anomaly_detection → modules/ts_anomaly_detection}/PatchTST_ad.yaml +0 -0
- /paddlex/configs/{ts_anomaly_detection → modules/ts_anomaly_detection}/TimesNet_ad.yaml +0 -0
- /paddlex/configs/{ts_classification → modules/ts_classification}/TimesNet_cls.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/DLinear.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/NLinear.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/Nonstationary.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/PatchTST.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/RLinear.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/TiDE.yaml +0 -0
- /paddlex/configs/{ts_forecast → modules/ts_forecast}/TimesNet.yaml +0 -0
- /paddlex/configs/{vehicle_attribute → modules/vehicle_attribute_recognition}/PP-LCNet_x1_0_vehicle_attribute.yaml +0 -0
- /paddlex/configs/{vehicle_detection → modules/vehicle_detection}/PP-YOLOE-L_vehicle.yaml +0 -0
- /paddlex/configs/{vehicle_detection → modules/vehicle_detection}/PP-YOLOE-S_vehicle.yaml +0 -0
- /paddlex/inference/{results/utils → common}/__init__.py +0 -0
- {paddlex-3.0.0b2.dist-info → paddlex-3.0.0rc0.dist-info}/LICENSE +0 -0
- {paddlex-3.0.0b2.dist-info → paddlex-3.0.0rc0.dist-info}/entry_points.txt +0 -0
- {paddlex-3.0.0b2.dist-info → paddlex-3.0.0rc0.dist-info}/top_level.txt +0 -0
@@ -1,332 +0,0 @@
|
|
1
|
-
# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
|
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
|
-
|
15
|
-
import os
|
16
|
-
from abc import abstractmethod
|
17
|
-
import lazy_paddle as paddle
|
18
|
-
import numpy as np
|
19
|
-
|
20
|
-
from ....utils.flags import FLAGS_json_format_model
|
21
|
-
from ....utils import logging
|
22
|
-
from ...utils.pp_option import PaddlePredictorOption
|
23
|
-
from ..base import BaseComponent
|
24
|
-
|
25
|
-
|
26
|
-
class Copy2GPU(BaseComponent):
|
27
|
-
|
28
|
-
def __init__(self, input_handlers):
|
29
|
-
super().__init__()
|
30
|
-
self.input_handlers = input_handlers
|
31
|
-
|
32
|
-
def apply(self, x):
|
33
|
-
for idx in range(len(x)):
|
34
|
-
self.input_handlers[idx].reshape(x[idx].shape)
|
35
|
-
self.input_handlers[idx].copy_from_cpu(x[idx])
|
36
|
-
|
37
|
-
|
38
|
-
class Copy2CPU(BaseComponent):
|
39
|
-
|
40
|
-
def __init__(self, output_handlers):
|
41
|
-
super().__init__()
|
42
|
-
self.output_handlers = output_handlers
|
43
|
-
|
44
|
-
def apply(self):
|
45
|
-
output = []
|
46
|
-
for out_tensor in self.output_handlers:
|
47
|
-
batch = out_tensor.copy_to_cpu()
|
48
|
-
output.append(batch)
|
49
|
-
return output
|
50
|
-
|
51
|
-
|
52
|
-
class Infer(BaseComponent):
|
53
|
-
|
54
|
-
def __init__(self, predictor):
|
55
|
-
super().__init__()
|
56
|
-
self.predictor = predictor
|
57
|
-
|
58
|
-
def apply(self):
|
59
|
-
self.predictor.run()
|
60
|
-
|
61
|
-
|
62
|
-
class BasePaddlePredictor(BaseComponent):
|
63
|
-
"""Predictor based on Paddle Inference"""
|
64
|
-
|
65
|
-
OUTPUT_KEYS = "pred"
|
66
|
-
DEAULT_OUTPUTS = {"pred": "pred"}
|
67
|
-
ENABLE_BATCH = True
|
68
|
-
|
69
|
-
def __init__(self, model_dir, model_prefix, option):
|
70
|
-
super().__init__()
|
71
|
-
self.model_dir = model_dir
|
72
|
-
self.model_prefix = model_prefix
|
73
|
-
self._update_option(option)
|
74
|
-
|
75
|
-
def _update_option(self, option):
|
76
|
-
if option:
|
77
|
-
if self.option and option == self.option:
|
78
|
-
return
|
79
|
-
self._option = option
|
80
|
-
self._reset()
|
81
|
-
|
82
|
-
@property
|
83
|
-
def option(self):
|
84
|
-
return self._option if hasattr(self, "_option") else None
|
85
|
-
|
86
|
-
@option.setter
|
87
|
-
def option(self, option):
|
88
|
-
self._update_option(option)
|
89
|
-
|
90
|
-
def _reset(self):
|
91
|
-
if not self.option:
|
92
|
-
self.option = PaddlePredictorOption()
|
93
|
-
logging.debug(f"Env: {self.option}")
|
94
|
-
(
|
95
|
-
predictor,
|
96
|
-
input_handlers,
|
97
|
-
output_handlers,
|
98
|
-
) = self._create()
|
99
|
-
self.copy2gpu = Copy2GPU(input_handlers)
|
100
|
-
self.copy2cpu = Copy2CPU(output_handlers)
|
101
|
-
self.infer = Infer(predictor)
|
102
|
-
self.option.changed = False
|
103
|
-
|
104
|
-
def _create(self):
|
105
|
-
"""_create"""
|
106
|
-
from lazy_paddle.inference import Config, create_predictor
|
107
|
-
|
108
|
-
model_postfix = ".json" if FLAGS_json_format_model else ".pdmodel"
|
109
|
-
model_file = (self.model_dir / f"{self.model_prefix}{model_postfix}").as_posix()
|
110
|
-
params_file = (self.model_dir / f"{self.model_prefix}.pdiparams").as_posix()
|
111
|
-
config = Config(model_file, params_file)
|
112
|
-
|
113
|
-
config.enable_memory_optim()
|
114
|
-
if self.option.device in ("gpu", "dcu"):
|
115
|
-
if self.option.device == "gpu":
|
116
|
-
config.exp_disable_mixed_precision_ops({"feed", "fetch"})
|
117
|
-
config.enable_use_gpu(100, self.option.device_id)
|
118
|
-
if self.option.device == "gpu":
|
119
|
-
# NOTE: The pptrt settings are not aligned with those of FD.
|
120
|
-
precision_map = {
|
121
|
-
"trt_int8": Config.Precision.Int8,
|
122
|
-
"trt_fp32": Config.Precision.Float32,
|
123
|
-
"trt_fp16": Config.Precision.Half,
|
124
|
-
}
|
125
|
-
if self.option.run_mode in precision_map.keys():
|
126
|
-
config.enable_tensorrt_engine(
|
127
|
-
workspace_size=(1 << 25) * self.option.batch_size,
|
128
|
-
max_batch_size=self.option.batch_size,
|
129
|
-
min_subgraph_size=self.option.min_subgraph_size,
|
130
|
-
precision_mode=precision_map[self.option.run_mode],
|
131
|
-
use_static=self.option.trt_use_static,
|
132
|
-
use_calib_mode=self.option.trt_calib_mode,
|
133
|
-
)
|
134
|
-
|
135
|
-
if self.option.shape_info_filename is not None:
|
136
|
-
if not os.path.exists(self.option.shape_info_filename):
|
137
|
-
config.collect_shape_range_info(
|
138
|
-
self.option.shape_info_filename
|
139
|
-
)
|
140
|
-
logging.info(
|
141
|
-
f"Dynamic shape info is collected into: {self.option.shape_info_filename}"
|
142
|
-
)
|
143
|
-
else:
|
144
|
-
logging.info(
|
145
|
-
f"A dynamic shape info file ( {self.option.shape_info_filename} ) already exists. \
|
146
|
-
No need to generate again."
|
147
|
-
)
|
148
|
-
config.enable_tuned_tensorrt_dynamic_shape(
|
149
|
-
self.option.shape_info_filename, True
|
150
|
-
)
|
151
|
-
elif self.option.device == "npu":
|
152
|
-
config.enable_custom_device("npu")
|
153
|
-
elif self.option.device == "xpu":
|
154
|
-
pass
|
155
|
-
elif self.option.device == "mlu":
|
156
|
-
config.enable_custom_device("mlu")
|
157
|
-
else:
|
158
|
-
assert self.option.device == "cpu"
|
159
|
-
config.disable_gpu()
|
160
|
-
if "mkldnn" in self.option.run_mode:
|
161
|
-
try:
|
162
|
-
config.enable_mkldnn()
|
163
|
-
if "bf16" in self.option.run_mode:
|
164
|
-
config.enable_mkldnn_bfloat16()
|
165
|
-
except Exception as e:
|
166
|
-
logging.warning(
|
167
|
-
"MKL-DNN is not available. We will disable MKL-DNN."
|
168
|
-
)
|
169
|
-
config.set_mkldnn_cache_capacity(-1)
|
170
|
-
else:
|
171
|
-
if hasattr(config, "disable_mkldnn"):
|
172
|
-
config.disable_mkldnn()
|
173
|
-
|
174
|
-
# Disable paddle inference logging
|
175
|
-
config.disable_glog_info()
|
176
|
-
|
177
|
-
config.set_cpu_math_library_num_threads(self.option.cpu_threads)
|
178
|
-
|
179
|
-
if self.option.device in ("cpu", "gpu"):
|
180
|
-
if not (
|
181
|
-
self.option.device == "gpu" and self.option.run_mode.startswith("trt")
|
182
|
-
):
|
183
|
-
if hasattr(config, "enable_new_ir"):
|
184
|
-
config.enable_new_ir(self.option.enable_new_ir)
|
185
|
-
if hasattr(config, "enable_new_executor"):
|
186
|
-
config.enable_new_executor()
|
187
|
-
config.set_optimization_level(3)
|
188
|
-
|
189
|
-
for del_p in self.option.delete_pass:
|
190
|
-
config.delete_pass(del_p)
|
191
|
-
|
192
|
-
if self.option.device in ("gpu", "dcu"):
|
193
|
-
if paddle.is_compiled_with_rocm():
|
194
|
-
# Delete unsupported passes in dcu
|
195
|
-
config.delete_pass("conv2d_add_act_fuse_pass")
|
196
|
-
config.delete_pass("conv2d_add_fuse_pass")
|
197
|
-
|
198
|
-
predictor = create_predictor(config)
|
199
|
-
|
200
|
-
# Get input and output handlers
|
201
|
-
input_names = predictor.get_input_names()
|
202
|
-
input_names.sort()
|
203
|
-
input_handlers = []
|
204
|
-
output_handlers = []
|
205
|
-
for input_name in input_names:
|
206
|
-
input_handler = predictor.get_input_handle(input_name)
|
207
|
-
input_handlers.append(input_handler)
|
208
|
-
output_names = predictor.get_output_names()
|
209
|
-
for output_name in output_names:
|
210
|
-
output_handler = predictor.get_output_handle(output_name)
|
211
|
-
output_handlers.append(output_handler)
|
212
|
-
return predictor, input_handlers, output_handlers
|
213
|
-
|
214
|
-
def apply(self, **kwargs):
|
215
|
-
if self.option.changed:
|
216
|
-
self._reset()
|
217
|
-
batches = self.to_batch(**kwargs)
|
218
|
-
self.copy2gpu.apply(batches)
|
219
|
-
self.infer.apply()
|
220
|
-
pred = self.copy2cpu.apply()
|
221
|
-
return self.format_output(pred)
|
222
|
-
|
223
|
-
@property
|
224
|
-
def sub_cmps(self):
|
225
|
-
return {
|
226
|
-
"Copy2GPU": self.copy2gpu,
|
227
|
-
"Infer": self.infer,
|
228
|
-
"Copy2CPU": self.copy2cpu,
|
229
|
-
}
|
230
|
-
|
231
|
-
@abstractmethod
|
232
|
-
def to_batch(self):
|
233
|
-
raise NotImplementedError
|
234
|
-
|
235
|
-
@abstractmethod
|
236
|
-
def format_output(self, pred):
|
237
|
-
return [{"pred": res} for res in zip(*pred)]
|
238
|
-
|
239
|
-
|
240
|
-
class ImagePredictor(BasePaddlePredictor):
|
241
|
-
INPUT_KEYS = "img"
|
242
|
-
OUTPUT_KEYS = "pred"
|
243
|
-
DEAULT_INPUTS = {"img": "img"}
|
244
|
-
DEAULT_OUTPUTS = {"pred": "pred"}
|
245
|
-
|
246
|
-
def to_batch(self, img):
|
247
|
-
return [np.stack(img, axis=0).astype(dtype=np.float32, copy=False)]
|
248
|
-
|
249
|
-
def format_output(self, pred):
|
250
|
-
return [{"pred": res} for res in zip(*pred)]
|
251
|
-
|
252
|
-
|
253
|
-
class ImageDetPredictor(BasePaddlePredictor):
|
254
|
-
|
255
|
-
INPUT_KEYS = [
|
256
|
-
["img", "scale_factors"],
|
257
|
-
["img", "scale_factors", "img_size"],
|
258
|
-
["img", "img_size"],
|
259
|
-
]
|
260
|
-
OUTPUT_KEYS = [["boxes"], ["boxes", "masks"]]
|
261
|
-
DEAULT_INPUTS = {"img": "img", "scale_factors": "scale_factors"}
|
262
|
-
DEAULT_OUTPUTS = None
|
263
|
-
|
264
|
-
def to_batch(self, img, scale_factors=[[1.0, 1.0]], img_size=None):
|
265
|
-
scale_factors = [scale_factor[::-1] for scale_factor in scale_factors]
|
266
|
-
if img_size is None:
|
267
|
-
return [
|
268
|
-
np.stack(img, axis=0).astype(dtype=np.float32, copy=False),
|
269
|
-
np.stack(scale_factors, axis=0).astype(dtype=np.float32, copy=False),
|
270
|
-
]
|
271
|
-
else:
|
272
|
-
img_size = [img_size[::-1] for img_size in img_size]
|
273
|
-
return [
|
274
|
-
np.stack(img_size, axis=0).astype(dtype=np.float32, copy=False),
|
275
|
-
np.stack(img, axis=0).astype(dtype=np.float32, copy=False),
|
276
|
-
np.stack(scale_factors, axis=0).astype(dtype=np.float32, copy=False),
|
277
|
-
]
|
278
|
-
|
279
|
-
def format_output(self, pred):
|
280
|
-
box_idx_start = 0
|
281
|
-
pred_box = []
|
282
|
-
|
283
|
-
if len(pred) == 4:
|
284
|
-
# Adapt to SOLOv2
|
285
|
-
pred_class_id = []
|
286
|
-
pred_mask = []
|
287
|
-
pred_class_id.append([pred[1], pred[2]])
|
288
|
-
pred_mask.append(pred[3])
|
289
|
-
return [
|
290
|
-
{
|
291
|
-
"class_id": np.array(pred_class_id[i]),
|
292
|
-
"masks": np.array(pred_mask[i]),
|
293
|
-
}
|
294
|
-
for i in range(len(pred_class_id))
|
295
|
-
]
|
296
|
-
|
297
|
-
if len(pred) == 3:
|
298
|
-
# Adapt to Instance Segmentation
|
299
|
-
pred_mask = []
|
300
|
-
for idx in range(len(pred[1])):
|
301
|
-
np_boxes_num = pred[1][idx]
|
302
|
-
box_idx_end = box_idx_start + np_boxes_num
|
303
|
-
np_boxes = pred[0][box_idx_start:box_idx_end]
|
304
|
-
pred_box.append(np_boxes)
|
305
|
-
if len(pred) == 3:
|
306
|
-
np_masks = pred[2][box_idx_start:box_idx_end]
|
307
|
-
pred_mask.append(np_masks)
|
308
|
-
box_idx_start = box_idx_end
|
309
|
-
|
310
|
-
if len(pred) == 3:
|
311
|
-
return [
|
312
|
-
{"boxes": np.array(pred_box[i]), "masks": np.array(pred_mask[i])}
|
313
|
-
for i in range(len(pred_box))
|
314
|
-
]
|
315
|
-
else:
|
316
|
-
return [{"boxes": np.array(res)} for res in pred_box]
|
317
|
-
|
318
|
-
|
319
|
-
class TSPPPredictor(BasePaddlePredictor):
|
320
|
-
|
321
|
-
INPUT_KEYS = "ts"
|
322
|
-
OUTPUT_KEYS = "pred"
|
323
|
-
DEAULT_INPUTS = {"ts": "ts"}
|
324
|
-
DEAULT_OUTPUTS = {"pred": "pred"}
|
325
|
-
|
326
|
-
def to_batch(self, ts):
|
327
|
-
n = len(ts[0])
|
328
|
-
x = [np.stack([lst[i] for lst in ts], axis=0) for i in range(n)]
|
329
|
-
return x
|
330
|
-
|
331
|
-
def format_output(self, pred):
|
332
|
-
return [{"pred": res} for res in zip(*pred)]
|
@@ -1,15 +0,0 @@
|
|
1
|
-
# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
|
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
|
-
|
15
|
-
from .faiss import FaissIndexer, FaissBuilder
|
@@ -1,359 +0,0 @@
|
|
1
|
-
# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
|
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
|
-
|
15
|
-
import os
|
16
|
-
import pickle
|
17
|
-
from pathlib import Path
|
18
|
-
import faiss
|
19
|
-
import numpy as np
|
20
|
-
|
21
|
-
from ....utils import logging
|
22
|
-
from ...utils.io import YAMLWriter, YAMLReader
|
23
|
-
from ..base import BaseComponent
|
24
|
-
|
25
|
-
|
26
|
-
class IndexData:
|
27
|
-
VECTOR_FN = "vector"
|
28
|
-
VECTOR_SUFFIX = ".index"
|
29
|
-
IDMAP_FN = "id_map"
|
30
|
-
IDMAP_SUFFIX = ".yaml"
|
31
|
-
|
32
|
-
def __init__(self, index, index_info):
|
33
|
-
self._index = index
|
34
|
-
self._index_info = index_info
|
35
|
-
self._id_map = index_info["id_map"]
|
36
|
-
self._metric_type = index_info["metric_type"]
|
37
|
-
self._index_type = index_info["index_type"]
|
38
|
-
|
39
|
-
@property
|
40
|
-
def index(self):
|
41
|
-
return self._index
|
42
|
-
|
43
|
-
@property
|
44
|
-
def index_bytes(self):
|
45
|
-
return faiss.serialize_index(self._index)
|
46
|
-
|
47
|
-
@property
|
48
|
-
def id_map(self):
|
49
|
-
return self._id_map
|
50
|
-
|
51
|
-
@property
|
52
|
-
def metric_type(self):
|
53
|
-
return self._metric_type
|
54
|
-
|
55
|
-
@property
|
56
|
-
def index_type(self):
|
57
|
-
return self._index_type
|
58
|
-
|
59
|
-
@property
|
60
|
-
def index_info(self):
|
61
|
-
return {
|
62
|
-
"index_type": self.index_type,
|
63
|
-
"metric_type": self.metric_type,
|
64
|
-
"id_map": self._convert_int(self.id_map),
|
65
|
-
}
|
66
|
-
|
67
|
-
def _convert_int(self, id_map):
|
68
|
-
return {int(k): str(v) for k, v in id_map.items()}
|
69
|
-
|
70
|
-
@staticmethod
|
71
|
-
def _convert_int64(id_map):
|
72
|
-
return {np.int64(k): str(v) for k, v in id_map.items()}
|
73
|
-
|
74
|
-
def save(self, save_dir):
|
75
|
-
save_dir = Path(save_dir)
|
76
|
-
save_dir.mkdir(parents=True, exist_ok=True)
|
77
|
-
vector_path = (save_dir / f"{self.VECTOR_FN}{self.VECTOR_SUFFIX}").as_posix()
|
78
|
-
index_info_path = (save_dir / f"{self.IDMAP_FN}{self.IDMAP_SUFFIX}").as_posix()
|
79
|
-
|
80
|
-
if self.metric_type in FaissBuilder.BINARY_METRIC_TYPE:
|
81
|
-
faiss.write_index_binary(self.index, vector_path)
|
82
|
-
else:
|
83
|
-
faiss.write_index(self.index, vector_path)
|
84
|
-
|
85
|
-
yaml_writer = YAMLWriter()
|
86
|
-
yaml_writer.write(
|
87
|
-
index_info_path,
|
88
|
-
self.index_info,
|
89
|
-
default_flow_style=False,
|
90
|
-
allow_unicode=True,
|
91
|
-
)
|
92
|
-
|
93
|
-
@classmethod
|
94
|
-
def load(cls, index):
|
95
|
-
if isinstance(index, str):
|
96
|
-
index_root = Path(index)
|
97
|
-
vector_path = index_root / f"{cls.VECTOR_FN}{cls.VECTOR_SUFFIX}"
|
98
|
-
index_info_path = index_root / f"{cls.IDMAP_FN}{cls.IDMAP_SUFFIX}"
|
99
|
-
|
100
|
-
assert (
|
101
|
-
vector_path.exists()
|
102
|
-
), f"Not found the {cls.VECTOR_FN}{cls.VECTOR_SUFFIX} file in {index}!"
|
103
|
-
assert (
|
104
|
-
index_info_path.exists()
|
105
|
-
), f"Not found the {cls.IDMAP_FN}{cls.IDMAP_SUFFIX} file in {index}!"
|
106
|
-
|
107
|
-
yaml_reader = YAMLReader()
|
108
|
-
index_info = yaml_reader.read(index_info_path)
|
109
|
-
assert (
|
110
|
-
"id_map" in index_info
|
111
|
-
and "metric_type" in index_info
|
112
|
-
and "index_type" in index_info
|
113
|
-
), f"The index_info file({index_info_path}) may have been damaged, `id_map` or `metric_type` or `index_type` not found in `index_info`."
|
114
|
-
id_map = IndexData._convert_int64(index_info["id_map"])
|
115
|
-
|
116
|
-
if index_info["metric_type"] in FaissBuilder.BINARY_METRIC_TYPE:
|
117
|
-
index = faiss.read_index_binary(vector_path.as_posix())
|
118
|
-
else:
|
119
|
-
index = faiss.read_index(vector_path.as_posix())
|
120
|
-
assert index.ntotal == len(
|
121
|
-
id_map
|
122
|
-
), "data number in index is not equal in in id_map"
|
123
|
-
|
124
|
-
return index, id_map, index_info["metric_type"], index_info["index_type"]
|
125
|
-
else:
|
126
|
-
assert isinstance(index, IndexData)
|
127
|
-
return index.index, index.id_map, index.metric_type, index.index_type
|
128
|
-
|
129
|
-
|
130
|
-
class FaissIndexer(BaseComponent):
|
131
|
-
|
132
|
-
INPUT_KEYS = "feature"
|
133
|
-
OUTPUT_KEYS = ["label", "score"]
|
134
|
-
DEAULT_INPUTS = {"feature": "feature"}
|
135
|
-
DEAULT_OUTPUTS = {"label": "label", "score": "score"}
|
136
|
-
|
137
|
-
ENABLE_BATCH = True
|
138
|
-
|
139
|
-
def __init__(
|
140
|
-
self,
|
141
|
-
index,
|
142
|
-
return_k=1,
|
143
|
-
score_thres=None,
|
144
|
-
hamming_radius=None,
|
145
|
-
):
|
146
|
-
super().__init__()
|
147
|
-
self._indexer, self.id_map, self.metric_type, index_type = IndexData.load(index)
|
148
|
-
self.return_k = return_k
|
149
|
-
if self.metric_type in FaissBuilder.BINARY_METRIC_TYPE:
|
150
|
-
self.hamming_radius = hamming_radius
|
151
|
-
else:
|
152
|
-
self.score_thres = score_thres
|
153
|
-
|
154
|
-
def apply(self, feature):
|
155
|
-
"""apply"""
|
156
|
-
scores_list, ids_list = self._indexer.search(np.array(feature), self.return_k)
|
157
|
-
preds = []
|
158
|
-
for scores, ids in zip(scores_list, ids_list):
|
159
|
-
labels = []
|
160
|
-
for id in ids:
|
161
|
-
if id > 0:
|
162
|
-
labels.append(self.id_map[id])
|
163
|
-
preds.append({"score": scores, "label": labels})
|
164
|
-
|
165
|
-
if self.metric_type in FaissBuilder.BINARY_METRIC_TYPE:
|
166
|
-
idxs = np.where(scores_list[:, 0] > self.hamming_radius)[0]
|
167
|
-
else:
|
168
|
-
idxs = np.where(scores_list[:, 0] < self.score_thres)[0]
|
169
|
-
for idx in idxs:
|
170
|
-
preds[idx] = {"score": None, "label": None}
|
171
|
-
return preds
|
172
|
-
|
173
|
-
|
174
|
-
class FaissBuilder:
|
175
|
-
|
176
|
-
SUPPORT_METRIC_TYPE = ("hamming", "IP", "L2")
|
177
|
-
SUPPORT_INDEX_TYPE = ("Flat", "IVF", "HNSW32")
|
178
|
-
BINARY_METRIC_TYPE = ("hamming",)
|
179
|
-
BINARY_SUPPORT_INDEX_TYPE = ("Flat", "IVF", "BinaryHash")
|
180
|
-
|
181
|
-
@classmethod
|
182
|
-
def _get_index_type(cls, metric_type, index_type, num=None):
|
183
|
-
# if IVF method, cal ivf number automaticlly
|
184
|
-
if index_type == "IVF":
|
185
|
-
index_type = index_type + str(min(int(num // 8), 65536))
|
186
|
-
if metric_type in cls.BINARY_METRIC_TYPE:
|
187
|
-
index_type += ",BFlat"
|
188
|
-
else:
|
189
|
-
index_type += ",Flat"
|
190
|
-
|
191
|
-
# for binary index, add B at head of index_type
|
192
|
-
if metric_type in cls.BINARY_METRIC_TYPE:
|
193
|
-
assert (
|
194
|
-
index_type in cls.BINARY_SUPPORT_INDEX_TYPE
|
195
|
-
), f"The metric type({metric_type}) only support {cls.BINARY_SUPPORT_INDEX_TYPE} index types!"
|
196
|
-
index_type = "B" + index_type
|
197
|
-
|
198
|
-
if index_type == "HNSW32":
|
199
|
-
logging.warning("The HNSW32 method dose not support 'remove' operation")
|
200
|
-
index_type = "HNSW32"
|
201
|
-
|
202
|
-
if index_type == "Flat":
|
203
|
-
index_type = "Flat"
|
204
|
-
|
205
|
-
return index_type
|
206
|
-
|
207
|
-
@classmethod
|
208
|
-
def _get_metric_type(cls, metric_type):
|
209
|
-
if metric_type == "hamming":
|
210
|
-
return faiss.METRIC_Hamming
|
211
|
-
elif metric_type == "jaccard":
|
212
|
-
return faiss.METRIC_Jaccard
|
213
|
-
elif metric_type == "IP":
|
214
|
-
return faiss.METRIC_INNER_PRODUCT
|
215
|
-
elif metric_type == "L2":
|
216
|
-
return faiss.METRIC_L2
|
217
|
-
|
218
|
-
@classmethod
|
219
|
-
def build(
|
220
|
-
cls,
|
221
|
-
gallery_imgs,
|
222
|
-
gallery_label,
|
223
|
-
predict_func,
|
224
|
-
metric_type="IP",
|
225
|
-
index_type="HNSW32",
|
226
|
-
):
|
227
|
-
assert (
|
228
|
-
index_type in cls.SUPPORT_INDEX_TYPE
|
229
|
-
), f"Supported index types only: {cls.SUPPORT_INDEX_TYPE}!"
|
230
|
-
|
231
|
-
assert (
|
232
|
-
metric_type in cls.SUPPORT_METRIC_TYPE
|
233
|
-
), f"Supported metric types only: {cls.SUPPORT_METRIC_TYPE}!"
|
234
|
-
|
235
|
-
if isinstance(gallery_label, str):
|
236
|
-
gallery_docs, gallery_list = cls.load_gallery(gallery_label, gallery_imgs)
|
237
|
-
else:
|
238
|
-
gallery_docs, gallery_list = gallery_label, gallery_imgs
|
239
|
-
|
240
|
-
features = [res["feature"] for res in predict_func(gallery_list)]
|
241
|
-
dtype = np.uint8 if metric_type in cls.BINARY_METRIC_TYPE else np.float32
|
242
|
-
features = np.array(features).astype(dtype)
|
243
|
-
vector_num, vector_dim = features.shape
|
244
|
-
|
245
|
-
if metric_type in cls.BINARY_METRIC_TYPE:
|
246
|
-
index = faiss.index_binary_factory(
|
247
|
-
vector_dim,
|
248
|
-
cls._get_index_type(metric_type, index_type, vector_num),
|
249
|
-
cls._get_metric_type(metric_type),
|
250
|
-
)
|
251
|
-
else:
|
252
|
-
index = faiss.index_factory(
|
253
|
-
vector_dim,
|
254
|
-
cls._get_index_type(metric_type, index_type, vector_num),
|
255
|
-
cls._get_metric_type(metric_type),
|
256
|
-
)
|
257
|
-
index = faiss.IndexIDMap2(index)
|
258
|
-
ids = {}
|
259
|
-
|
260
|
-
# calculate id for new data
|
261
|
-
index, ids = cls._add_gallery(
|
262
|
-
metric_type, index, ids, features, gallery_docs, mode="new"
|
263
|
-
)
|
264
|
-
return IndexData(
|
265
|
-
index, {"id_map": ids, "metric_type": metric_type, "index_type": index_type}
|
266
|
-
)
|
267
|
-
|
268
|
-
@classmethod
|
269
|
-
def remove(
|
270
|
-
cls,
|
271
|
-
remove_ids,
|
272
|
-
index,
|
273
|
-
):
|
274
|
-
index, ids, metric_type, index_type = IndexData.load(index)
|
275
|
-
if index_type == "HNSW32":
|
276
|
-
raise RuntimeError(
|
277
|
-
"The index_type: HNSW32 dose not support 'remove' operation"
|
278
|
-
)
|
279
|
-
if isinstance(remove_ids, str):
|
280
|
-
lines = []
|
281
|
-
with open(remove_ids) as f:
|
282
|
-
lines = f.readlines()
|
283
|
-
remove_ids = []
|
284
|
-
for line in lines:
|
285
|
-
id_ = int(line.strip().split(" ")[0])
|
286
|
-
remove_ids.append(id_)
|
287
|
-
remove_ids = np.asarray(remove_ids)
|
288
|
-
else:
|
289
|
-
remove_ids = np.asarray(remove_ids)
|
290
|
-
|
291
|
-
# remove ids in id_map, remove index data in faiss index
|
292
|
-
index.remove_ids(remove_ids)
|
293
|
-
ids = {k: v for k, v in ids.items() if k not in remove_ids}
|
294
|
-
return IndexData(
|
295
|
-
index, {"id_map": ids, "metric_type": metric_type, "index_type": index_type}
|
296
|
-
)
|
297
|
-
|
298
|
-
@classmethod
|
299
|
-
def append(cls, gallery_imgs, gallery_label, predict_func, index):
|
300
|
-
index, ids, metric_type, index_type = IndexData.load(index)
|
301
|
-
assert (
|
302
|
-
metric_type in cls.SUPPORT_METRIC_TYPE
|
303
|
-
), f"Supported metric types only: {cls.SUPPORT_METRIC_TYPE}!"
|
304
|
-
|
305
|
-
if isinstance(gallery_label, str):
|
306
|
-
gallery_docs, gallery_list = cls.load_gallery(gallery_label, gallery_imgs)
|
307
|
-
else:
|
308
|
-
gallery_docs, gallery_list = gallery_label, gallery_imgs
|
309
|
-
|
310
|
-
features = [res["feature"] for res in predict_func(gallery_list)]
|
311
|
-
dtype = np.uint8 if metric_type in cls.BINARY_METRIC_TYPE else np.float32
|
312
|
-
features = np.array(features).astype(dtype)
|
313
|
-
|
314
|
-
# calculate id for new data
|
315
|
-
index, ids = cls._add_gallery(
|
316
|
-
metric_type, index, ids, features, gallery_docs, mode="append"
|
317
|
-
)
|
318
|
-
return IndexData(
|
319
|
-
index, {"id_map": ids, "metric_type": metric_type, "index_type": index_type}
|
320
|
-
)
|
321
|
-
|
322
|
-
@classmethod
|
323
|
-
def _add_gallery(
|
324
|
-
cls, metric_type, index, ids, gallery_features, gallery_docs, mode
|
325
|
-
):
|
326
|
-
start_id = max(ids.keys()) + 1 if ids else 0
|
327
|
-
ids_now = (np.arange(0, len(gallery_docs)) + start_id).astype(np.int64)
|
328
|
-
|
329
|
-
# only train when new index file
|
330
|
-
if mode == "new":
|
331
|
-
if metric_type in cls.BINARY_METRIC_TYPE:
|
332
|
-
index.add(gallery_features)
|
333
|
-
else:
|
334
|
-
index.train(gallery_features)
|
335
|
-
|
336
|
-
if metric_type not in cls.BINARY_METRIC_TYPE:
|
337
|
-
index.add_with_ids(gallery_features, ids_now)
|
338
|
-
# TODO(gaotingquan): how append when using hamming metric type
|
339
|
-
# else:
|
340
|
-
# pass
|
341
|
-
|
342
|
-
for i, d in zip(list(ids_now), gallery_docs):
|
343
|
-
ids[i] = d
|
344
|
-
return index, ids
|
345
|
-
|
346
|
-
@classmethod
|
347
|
-
def load_gallery(cls, gallery_label_path, gallery_imgs_root="", delimiter=" "):
|
348
|
-
lines = []
|
349
|
-
files = []
|
350
|
-
labels = []
|
351
|
-
root = Path(gallery_imgs_root)
|
352
|
-
with open(gallery_label_path, "r", encoding="utf-8") as f:
|
353
|
-
lines = f.readlines()
|
354
|
-
for line in lines:
|
355
|
-
path, label = line.strip().split(delimiter)
|
356
|
-
file_path = root / path
|
357
|
-
files.append(file_path.as_posix())
|
358
|
-
labels.append(label)
|
359
|
-
return labels, files
|