paddlex 2.1.0__py3-none-any.whl → 3.0.0b2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- paddlex/.version +1 -0
- paddlex/__init__.py +51 -19
- paddlex/__main__.py +40 -0
- paddlex/configs/anomaly_detection/STFPM.yaml +41 -0
- paddlex/configs/doc_text_orientation/PP-LCNet_x1_0_doc_ori.yaml +41 -0
- paddlex/configs/face_detection/BlazeFace-FPN-SSH.yaml +40 -0
- paddlex/configs/face_detection/BlazeFace.yaml +40 -0
- paddlex/configs/face_detection/PP-YOLOE_plus-S_face.yaml +40 -0
- paddlex/configs/face_detection/PicoDet_LCNet_x2_5_face.yaml +40 -0
- paddlex/configs/face_recognition/MobileFaceNet.yaml +44 -0
- paddlex/configs/face_recognition/ResNet50_face.yaml +44 -0
- paddlex/configs/formula_recognition/LaTeX_OCR_rec.yaml +40 -0
- paddlex/configs/general_recognition/PP-ShiTuV2_rec.yaml +42 -0
- paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_base.yaml +42 -0
- paddlex/configs/general_recognition/PP-ShiTuV2_rec_CLIP_vit_large.yaml +41 -0
- paddlex/configs/human_detection/PP-YOLOE-L_human.yaml +42 -0
- paddlex/configs/human_detection/PP-YOLOE-S_human.yaml +42 -0
- paddlex/configs/image_classification/CLIP_vit_base_patch16_224.yaml +41 -0
- paddlex/configs/image_classification/CLIP_vit_large_patch14_224.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_base_224.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_base_384.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_large_224.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_large_384.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_small.yaml +41 -0
- paddlex/configs/image_classification/ConvNeXt_tiny.yaml +41 -0
- paddlex/configs/image_classification/FasterNet-L.yaml +40 -0
- paddlex/configs/image_classification/FasterNet-M.yaml +40 -0
- paddlex/configs/image_classification/FasterNet-S.yaml +40 -0
- paddlex/configs/image_classification/FasterNet-T0.yaml +40 -0
- paddlex/configs/image_classification/FasterNet-T1.yaml +40 -0
- paddlex/configs/image_classification/FasterNet-T2.yaml +40 -0
- paddlex/configs/image_classification/MobileNetV1_x0_25.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV1_x0_5.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV1_x0_75.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV1_x1_0.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV2_x0_25.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV2_x0_5.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV2_x1_0.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV2_x1_5.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV2_x2_0.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_large_x0_35.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_large_x0_5.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_large_x0_75.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_large_x1_0.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_large_x1_25.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_small_x0_35.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_small_x0_5.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_small_x0_75.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_small_x1_0.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV3_small_x1_25.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV4_conv_large.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV4_conv_medium.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV4_conv_small.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV4_hybrid_large.yaml +41 -0
- paddlex/configs/image_classification/MobileNetV4_hybrid_medium.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B0.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B1.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B2.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B3.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B4.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B5.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNetV2-B6.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNet_base.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNet_small.yaml +41 -0
- paddlex/configs/image_classification/PP-HGNet_tiny.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNetV2_base.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNetV2_large.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNetV2_small.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x0_25.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x0_35.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x0_5.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x0_75.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x1_0.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x1_5.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x2_0.yaml +41 -0
- paddlex/configs/image_classification/PP-LCNet_x2_5.yaml +41 -0
- paddlex/configs/image_classification/ResNet101.yaml +41 -0
- paddlex/configs/image_classification/ResNet101_vd.yaml +41 -0
- paddlex/configs/image_classification/ResNet152.yaml +41 -0
- paddlex/configs/image_classification/ResNet152_vd.yaml +41 -0
- paddlex/configs/image_classification/ResNet18.yaml +41 -0
- paddlex/configs/image_classification/ResNet18_vd.yaml +41 -0
- paddlex/configs/image_classification/ResNet200_vd.yaml +41 -0
- paddlex/configs/image_classification/ResNet34.yaml +41 -0
- paddlex/configs/image_classification/ResNet34_vd.yaml +41 -0
- paddlex/configs/image_classification/ResNet50.yaml +41 -0
- paddlex/configs/image_classification/ResNet50_vd.yaml +41 -0
- paddlex/configs/image_classification/StarNet-S1.yaml +41 -0
- paddlex/configs/image_classification/StarNet-S2.yaml +41 -0
- paddlex/configs/image_classification/StarNet-S3.yaml +41 -0
- paddlex/configs/image_classification/StarNet-S4.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_base_patch4_window12_384.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_base_patch4_window7_224.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_large_patch4_window12_384.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_large_patch4_window7_224.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_small_patch4_window7_224.yaml +41 -0
- paddlex/configs/image_classification/SwinTransformer_tiny_patch4_window7_224.yaml +41 -0
- paddlex/configs/image_unwarping/UVDoc.yaml +12 -0
- paddlex/configs/instance_segmentation/Cascade-MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/Mask-RT-DETR-H.yaml +40 -0
- paddlex/configs/instance_segmentation/Mask-RT-DETR-L.yaml +40 -0
- paddlex/configs/instance_segmentation/Mask-RT-DETR-M.yaml +40 -0
- paddlex/configs/instance_segmentation/Mask-RT-DETR-S.yaml +40 -0
- paddlex/configs/instance_segmentation/Mask-RT-DETR-X.yaml +40 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNeXt101-vd-FPN.yaml +39 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNet101-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNet101-vd-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNet50-vd-FPN.yaml +40 -0
- paddlex/configs/instance_segmentation/MaskRCNN-ResNet50.yaml +40 -0
- paddlex/configs/instance_segmentation/PP-YOLOE_seg-S.yaml +40 -0
- paddlex/configs/instance_segmentation/SOLOv2.yaml +40 -0
- paddlex/configs/mainbody_detection/PP-ShiTuV2_det.yaml +41 -0
- paddlex/configs/multilabel_classification/CLIP_vit_base_patch16_448_ML.yaml +41 -0
- paddlex/configs/multilabel_classification/PP-HGNetV2-B0_ML.yaml +41 -0
- paddlex/configs/multilabel_classification/PP-HGNetV2-B4_ML.yaml +41 -0
- paddlex/configs/multilabel_classification/PP-HGNetV2-B6_ML.yaml +41 -0
- paddlex/configs/multilabel_classification/PP-LCNet_x1_0_ML.yaml +41 -0
- paddlex/configs/multilabel_classification/ResNet50_ML.yaml +41 -0
- paddlex/configs/object_detection/Cascade-FasterRCNN-ResNet50-FPN.yaml +41 -0
- paddlex/configs/object_detection/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/object_detection/CenterNet-DLA-34.yaml +41 -0
- paddlex/configs/object_detection/CenterNet-ResNet50.yaml +41 -0
- paddlex/configs/object_detection/DETR-R50.yaml +42 -0
- paddlex/configs/object_detection/FCOS-ResNet50.yaml +41 -0
- paddlex/configs/object_detection/FasterRCNN-ResNeXt101-vd-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet101-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet101.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet34-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet50-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet50-vd-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-ResNet50.yaml +42 -0
- paddlex/configs/object_detection/FasterRCNN-Swin-Tiny-FPN.yaml +42 -0
- paddlex/configs/object_detection/PP-YOLOE_plus-L.yaml +40 -0
- paddlex/configs/object_detection/PP-YOLOE_plus-M.yaml +40 -0
- paddlex/configs/object_detection/PP-YOLOE_plus-S.yaml +40 -0
- paddlex/configs/object_detection/PP-YOLOE_plus-X.yaml +40 -0
- paddlex/configs/object_detection/PicoDet-L.yaml +40 -0
- paddlex/configs/object_detection/PicoDet-M.yaml +42 -0
- paddlex/configs/object_detection/PicoDet-S.yaml +40 -0
- paddlex/configs/object_detection/PicoDet-XS.yaml +42 -0
- paddlex/configs/object_detection/RT-DETR-H.yaml +40 -0
- paddlex/configs/object_detection/RT-DETR-L.yaml +40 -0
- paddlex/configs/object_detection/RT-DETR-R18.yaml +40 -0
- paddlex/configs/object_detection/RT-DETR-R50.yaml +40 -0
- paddlex/configs/object_detection/RT-DETR-X.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-L.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-M.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-N.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-S.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-T.yaml +40 -0
- paddlex/configs/object_detection/YOLOX-X.yaml +40 -0
- paddlex/configs/object_detection/YOLOv3-DarkNet53.yaml +40 -0
- paddlex/configs/object_detection/YOLOv3-MobileNetV3.yaml +40 -0
- paddlex/configs/object_detection/YOLOv3-ResNet50_vd_DCN.yaml +40 -0
- paddlex/configs/pedestrian_attribute/PP-LCNet_x1_0_pedestrian_attribute.yaml +41 -0
- paddlex/configs/semantic_segmentation/Deeplabv3-R101.yaml +40 -0
- paddlex/configs/semantic_segmentation/Deeplabv3-R50.yaml +40 -0
- paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R101.yaml +40 -0
- paddlex/configs/semantic_segmentation/Deeplabv3_Plus-R50.yaml +40 -0
- paddlex/configs/semantic_segmentation/OCRNet_HRNet-W18.yaml +40 -0
- paddlex/configs/semantic_segmentation/OCRNet_HRNet-W48.yaml +40 -0
- paddlex/configs/semantic_segmentation/PP-LiteSeg-B.yaml +41 -0
- paddlex/configs/semantic_segmentation/PP-LiteSeg-T.yaml +40 -0
- paddlex/configs/semantic_segmentation/SeaFormer_base.yaml +40 -0
- paddlex/configs/semantic_segmentation/SeaFormer_large.yaml +40 -0
- paddlex/configs/semantic_segmentation/SeaFormer_small.yaml +40 -0
- paddlex/configs/semantic_segmentation/SeaFormer_tiny.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B0.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B1.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B2.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B3.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B4.yaml +40 -0
- paddlex/configs/semantic_segmentation/SegFormer-B5.yaml +40 -0
- paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-L.yaml +42 -0
- paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-S.yaml +42 -0
- paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml +42 -0
- paddlex/configs/structure_analysis/PicoDet-L_layout_17cls.yaml +40 -0
- paddlex/configs/structure_analysis/PicoDet-L_layout_3cls.yaml +40 -0
- paddlex/configs/structure_analysis/PicoDet-S_layout_17cls.yaml +40 -0
- paddlex/configs/structure_analysis/PicoDet-S_layout_3cls.yaml +40 -0
- paddlex/configs/structure_analysis/PicoDet_layout_1x.yaml +40 -0
- paddlex/configs/structure_analysis/PicoDet_layout_1x_table.yaml +40 -0
- paddlex/configs/structure_analysis/RT-DETR-H_layout_17cls.yaml +40 -0
- paddlex/configs/structure_analysis/RT-DETR-H_layout_3cls.yaml +40 -0
- paddlex/configs/table_recognition/SLANet.yaml +39 -0
- paddlex/configs/table_recognition/SLANet_plus.yaml +39 -0
- paddlex/configs/text_detection/PP-OCRv4_mobile_det.yaml +40 -0
- paddlex/configs/text_detection/PP-OCRv4_server_det.yaml +40 -0
- paddlex/configs/text_detection_seal/PP-OCRv4_mobile_seal_det.yaml +40 -0
- paddlex/configs/text_detection_seal/PP-OCRv4_server_seal_det.yaml +40 -0
- paddlex/configs/text_recognition/PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/text_recognition/PP-OCRv4_server_rec.yaml +39 -0
- paddlex/configs/text_recognition/ch_RepSVTR_rec.yaml +39 -0
- paddlex/configs/text_recognition/ch_SVTRv2_rec.yaml +39 -0
- paddlex/configs/ts_anomaly_detection/AutoEncoder_ad.yaml +37 -0
- paddlex/configs/ts_anomaly_detection/DLinear_ad.yaml +37 -0
- paddlex/configs/ts_anomaly_detection/Nonstationary_ad.yaml +37 -0
- paddlex/configs/ts_anomaly_detection/PatchTST_ad.yaml +37 -0
- paddlex/configs/ts_anomaly_detection/TimesNet_ad.yaml +37 -0
- paddlex/configs/ts_classification/TimesNet_cls.yaml +37 -0
- paddlex/configs/ts_forecast/DLinear.yaml +38 -0
- paddlex/configs/ts_forecast/NLinear.yaml +38 -0
- paddlex/configs/ts_forecast/Nonstationary.yaml +38 -0
- paddlex/configs/ts_forecast/PatchTST.yaml +38 -0
- paddlex/configs/ts_forecast/RLinear.yaml +38 -0
- paddlex/configs/ts_forecast/TiDE.yaml +38 -0
- paddlex/configs/ts_forecast/TimesNet.yaml +38 -0
- paddlex/configs/vehicle_attribute/PP-LCNet_x1_0_vehicle_attribute.yaml +41 -0
- paddlex/configs/vehicle_detection/PP-YOLOE-L_vehicle.yaml +41 -0
- paddlex/configs/vehicle_detection/PP-YOLOE-S_vehicle.yaml +42 -0
- paddlex/engine.py +54 -0
- paddlex/inference/__init__.py +17 -0
- paddlex/inference/components/__init__.py +18 -0
- paddlex/inference/components/base.py +292 -0
- paddlex/inference/components/llm/__init__.py +25 -0
- paddlex/inference/components/llm/base.py +65 -0
- paddlex/inference/components/llm/erniebot.py +212 -0
- paddlex/inference/components/paddle_predictor/__init__.py +20 -0
- paddlex/inference/components/paddle_predictor/predictor.py +332 -0
- paddlex/inference/components/retrieval/__init__.py +15 -0
- paddlex/inference/components/retrieval/faiss.py +359 -0
- paddlex/inference/components/task_related/__init__.py +33 -0
- paddlex/inference/components/task_related/clas.py +124 -0
- paddlex/inference/components/task_related/det.py +284 -0
- paddlex/inference/components/task_related/instance_seg.py +89 -0
- paddlex/inference/components/task_related/seal_det_warp.py +940 -0
- paddlex/inference/components/task_related/seg.py +40 -0
- paddlex/inference/components/task_related/table_rec.py +191 -0
- paddlex/inference/components/task_related/text_det.py +895 -0
- paddlex/inference/components/task_related/text_rec.py +353 -0
- paddlex/inference/components/task_related/warp.py +43 -0
- paddlex/inference/components/transforms/__init__.py +16 -0
- paddlex/inference/components/transforms/image/__init__.py +15 -0
- paddlex/inference/components/transforms/image/common.py +598 -0
- paddlex/inference/components/transforms/image/funcs.py +58 -0
- paddlex/inference/components/transforms/read_data.py +67 -0
- paddlex/inference/components/transforms/ts/__init__.py +15 -0
- paddlex/inference/components/transforms/ts/common.py +393 -0
- paddlex/inference/components/transforms/ts/funcs.py +424 -0
- paddlex/inference/models/__init__.py +106 -0
- paddlex/inference/models/anomaly_detection.py +87 -0
- paddlex/inference/models/base/__init__.py +16 -0
- paddlex/inference/models/base/base_predictor.py +76 -0
- paddlex/inference/models/base/basic_predictor.py +122 -0
- paddlex/inference/models/face_recognition.py +21 -0
- paddlex/inference/models/formula_recognition.py +55 -0
- paddlex/inference/models/general_recognition.py +99 -0
- paddlex/inference/models/image_classification.py +101 -0
- paddlex/inference/models/image_unwarping.py +43 -0
- paddlex/inference/models/instance_segmentation.py +66 -0
- paddlex/inference/models/multilabel_classification.py +33 -0
- paddlex/inference/models/object_detection.py +129 -0
- paddlex/inference/models/semantic_segmentation.py +86 -0
- paddlex/inference/models/table_recognition.py +106 -0
- paddlex/inference/models/text_detection.py +105 -0
- paddlex/inference/models/text_recognition.py +78 -0
- paddlex/inference/models/ts_ad.py +68 -0
- paddlex/inference/models/ts_cls.py +57 -0
- paddlex/inference/models/ts_fc.py +73 -0
- paddlex/inference/pipelines/__init__.py +127 -0
- paddlex/inference/pipelines/attribute_recognition.py +92 -0
- paddlex/inference/pipelines/base.py +86 -0
- paddlex/inference/pipelines/face_recognition.py +49 -0
- paddlex/inference/pipelines/formula_recognition.py +102 -0
- paddlex/inference/pipelines/layout_parsing/__init__.py +15 -0
- paddlex/inference/pipelines/layout_parsing/layout_parsing.py +362 -0
- paddlex/inference/pipelines/ocr.py +80 -0
- paddlex/inference/pipelines/pp_shitu_v2.py +152 -0
- paddlex/inference/pipelines/ppchatocrv3/__init__.py +15 -0
- paddlex/inference/pipelines/ppchatocrv3/ch_prompt.yaml +14 -0
- paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py +717 -0
- paddlex/inference/pipelines/ppchatocrv3/utils.py +168 -0
- paddlex/inference/pipelines/seal_recognition.py +152 -0
- paddlex/inference/pipelines/serving/__init__.py +17 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/__init__.py +205 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/anomaly_detection.py +80 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/face_recognition.py +317 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/formula_recognition.py +119 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/image_classification.py +101 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/instance_segmentation.py +112 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/layout_parsing.py +205 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/multi_label_image_classification.py +90 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/object_detection.py +90 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/ocr.py +98 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/pedestrian_attribute_recognition.py +102 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/pp_shitu_v2.py +319 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/ppchatocrv3.py +445 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/seal_recognition.py +110 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/semantic_segmentation.py +82 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/small_object_detection.py +92 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/table_recognition.py +110 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_ad.py +68 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_cls.py +68 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/ts_fc.py +68 -0
- paddlex/inference/pipelines/serving/_pipeline_apps/vehicle_attribute_recognition.py +102 -0
- paddlex/inference/pipelines/serving/app.py +164 -0
- paddlex/inference/pipelines/serving/models.py +30 -0
- paddlex/inference/pipelines/serving/server.py +25 -0
- paddlex/inference/pipelines/serving/storage.py +161 -0
- paddlex/inference/pipelines/serving/utils.py +190 -0
- paddlex/inference/pipelines/single_model_pipeline.py +76 -0
- paddlex/inference/pipelines/table_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/table_recognition/table_recognition.py +193 -0
- paddlex/inference/pipelines/table_recognition/utils.py +457 -0
- paddlex/inference/results/__init__.py +31 -0
- paddlex/inference/results/attribute_rec.py +89 -0
- paddlex/inference/results/base.py +43 -0
- paddlex/inference/results/chat_ocr.py +158 -0
- paddlex/inference/results/clas.py +133 -0
- paddlex/inference/results/det.py +86 -0
- paddlex/inference/results/face_rec.py +34 -0
- paddlex/inference/results/formula_rec.py +363 -0
- paddlex/inference/results/instance_seg.py +152 -0
- paddlex/inference/results/ocr.py +157 -0
- paddlex/inference/results/seal_rec.py +50 -0
- paddlex/inference/results/seg.py +72 -0
- paddlex/inference/results/shitu.py +35 -0
- paddlex/inference/results/table_rec.py +109 -0
- paddlex/inference/results/text_det.py +33 -0
- paddlex/inference/results/text_rec.py +66 -0
- paddlex/inference/results/ts.py +37 -0
- paddlex/inference/results/utils/__init__.py +13 -0
- paddlex/inference/results/utils/mixin.py +204 -0
- paddlex/inference/results/warp.py +31 -0
- paddlex/inference/utils/__init__.py +13 -0
- paddlex/inference/utils/benchmark.py +214 -0
- paddlex/inference/utils/color_map.py +123 -0
- paddlex/inference/utils/get_pipeline_path.py +26 -0
- paddlex/inference/utils/io/__init__.py +33 -0
- paddlex/inference/utils/io/readers.py +353 -0
- paddlex/inference/utils/io/style.py +374 -0
- paddlex/inference/utils/io/tablepyxl.py +149 -0
- paddlex/inference/utils/io/writers.py +376 -0
- paddlex/inference/utils/new_ir_blacklist.py +22 -0
- paddlex/inference/utils/official_models.py +286 -0
- paddlex/inference/utils/pp_option.py +236 -0
- paddlex/inference/utils/process_hook.py +54 -0
- paddlex/model.py +106 -0
- paddlex/modules/__init__.py +105 -0
- paddlex/modules/anomaly_detection/__init__.py +18 -0
- paddlex/modules/anomaly_detection/dataset_checker/__init__.py +95 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +79 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/check_dataset.py +87 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +230 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/visualizer.py +71 -0
- paddlex/modules/anomaly_detection/evaluator.py +58 -0
- paddlex/modules/anomaly_detection/exportor.py +22 -0
- paddlex/modules/anomaly_detection/model_list.py +16 -0
- paddlex/modules/anomaly_detection/trainer.py +71 -0
- paddlex/modules/base/__init__.py +18 -0
- paddlex/modules/base/build_model.py +34 -0
- paddlex/modules/base/dataset_checker/__init__.py +16 -0
- paddlex/modules/base/dataset_checker/dataset_checker.py +164 -0
- paddlex/modules/base/dataset_checker/utils.py +110 -0
- paddlex/modules/base/evaluator.py +154 -0
- paddlex/modules/base/exportor.py +121 -0
- paddlex/modules/base/trainer.py +111 -0
- paddlex/modules/face_recognition/__init__.py +18 -0
- paddlex/modules/face_recognition/dataset_checker/__init__.py +71 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/__init__.py +16 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/check_dataset.py +174 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/visualizer.py +156 -0
- paddlex/modules/face_recognition/evaluator.py +52 -0
- paddlex/modules/face_recognition/exportor.py +22 -0
- paddlex/modules/face_recognition/model_list.py +15 -0
- paddlex/modules/face_recognition/trainer.py +97 -0
- paddlex/modules/formula_recognition/__init__.py +13 -0
- paddlex/modules/formula_recognition/model_list.py +17 -0
- paddlex/modules/general_recognition/__init__.py +18 -0
- paddlex/modules/general_recognition/dataset_checker/__init__.py +107 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/analyse_dataset.py +98 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/check_dataset.py +100 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/convert_dataset.py +99 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/visualizer.py +150 -0
- paddlex/modules/general_recognition/evaluator.py +31 -0
- paddlex/modules/general_recognition/exportor.py +22 -0
- paddlex/modules/general_recognition/model_list.py +19 -0
- paddlex/modules/general_recognition/trainer.py +52 -0
- paddlex/modules/image_classification/__init__.py +18 -0
- paddlex/modules/image_classification/dataset_checker/__init__.py +104 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/analyse_dataset.py +93 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/check_dataset.py +131 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/convert_dataset.py +51 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/visualizer.py +156 -0
- paddlex/modules/image_classification/evaluator.py +43 -0
- paddlex/modules/image_classification/exportor.py +22 -0
- paddlex/modules/image_classification/model_list.py +97 -0
- paddlex/modules/image_classification/trainer.py +82 -0
- paddlex/modules/image_unwarping/__init__.py +13 -0
- paddlex/modules/image_unwarping/model_list.py +17 -0
- paddlex/modules/instance_segmentation/__init__.py +18 -0
- paddlex/modules/instance_segmentation/dataset_checker/__init__.py +93 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/analyse_dataset.py +78 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/check_dataset.py +92 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/convert_dataset.py +241 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/split_dataset.py +119 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/visualizer.py +221 -0
- paddlex/modules/instance_segmentation/evaluator.py +32 -0
- paddlex/modules/instance_segmentation/exportor.py +22 -0
- paddlex/modules/instance_segmentation/model_list.py +33 -0
- paddlex/modules/instance_segmentation/trainer.py +31 -0
- paddlex/modules/multilabel_classification/__init__.py +18 -0
- paddlex/modules/multilabel_classification/dataset_checker/__init__.py +106 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/analyse_dataset.py +95 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/check_dataset.py +131 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/convert_dataset.py +117 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/multilabel_classification/evaluator.py +43 -0
- paddlex/modules/multilabel_classification/exportor.py +22 -0
- paddlex/modules/multilabel_classification/model_list.py +24 -0
- paddlex/modules/multilabel_classification/trainer.py +85 -0
- paddlex/modules/object_detection/__init__.py +18 -0
- paddlex/modules/object_detection/dataset_checker/__init__.py +115 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/analyse_dataset.py +80 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/check_dataset.py +86 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/convert_dataset.py +433 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/split_dataset.py +119 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/visualizer.py +192 -0
- paddlex/modules/object_detection/evaluator.py +41 -0
- paddlex/modules/object_detection/exportor.py +22 -0
- paddlex/modules/object_detection/model_list.py +74 -0
- paddlex/modules/object_detection/trainer.py +85 -0
- paddlex/modules/semantic_segmentation/__init__.py +18 -0
- paddlex/modules/semantic_segmentation/dataset_checker/__init__.py +95 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/analyse_dataset.py +73 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/check_dataset.py +80 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/convert_dataset.py +162 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/visualizer.py +71 -0
- paddlex/modules/semantic_segmentation/evaluator.py +58 -0
- paddlex/modules/semantic_segmentation/exportor.py +22 -0
- paddlex/modules/semantic_segmentation/model_list.py +35 -0
- paddlex/modules/semantic_segmentation/trainer.py +71 -0
- paddlex/modules/table_recognition/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/__init__.py +83 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/analyse_dataset.py +58 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py +87 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/split_dataset.py +79 -0
- paddlex/modules/table_recognition/evaluator.py +43 -0
- paddlex/modules/table_recognition/exportor.py +22 -0
- paddlex/modules/table_recognition/model_list.py +19 -0
- paddlex/modules/table_recognition/trainer.py +70 -0
- paddlex/modules/text_detection/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/__init__.py +94 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/analyse_dataset.py +217 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py +96 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/split_dataset.py +140 -0
- paddlex/modules/text_detection/evaluator.py +41 -0
- paddlex/modules/text_detection/exportor.py +22 -0
- paddlex/modules/text_detection/model_list.py +22 -0
- paddlex/modules/text_detection/trainer.py +68 -0
- paddlex/modules/text_recognition/__init__.py +18 -0
- paddlex/modules/text_recognition/dataset_checker/__init__.py +114 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/analyse_dataset.py +161 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py +97 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/convert_dataset.py +94 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/text_recognition/evaluator.py +63 -0
- paddlex/modules/text_recognition/exportor.py +25 -0
- paddlex/modules/text_recognition/model_list.py +20 -0
- paddlex/modules/text_recognition/trainer.py +105 -0
- paddlex/modules/ts_anomaly_detection/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py +97 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +27 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +78 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_anomaly_detection/evaluator.py +67 -0
- paddlex/modules/ts_anomaly_detection/exportor.py +45 -0
- paddlex/modules/ts_anomaly_detection/model_list.py +22 -0
- paddlex/modules/ts_anomaly_detection/trainer.py +97 -0
- paddlex/modules/ts_classification/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/__init__.py +97 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/analyse_dataset.py +74 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/convert_dataset.py +78 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/split_dataset.py +88 -0
- paddlex/modules/ts_classification/evaluator.py +66 -0
- paddlex/modules/ts_classification/exportor.py +45 -0
- paddlex/modules/ts_classification/model_list.py +18 -0
- paddlex/modules/ts_classification/trainer.py +92 -0
- paddlex/modules/ts_forecast/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/__init__.py +97 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/analyse_dataset.py +27 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/convert_dataset.py +77 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_forecast/evaluator.py +66 -0
- paddlex/modules/ts_forecast/exportor.py +45 -0
- paddlex/modules/ts_forecast/model_list.py +24 -0
- paddlex/modules/ts_forecast/trainer.py +92 -0
- paddlex/paddlex_cli.py +197 -0
- paddlex/pipelines/OCR.yaml +8 -0
- paddlex/pipelines/PP-ChatOCRv3-doc.yaml +27 -0
- paddlex/pipelines/PP-ShiTuV2.yaml +13 -0
- paddlex/pipelines/anomaly_detection.yaml +7 -0
- paddlex/pipelines/face_recognition.yaml +13 -0
- paddlex/pipelines/formula_recognition.yaml +8 -0
- paddlex/pipelines/image_classification.yaml +7 -0
- paddlex/pipelines/instance_segmentation.yaml +7 -0
- paddlex/pipelines/layout_parsing.yaml +14 -0
- paddlex/pipelines/multi_label_image_classification.yaml +7 -0
- paddlex/pipelines/object_detection.yaml +7 -0
- paddlex/pipelines/pedestrian_attribute_recognition.yaml +7 -0
- paddlex/pipelines/seal_recognition.yaml +10 -0
- paddlex/pipelines/semantic_segmentation.yaml +7 -0
- paddlex/pipelines/small_object_detection.yaml +7 -0
- paddlex/pipelines/table_recognition.yaml +12 -0
- paddlex/pipelines/ts_ad.yaml +7 -0
- paddlex/pipelines/ts_cls.yaml +7 -0
- paddlex/pipelines/ts_fc.yaml +7 -0
- paddlex/pipelines/vehicle_attribute_recognition.yaml +7 -0
- paddlex/repo_apis/PaddleClas_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleClas_api/cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleClas_api/cls/config.py +594 -0
- paddlex/repo_apis/PaddleClas_api/cls/model.py +349 -0
- paddlex/repo_apis/PaddleClas_api/cls/register.py +890 -0
- paddlex/repo_apis/PaddleClas_api/cls/runner.py +219 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/__init__.py +18 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/config.py +141 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/model.py +23 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/register.py +68 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/runner.py +55 -0
- paddlex/repo_apis/PaddleDetection_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleDetection_api/config_helper.py +280 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/__init__.py +18 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/config.py +454 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py +397 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/register.py +263 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/runner.py +226 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/config.py +517 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/model.py +424 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py +139 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/register.py +927 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/runner.py +226 -0
- paddlex/repo_apis/PaddleNLP_api/__init__.py +13 -0
- paddlex/repo_apis/PaddleOCR_api/__init__.py +20 -0
- paddlex/repo_apis/PaddleOCR_api/config_utils.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/config.py +64 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/model.py +126 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/register.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/runner.py +51 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/config.py +62 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/model.py +72 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/register.py +72 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/runner.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/config.py +542 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/model.py +396 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/register.py +80 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/runner.py +240 -0
- paddlex/repo_apis/PaddleSeg_api/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/base_seg_config.py +134 -0
- paddlex/repo_apis/PaddleSeg_api/seg/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/seg/config.py +177 -0
- paddlex/repo_apis/PaddleSeg_api/seg/model.py +481 -0
- paddlex/repo_apis/PaddleSeg_api/seg/register.py +253 -0
- paddlex/repo_apis/PaddleSeg_api/seg/runner.py +262 -0
- paddlex/repo_apis/PaddleTS_api/__init__.py +19 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/config.py +89 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/register.py +146 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/__init__.py +13 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/config.py +222 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/model.py +272 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/config.py +73 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/register.py +59 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/config.py +137 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/register.py +186 -0
- paddlex/repo_apis/__init__.py +13 -0
- paddlex/repo_apis/base/__init__.py +23 -0
- paddlex/repo_apis/base/config.py +238 -0
- paddlex/repo_apis/base/model.py +571 -0
- paddlex/repo_apis/base/register.py +135 -0
- paddlex/repo_apis/base/runner.py +390 -0
- paddlex/repo_apis/base/utils/__init__.py +13 -0
- paddlex/repo_apis/base/utils/arg.py +64 -0
- paddlex/repo_apis/base/utils/subprocess.py +107 -0
- paddlex/repo_manager/__init__.py +24 -0
- paddlex/repo_manager/core.py +271 -0
- paddlex/repo_manager/meta.py +143 -0
- paddlex/repo_manager/repo.py +396 -0
- paddlex/repo_manager/requirements.txt +18 -0
- paddlex/repo_manager/utils.py +298 -0
- paddlex/utils/__init__.py +1 -12
- paddlex/utils/cache.py +148 -0
- paddlex/utils/config.py +214 -0
- paddlex/utils/device.py +103 -0
- paddlex/utils/download.py +168 -182
- paddlex/utils/errors/__init__.py +17 -0
- paddlex/utils/errors/dataset_checker.py +78 -0
- paddlex/utils/errors/others.py +152 -0
- paddlex/utils/file_interface.py +212 -0
- paddlex/utils/flags.py +61 -0
- paddlex/utils/fonts/PingFang-SC-Regular.ttf +0 -0
- paddlex/utils/fonts/__init__.py +24 -0
- paddlex/utils/func_register.py +41 -0
- paddlex/utils/interactive_get_pipeline.py +55 -0
- paddlex/utils/lazy_loader.py +66 -0
- paddlex/utils/logging.py +132 -33
- paddlex/utils/misc.py +201 -0
- paddlex/utils/result_saver.py +59 -0
- paddlex/utils/subclass_register.py +101 -0
- paddlex/version.py +54 -0
- paddlex-3.0.0b2.dist-info/LICENSE +169 -0
- paddlex-3.0.0b2.dist-info/METADATA +760 -0
- paddlex-3.0.0b2.dist-info/RECORD +646 -0
- paddlex-3.0.0b2.dist-info/WHEEL +5 -0
- paddlex-3.0.0b2.dist-info/entry_points.txt +2 -0
- paddlex-3.0.0b2.dist-info/top_level.txt +1 -0
- PaddleClas/__init__.py +0 -16
- PaddleClas/deploy/__init__.py +0 -1
- PaddleClas/deploy/paddleserving/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/classification_web_service.py +0 -74
- PaddleClas/deploy/paddleserving/cpu_utilization.py +0 -4
- PaddleClas/deploy/paddleserving/pipeline_http_client.py +0 -20
- PaddleClas/deploy/paddleserving/pipeline_rpc_client.py +0 -33
- PaddleClas/deploy/paddleserving/recognition/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/recognition/pipeline_http_client.py +0 -21
- PaddleClas/deploy/paddleserving/recognition/pipeline_rpc_client.py +0 -34
- PaddleClas/deploy/paddleserving/recognition/recognition_web_service.py +0 -209
- PaddleClas/deploy/python/__init__.py +0 -0
- PaddleClas/deploy/python/build_gallery.py +0 -214
- PaddleClas/deploy/python/det_preprocess.py +0 -205
- PaddleClas/deploy/python/postprocess.py +0 -161
- PaddleClas/deploy/python/predict_cls.py +0 -142
- PaddleClas/deploy/python/predict_det.py +0 -158
- PaddleClas/deploy/python/predict_rec.py +0 -138
- PaddleClas/deploy/python/predict_system.py +0 -144
- PaddleClas/deploy/python/preprocess.py +0 -337
- PaddleClas/deploy/utils/__init__.py +0 -5
- PaddleClas/deploy/utils/config.py +0 -197
- PaddleClas/deploy/utils/draw_bbox.py +0 -61
- PaddleClas/deploy/utils/encode_decode.py +0 -31
- PaddleClas/deploy/utils/get_image_list.py +0 -49
- PaddleClas/deploy/utils/logger.py +0 -120
- PaddleClas/deploy/utils/predictor.py +0 -71
- PaddleClas/deploy/vector_search/__init__.py +0 -1
- PaddleClas/deploy/vector_search/interface.py +0 -272
- PaddleClas/deploy/vector_search/test.py +0 -34
- PaddleClas/hubconf.py +0 -788
- PaddleClas/paddleclas.py +0 -552
- PaddleClas/ppcls/__init__.py +0 -20
- PaddleClas/ppcls/arch/__init__.py +0 -127
- PaddleClas/ppcls/arch/backbone/__init__.py +0 -80
- PaddleClas/ppcls/arch/backbone/base/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/base/theseus_layer.py +0 -126
- PaddleClas/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- PaddleClas/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- PaddleClas/ppcls/arch/backbone/legendary_models/hrnet.py +0 -744
- PaddleClas/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- PaddleClas/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- PaddleClas/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- PaddleClas/ppcls/arch/backbone/legendary_models/vgg.py +0 -231
- PaddleClas/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- PaddleClas/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- PaddleClas/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- PaddleClas/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- PaddleClas/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- PaddleClas/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- PaddleClas/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- PaddleClas/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- PaddleClas/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- PaddleClas/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- PaddleClas/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- PaddleClas/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- PaddleClas/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- PaddleClas/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- PaddleClas/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- PaddleClas/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- PaddleClas/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- PaddleClas/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- PaddleClas/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- PaddleClas/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- PaddleClas/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- PaddleClas/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- PaddleClas/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- PaddleClas/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- PaddleClas/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- PaddleClas/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- PaddleClas/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- PaddleClas/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- PaddleClas/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- PaddleClas/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- PaddleClas/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- PaddleClas/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- PaddleClas/ppcls/arch/gears/__init__.py +0 -32
- PaddleClas/ppcls/arch/gears/arcmargin.py +0 -72
- PaddleClas/ppcls/arch/gears/circlemargin.py +0 -59
- PaddleClas/ppcls/arch/gears/cosmargin.py +0 -55
- PaddleClas/ppcls/arch/gears/fc.py +0 -35
- PaddleClas/ppcls/arch/gears/identity_head.py +0 -9
- PaddleClas/ppcls/arch/gears/vehicle_neck.py +0 -52
- PaddleClas/ppcls/arch/utils.py +0 -53
- PaddleClas/ppcls/data/__init__.py +0 -144
- PaddleClas/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- PaddleClas/ppcls/data/dataloader/__init__.py +0 -9
- PaddleClas/ppcls/data/dataloader/common_dataset.py +0 -84
- PaddleClas/ppcls/data/dataloader/dali.py +0 -319
- PaddleClas/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- PaddleClas/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- PaddleClas/ppcls/data/dataloader/logo_dataset.py +0 -46
- PaddleClas/ppcls/data/dataloader/mix_dataset.py +0 -49
- PaddleClas/ppcls/data/dataloader/mix_sampler.py +0 -79
- PaddleClas/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- PaddleClas/ppcls/data/dataloader/pk_sampler.py +0 -105
- PaddleClas/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- PaddleClas/ppcls/data/postprocess/__init__.py +0 -41
- PaddleClas/ppcls/data/postprocess/topk.py +0 -85
- PaddleClas/ppcls/data/preprocess/__init__.py +0 -100
- PaddleClas/ppcls/data/preprocess/batch_ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- PaddleClas/ppcls/data/preprocess/ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- PaddleClas/ppcls/data/preprocess/ops/cutout.py +0 -41
- PaddleClas/ppcls/data/preprocess/ops/fmix.py +0 -217
- PaddleClas/ppcls/data/preprocess/ops/functional.py +0 -138
- PaddleClas/ppcls/data/preprocess/ops/grid.py +0 -89
- PaddleClas/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- PaddleClas/ppcls/data/preprocess/ops/operators.py +0 -384
- PaddleClas/ppcls/data/preprocess/ops/randaugment.py +0 -106
- PaddleClas/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- PaddleClas/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- PaddleClas/ppcls/data/utils/__init__.py +0 -13
- PaddleClas/ppcls/data/utils/get_image_list.py +0 -49
- PaddleClas/ppcls/engine/__init__.py +0 -0
- PaddleClas/ppcls/engine/engine.py +0 -436
- PaddleClas/ppcls/engine/evaluation/__init__.py +0 -16
- PaddleClas/ppcls/engine/evaluation/classification.py +0 -143
- PaddleClas/ppcls/engine/evaluation/retrieval.py +0 -169
- PaddleClas/ppcls/engine/slim/__init__.py +0 -16
- PaddleClas/ppcls/engine/slim/prune.py +0 -66
- PaddleClas/ppcls/engine/slim/quant.py +0 -55
- PaddleClas/ppcls/engine/train/__init__.py +0 -14
- PaddleClas/ppcls/engine/train/train.py +0 -79
- PaddleClas/ppcls/engine/train/utils.py +0 -72
- PaddleClas/ppcls/loss/__init__.py +0 -65
- PaddleClas/ppcls/loss/celoss.py +0 -67
- PaddleClas/ppcls/loss/centerloss.py +0 -54
- PaddleClas/ppcls/loss/comfunc.py +0 -45
- PaddleClas/ppcls/loss/deephashloss.py +0 -92
- PaddleClas/ppcls/loss/distanceloss.py +0 -43
- PaddleClas/ppcls/loss/distillationloss.py +0 -141
- PaddleClas/ppcls/loss/dmlloss.py +0 -46
- PaddleClas/ppcls/loss/emlloss.py +0 -97
- PaddleClas/ppcls/loss/googlenetloss.py +0 -41
- PaddleClas/ppcls/loss/msmloss.py +0 -78
- PaddleClas/ppcls/loss/multilabelloss.py +0 -43
- PaddleClas/ppcls/loss/npairsloss.py +0 -38
- PaddleClas/ppcls/loss/pairwisecosface.py +0 -55
- PaddleClas/ppcls/loss/supconloss.py +0 -108
- PaddleClas/ppcls/loss/trihardloss.py +0 -82
- PaddleClas/ppcls/loss/triplet.py +0 -137
- PaddleClas/ppcls/metric/__init__.py +0 -51
- PaddleClas/ppcls/metric/metrics.py +0 -308
- PaddleClas/ppcls/optimizer/__init__.py +0 -72
- PaddleClas/ppcls/optimizer/learning_rate.py +0 -326
- PaddleClas/ppcls/optimizer/optimizer.py +0 -207
- PaddleClas/ppcls/utils/__init__.py +0 -27
- PaddleClas/ppcls/utils/check.py +0 -151
- PaddleClas/ppcls/utils/config.py +0 -210
- PaddleClas/ppcls/utils/download.py +0 -319
- PaddleClas/ppcls/utils/ema.py +0 -63
- PaddleClas/ppcls/utils/logger.py +0 -137
- PaddleClas/ppcls/utils/metrics.py +0 -107
- PaddleClas/ppcls/utils/misc.py +0 -63
- PaddleClas/ppcls/utils/model_zoo.py +0 -213
- PaddleClas/ppcls/utils/profiler.py +0 -111
- PaddleClas/ppcls/utils/save_load.py +0 -136
- PaddleClas/setup.py +0 -58
- PaddleClas/tools/__init__.py +0 -15
- PaddleClas/tools/eval.py +0 -31
- PaddleClas/tools/export_model.py +0 -34
- PaddleClas/tools/infer.py +0 -31
- PaddleClas/tools/train.py +0 -32
- paddlex/cls.py +0 -82
- paddlex/command.py +0 -215
- paddlex/cv/__init__.py +0 -17
- paddlex/cv/datasets/__init__.py +0 -18
- paddlex/cv/datasets/coco.py +0 -208
- paddlex/cv/datasets/imagenet.py +0 -88
- paddlex/cv/datasets/seg_dataset.py +0 -91
- paddlex/cv/datasets/voc.py +0 -445
- paddlex/cv/models/__init__.py +0 -18
- paddlex/cv/models/base.py +0 -631
- paddlex/cv/models/classifier.py +0 -989
- paddlex/cv/models/detector.py +0 -2292
- paddlex/cv/models/load_model.py +0 -148
- paddlex/cv/models/segmenter.py +0 -768
- paddlex/cv/models/slim/__init__.py +0 -13
- paddlex/cv/models/slim/prune.py +0 -55
- paddlex/cv/models/utils/__init__.py +0 -13
- paddlex/cv/models/utils/det_metrics/__init__.py +0 -15
- paddlex/cv/models/utils/det_metrics/coco_utils.py +0 -476
- paddlex/cv/models/utils/det_metrics/metrics.py +0 -220
- paddlex/cv/models/utils/infer_nets.py +0 -45
- paddlex/cv/models/utils/seg_metrics.py +0 -62
- paddlex/cv/models/utils/visualize.py +0 -399
- paddlex/cv/transforms/__init__.py +0 -46
- paddlex/cv/transforms/batch_operators.py +0 -286
- paddlex/cv/transforms/box_utils.py +0 -41
- paddlex/cv/transforms/functions.py +0 -193
- paddlex/cv/transforms/operators.py +0 -1402
- paddlex/deploy.py +0 -268
- paddlex/det.py +0 -49
- paddlex/paddleseg/__init__.py +0 -17
- paddlex/paddleseg/core/__init__.py +0 -20
- paddlex/paddleseg/core/infer.py +0 -289
- paddlex/paddleseg/core/predict.py +0 -145
- paddlex/paddleseg/core/train.py +0 -258
- paddlex/paddleseg/core/val.py +0 -172
- paddlex/paddleseg/cvlibs/__init__.py +0 -17
- paddlex/paddleseg/cvlibs/callbacks.py +0 -279
- paddlex/paddleseg/cvlibs/config.py +0 -359
- paddlex/paddleseg/cvlibs/manager.py +0 -142
- paddlex/paddleseg/cvlibs/param_init.py +0 -91
- paddlex/paddleseg/datasets/__init__.py +0 -21
- paddlex/paddleseg/datasets/ade.py +0 -112
- paddlex/paddleseg/datasets/cityscapes.py +0 -86
- paddlex/paddleseg/datasets/cocostuff.py +0 -79
- paddlex/paddleseg/datasets/dataset.py +0 -164
- paddlex/paddleseg/datasets/mini_deep_globe_road_extraction.py +0 -95
- paddlex/paddleseg/datasets/optic_disc_seg.py +0 -97
- paddlex/paddleseg/datasets/pascal_context.py +0 -80
- paddlex/paddleseg/datasets/voc.py +0 -113
- paddlex/paddleseg/models/__init__.py +0 -39
- paddlex/paddleseg/models/ann.py +0 -436
- paddlex/paddleseg/models/attention_unet.py +0 -189
- paddlex/paddleseg/models/backbones/__init__.py +0 -18
- paddlex/paddleseg/models/backbones/hrnet.py +0 -815
- paddlex/paddleseg/models/backbones/mobilenetv3.py +0 -365
- paddlex/paddleseg/models/backbones/resnet_vd.py +0 -364
- paddlex/paddleseg/models/backbones/xception_deeplab.py +0 -415
- paddlex/paddleseg/models/bisenet.py +0 -311
- paddlex/paddleseg/models/danet.py +0 -220
- paddlex/paddleseg/models/decoupled_segnet.py +0 -233
- paddlex/paddleseg/models/deeplab.py +0 -258
- paddlex/paddleseg/models/dnlnet.py +0 -231
- paddlex/paddleseg/models/emanet.py +0 -219
- paddlex/paddleseg/models/fast_scnn.py +0 -318
- paddlex/paddleseg/models/fcn.py +0 -135
- paddlex/paddleseg/models/gcnet.py +0 -223
- paddlex/paddleseg/models/gscnn.py +0 -357
- paddlex/paddleseg/models/hardnet.py +0 -309
- paddlex/paddleseg/models/isanet.py +0 -202
- paddlex/paddleseg/models/layers/__init__.py +0 -19
- paddlex/paddleseg/models/layers/activation.py +0 -73
- paddlex/paddleseg/models/layers/attention.py +0 -146
- paddlex/paddleseg/models/layers/layer_libs.py +0 -168
- paddlex/paddleseg/models/layers/nonlocal2d.py +0 -155
- paddlex/paddleseg/models/layers/pyramid_pool.py +0 -182
- paddlex/paddleseg/models/losses/__init__.py +0 -27
- paddlex/paddleseg/models/losses/binary_cross_entropy_loss.py +0 -174
- paddlex/paddleseg/models/losses/bootstrapped_cross_entropy.py +0 -73
- paddlex/paddleseg/models/losses/cross_entropy_loss.py +0 -94
- paddlex/paddleseg/models/losses/decoupledsegnet_relax_boundary_loss.py +0 -129
- paddlex/paddleseg/models/losses/dice_loss.py +0 -61
- paddlex/paddleseg/models/losses/edge_attention_loss.py +0 -78
- paddlex/paddleseg/models/losses/gscnn_dual_task_loss.py +0 -141
- paddlex/paddleseg/models/losses/l1_loss.py +0 -76
- paddlex/paddleseg/models/losses/lovasz_loss.py +0 -222
- paddlex/paddleseg/models/losses/mean_square_error_loss.py +0 -65
- paddlex/paddleseg/models/losses/mixed_loss.py +0 -58
- paddlex/paddleseg/models/losses/ohem_cross_entropy_loss.py +0 -99
- paddlex/paddleseg/models/losses/ohem_edge_attention_loss.py +0 -114
- paddlex/paddleseg/models/ocrnet.py +0 -248
- paddlex/paddleseg/models/pspnet.py +0 -147
- paddlex/paddleseg/models/sfnet.py +0 -236
- paddlex/paddleseg/models/shufflenet_slim.py +0 -268
- paddlex/paddleseg/models/u2net.py +0 -574
- paddlex/paddleseg/models/unet.py +0 -155
- paddlex/paddleseg/models/unet_3plus.py +0 -316
- paddlex/paddleseg/models/unet_plusplus.py +0 -237
- paddlex/paddleseg/transforms/__init__.py +0 -16
- paddlex/paddleseg/transforms/functional.py +0 -161
- paddlex/paddleseg/transforms/transforms.py +0 -937
- paddlex/paddleseg/utils/__init__.py +0 -22
- paddlex/paddleseg/utils/config_check.py +0 -60
- paddlex/paddleseg/utils/download.py +0 -163
- paddlex/paddleseg/utils/env/__init__.py +0 -16
- paddlex/paddleseg/utils/env/seg_env.py +0 -56
- paddlex/paddleseg/utils/env/sys_env.py +0 -122
- paddlex/paddleseg/utils/logger.py +0 -48
- paddlex/paddleseg/utils/metrics.py +0 -146
- paddlex/paddleseg/utils/progbar.py +0 -212
- paddlex/paddleseg/utils/timer.py +0 -53
- paddlex/paddleseg/utils/utils.py +0 -120
- paddlex/paddleseg/utils/visualize.py +0 -90
- paddlex/ppcls/__init__.py +0 -20
- paddlex/ppcls/arch/__init__.py +0 -127
- paddlex/ppcls/arch/backbone/__init__.py +0 -80
- paddlex/ppcls/arch/backbone/base/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/base/theseus_layer.py +0 -130
- paddlex/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- paddlex/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- paddlex/ppcls/arch/backbone/legendary_models/hrnet.py +0 -748
- paddlex/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- paddlex/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- paddlex/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- paddlex/ppcls/arch/backbone/legendary_models/vgg.py +0 -235
- paddlex/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- paddlex/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- paddlex/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- paddlex/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- paddlex/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- paddlex/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- paddlex/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- paddlex/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- paddlex/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- paddlex/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- paddlex/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- paddlex/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- paddlex/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- paddlex/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- paddlex/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- paddlex/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- paddlex/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- paddlex/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- paddlex/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- paddlex/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- paddlex/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- paddlex/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- paddlex/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- paddlex/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- paddlex/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- paddlex/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- paddlex/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- paddlex/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- paddlex/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- paddlex/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- paddlex/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- paddlex/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- paddlex/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- paddlex/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- paddlex/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- paddlex/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- paddlex/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- paddlex/ppcls/arch/gears/__init__.py +0 -32
- paddlex/ppcls/arch/gears/arcmargin.py +0 -72
- paddlex/ppcls/arch/gears/circlemargin.py +0 -59
- paddlex/ppcls/arch/gears/cosmargin.py +0 -55
- paddlex/ppcls/arch/gears/fc.py +0 -35
- paddlex/ppcls/arch/gears/identity_head.py +0 -9
- paddlex/ppcls/arch/gears/vehicle_neck.py +0 -52
- paddlex/ppcls/arch/utils.py +0 -53
- paddlex/ppcls/data/__init__.py +0 -144
- paddlex/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- paddlex/ppcls/data/dataloader/__init__.py +0 -9
- paddlex/ppcls/data/dataloader/common_dataset.py +0 -84
- paddlex/ppcls/data/dataloader/dali.py +0 -319
- paddlex/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- paddlex/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- paddlex/ppcls/data/dataloader/logo_dataset.py +0 -46
- paddlex/ppcls/data/dataloader/mix_dataset.py +0 -49
- paddlex/ppcls/data/dataloader/mix_sampler.py +0 -79
- paddlex/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- paddlex/ppcls/data/dataloader/pk_sampler.py +0 -105
- paddlex/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- paddlex/ppcls/data/postprocess/__init__.py +0 -41
- paddlex/ppcls/data/postprocess/topk.py +0 -85
- paddlex/ppcls/data/preprocess/__init__.py +0 -100
- paddlex/ppcls/data/preprocess/batch_ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- paddlex/ppcls/data/preprocess/ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- paddlex/ppcls/data/preprocess/ops/cutout.py +0 -41
- paddlex/ppcls/data/preprocess/ops/fmix.py +0 -217
- paddlex/ppcls/data/preprocess/ops/functional.py +0 -141
- paddlex/ppcls/data/preprocess/ops/grid.py +0 -89
- paddlex/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- paddlex/ppcls/data/preprocess/ops/operators.py +0 -384
- paddlex/ppcls/data/preprocess/ops/randaugment.py +0 -106
- paddlex/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- paddlex/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- paddlex/ppcls/data/utils/__init__.py +0 -13
- paddlex/ppcls/data/utils/get_image_list.py +0 -49
- paddlex/ppcls/engine/__init__.py +0 -0
- paddlex/ppcls/engine/engine.py +0 -436
- paddlex/ppcls/engine/evaluation/__init__.py +0 -16
- paddlex/ppcls/engine/evaluation/classification.py +0 -143
- paddlex/ppcls/engine/evaluation/retrieval.py +0 -169
- paddlex/ppcls/engine/slim/__init__.py +0 -16
- paddlex/ppcls/engine/slim/prune.py +0 -66
- paddlex/ppcls/engine/slim/quant.py +0 -55
- paddlex/ppcls/engine/train/__init__.py +0 -14
- paddlex/ppcls/engine/train/train.py +0 -79
- paddlex/ppcls/engine/train/utils.py +0 -72
- paddlex/ppcls/loss/__init__.py +0 -65
- paddlex/ppcls/loss/celoss.py +0 -67
- paddlex/ppcls/loss/centerloss.py +0 -54
- paddlex/ppcls/loss/comfunc.py +0 -45
- paddlex/ppcls/loss/deephashloss.py +0 -96
- paddlex/ppcls/loss/distanceloss.py +0 -43
- paddlex/ppcls/loss/distillationloss.py +0 -141
- paddlex/ppcls/loss/dmlloss.py +0 -46
- paddlex/ppcls/loss/emlloss.py +0 -97
- paddlex/ppcls/loss/googlenetloss.py +0 -42
- paddlex/ppcls/loss/msmloss.py +0 -78
- paddlex/ppcls/loss/multilabelloss.py +0 -43
- paddlex/ppcls/loss/npairsloss.py +0 -38
- paddlex/ppcls/loss/pairwisecosface.py +0 -59
- paddlex/ppcls/loss/supconloss.py +0 -108
- paddlex/ppcls/loss/trihardloss.py +0 -82
- paddlex/ppcls/loss/triplet.py +0 -137
- paddlex/ppcls/metric/__init__.py +0 -51
- paddlex/ppcls/metric/metrics.py +0 -308
- paddlex/ppcls/optimizer/__init__.py +0 -72
- paddlex/ppcls/optimizer/learning_rate.py +0 -326
- paddlex/ppcls/optimizer/optimizer.py +0 -208
- paddlex/ppcls/utils/__init__.py +0 -27
- paddlex/ppcls/utils/check.py +0 -151
- paddlex/ppcls/utils/config.py +0 -210
- paddlex/ppcls/utils/download.py +0 -319
- paddlex/ppcls/utils/ema.py +0 -63
- paddlex/ppcls/utils/logger.py +0 -137
- paddlex/ppcls/utils/metrics.py +0 -112
- paddlex/ppcls/utils/misc.py +0 -63
- paddlex/ppcls/utils/model_zoo.py +0 -213
- paddlex/ppcls/utils/profiler.py +0 -111
- paddlex/ppcls/utils/save_load.py +0 -136
- paddlex/ppdet/__init__.py +0 -16
- paddlex/ppdet/core/__init__.py +0 -15
- paddlex/ppdet/core/config/__init__.py +0 -13
- paddlex/ppdet/core/config/schema.py +0 -248
- paddlex/ppdet/core/config/yaml_helpers.py +0 -118
- paddlex/ppdet/core/workspace.py +0 -278
- paddlex/ppdet/data/__init__.py +0 -21
- paddlex/ppdet/data/crop_utils/__init__.py +0 -13
- paddlex/ppdet/data/crop_utils/annotation_cropper.py +0 -585
- paddlex/ppdet/data/crop_utils/chip_box_utils.py +0 -170
- paddlex/ppdet/data/reader.py +0 -302
- paddlex/ppdet/data/shm_utils.py +0 -67
- paddlex/ppdet/data/source/__init__.py +0 -29
- paddlex/ppdet/data/source/category.py +0 -904
- paddlex/ppdet/data/source/coco.py +0 -251
- paddlex/ppdet/data/source/dataset.py +0 -197
- paddlex/ppdet/data/source/keypoint_coco.py +0 -669
- paddlex/ppdet/data/source/mot.py +0 -636
- paddlex/ppdet/data/source/sniper_coco.py +0 -191
- paddlex/ppdet/data/source/voc.py +0 -231
- paddlex/ppdet/data/source/widerface.py +0 -180
- paddlex/ppdet/data/transform/__init__.py +0 -28
- paddlex/ppdet/data/transform/atss_assigner.py +0 -270
- paddlex/ppdet/data/transform/autoaugment_utils.py +0 -1591
- paddlex/ppdet/data/transform/batch_operators.py +0 -1080
- paddlex/ppdet/data/transform/gridmask_utils.py +0 -86
- paddlex/ppdet/data/transform/keypoint_operators.py +0 -868
- paddlex/ppdet/data/transform/mot_operators.py +0 -628
- paddlex/ppdet/data/transform/op_helper.py +0 -498
- paddlex/ppdet/data/transform/operators.py +0 -3025
- paddlex/ppdet/engine/__init__.py +0 -30
- paddlex/ppdet/engine/callbacks.py +0 -340
- paddlex/ppdet/engine/env.py +0 -50
- paddlex/ppdet/engine/export_utils.py +0 -177
- paddlex/ppdet/engine/tracker.py +0 -538
- paddlex/ppdet/engine/trainer.py +0 -723
- paddlex/ppdet/metrics/__init__.py +0 -29
- paddlex/ppdet/metrics/coco_utils.py +0 -184
- paddlex/ppdet/metrics/json_results.py +0 -149
- paddlex/ppdet/metrics/keypoint_metrics.py +0 -401
- paddlex/ppdet/metrics/map_utils.py +0 -444
- paddlex/ppdet/metrics/mcmot_metrics.py +0 -470
- paddlex/ppdet/metrics/metrics.py +0 -434
- paddlex/ppdet/metrics/mot_metrics.py +0 -1236
- paddlex/ppdet/metrics/munkres.py +0 -428
- paddlex/ppdet/metrics/widerface_utils.py +0 -393
- paddlex/ppdet/model_zoo/__init__.py +0 -18
- paddlex/ppdet/model_zoo/model_zoo.py +0 -84
- paddlex/ppdet/modeling/__init__.py +0 -45
- paddlex/ppdet/modeling/architectures/__init__.py +0 -51
- paddlex/ppdet/modeling/architectures/blazeface.py +0 -91
- paddlex/ppdet/modeling/architectures/cascade_rcnn.py +0 -144
- paddlex/ppdet/modeling/architectures/centernet.py +0 -108
- paddlex/ppdet/modeling/architectures/deepsort.py +0 -69
- paddlex/ppdet/modeling/architectures/detr.py +0 -93
- paddlex/ppdet/modeling/architectures/fairmot.py +0 -100
- paddlex/ppdet/modeling/architectures/faster_rcnn.py +0 -106
- paddlex/ppdet/modeling/architectures/fcos.py +0 -105
- paddlex/ppdet/modeling/architectures/gfl.py +0 -87
- paddlex/ppdet/modeling/architectures/jde.py +0 -111
- paddlex/ppdet/modeling/architectures/keypoint_hrhrnet.py +0 -287
- paddlex/ppdet/modeling/architectures/keypoint_hrnet.py +0 -267
- paddlex/ppdet/modeling/architectures/mask_rcnn.py +0 -135
- paddlex/ppdet/modeling/architectures/meta_arch.py +0 -128
- paddlex/ppdet/modeling/architectures/picodet.py +0 -91
- paddlex/ppdet/modeling/architectures/s2anet.py +0 -102
- paddlex/ppdet/modeling/architectures/solov2.py +0 -110
- paddlex/ppdet/modeling/architectures/sparse_rcnn.py +0 -99
- paddlex/ppdet/modeling/architectures/ssd.py +0 -93
- paddlex/ppdet/modeling/architectures/tood.py +0 -78
- paddlex/ppdet/modeling/architectures/ttfnet.py +0 -98
- paddlex/ppdet/modeling/architectures/yolo.py +0 -124
- paddlex/ppdet/modeling/assigners/__init__.py +0 -23
- paddlex/ppdet/modeling/assigners/atss_assigner.py +0 -211
- paddlex/ppdet/modeling/assigners/simota_assigner.py +0 -262
- paddlex/ppdet/modeling/assigners/task_aligned_assigner.py +0 -158
- paddlex/ppdet/modeling/assigners/utils.py +0 -195
- paddlex/ppdet/modeling/backbones/__init__.py +0 -49
- paddlex/ppdet/modeling/backbones/blazenet.py +0 -323
- paddlex/ppdet/modeling/backbones/darknet.py +0 -340
- paddlex/ppdet/modeling/backbones/dla.py +0 -244
- paddlex/ppdet/modeling/backbones/esnet.py +0 -290
- paddlex/ppdet/modeling/backbones/ghostnet.py +0 -470
- paddlex/ppdet/modeling/backbones/hardnet.py +0 -224
- paddlex/ppdet/modeling/backbones/hrnet.py +0 -727
- paddlex/ppdet/modeling/backbones/lcnet.py +0 -259
- paddlex/ppdet/modeling/backbones/lite_hrnet.py +0 -886
- paddlex/ppdet/modeling/backbones/mobilenet_v1.py +0 -418
- paddlex/ppdet/modeling/backbones/mobilenet_v3.py +0 -483
- paddlex/ppdet/modeling/backbones/name_adapter.py +0 -69
- paddlex/ppdet/modeling/backbones/res2net.py +0 -358
- paddlex/ppdet/modeling/backbones/resnet.py +0 -613
- paddlex/ppdet/modeling/backbones/senet.py +0 -139
- paddlex/ppdet/modeling/backbones/shufflenet_v2.py +0 -246
- paddlex/ppdet/modeling/backbones/swin_transformer.py +0 -743
- paddlex/ppdet/modeling/backbones/vgg.py +0 -210
- paddlex/ppdet/modeling/bbox_utils.py +0 -778
- paddlex/ppdet/modeling/heads/__init__.py +0 -53
- paddlex/ppdet/modeling/heads/bbox_head.py +0 -377
- paddlex/ppdet/modeling/heads/cascade_head.py +0 -284
- paddlex/ppdet/modeling/heads/centernet_head.py +0 -292
- paddlex/ppdet/modeling/heads/detr_head.py +0 -368
- paddlex/ppdet/modeling/heads/face_head.py +0 -110
- paddlex/ppdet/modeling/heads/fcos_head.py +0 -259
- paddlex/ppdet/modeling/heads/gfl_head.py +0 -487
- paddlex/ppdet/modeling/heads/keypoint_hrhrnet_head.py +0 -108
- paddlex/ppdet/modeling/heads/mask_head.py +0 -250
- paddlex/ppdet/modeling/heads/pico_head.py +0 -278
- paddlex/ppdet/modeling/heads/roi_extractor.py +0 -111
- paddlex/ppdet/modeling/heads/s2anet_head.py +0 -1056
- paddlex/ppdet/modeling/heads/simota_head.py +0 -506
- paddlex/ppdet/modeling/heads/solov2_head.py +0 -560
- paddlex/ppdet/modeling/heads/sparsercnn_head.py +0 -375
- paddlex/ppdet/modeling/heads/ssd_head.py +0 -215
- paddlex/ppdet/modeling/heads/tood_head.py +0 -366
- paddlex/ppdet/modeling/heads/ttf_head.py +0 -316
- paddlex/ppdet/modeling/heads/yolo_head.py +0 -124
- paddlex/ppdet/modeling/initializer.py +0 -317
- paddlex/ppdet/modeling/keypoint_utils.py +0 -342
- paddlex/ppdet/modeling/layers.py +0 -1430
- paddlex/ppdet/modeling/losses/__init__.py +0 -43
- paddlex/ppdet/modeling/losses/ctfocal_loss.py +0 -68
- paddlex/ppdet/modeling/losses/detr_loss.py +0 -233
- paddlex/ppdet/modeling/losses/fairmot_loss.py +0 -41
- paddlex/ppdet/modeling/losses/fcos_loss.py +0 -225
- paddlex/ppdet/modeling/losses/gfocal_loss.py +0 -217
- paddlex/ppdet/modeling/losses/iou_aware_loss.py +0 -47
- paddlex/ppdet/modeling/losses/iou_loss.py +0 -210
- paddlex/ppdet/modeling/losses/jde_loss.py +0 -193
- paddlex/ppdet/modeling/losses/keypoint_loss.py +0 -229
- paddlex/ppdet/modeling/losses/solov2_loss.py +0 -101
- paddlex/ppdet/modeling/losses/sparsercnn_loss.py +0 -425
- paddlex/ppdet/modeling/losses/ssd_loss.py +0 -170
- paddlex/ppdet/modeling/losses/varifocal_loss.py +0 -152
- paddlex/ppdet/modeling/losses/yolo_loss.py +0 -212
- paddlex/ppdet/modeling/mot/__init__.py +0 -25
- paddlex/ppdet/modeling/mot/matching/__init__.py +0 -19
- paddlex/ppdet/modeling/mot/matching/deepsort_matching.py +0 -382
- paddlex/ppdet/modeling/mot/matching/jde_matching.py +0 -144
- paddlex/ppdet/modeling/mot/motion/__init__.py +0 -17
- paddlex/ppdet/modeling/mot/motion/kalman_filter.py +0 -270
- paddlex/ppdet/modeling/mot/tracker/__init__.py +0 -23
- paddlex/ppdet/modeling/mot/tracker/base_jde_tracker.py +0 -297
- paddlex/ppdet/modeling/mot/tracker/base_sde_tracker.py +0 -156
- paddlex/ppdet/modeling/mot/tracker/deepsort_tracker.py +0 -188
- paddlex/ppdet/modeling/mot/tracker/jde_tracker.py +0 -277
- paddlex/ppdet/modeling/mot/utils.py +0 -263
- paddlex/ppdet/modeling/mot/visualization.py +0 -150
- paddlex/ppdet/modeling/necks/__init__.py +0 -30
- paddlex/ppdet/modeling/necks/bifpn.py +0 -302
- paddlex/ppdet/modeling/necks/blazeface_fpn.py +0 -216
- paddlex/ppdet/modeling/necks/centernet_fpn.py +0 -426
- paddlex/ppdet/modeling/necks/csp_pan.py +0 -364
- paddlex/ppdet/modeling/necks/fpn.py +0 -231
- paddlex/ppdet/modeling/necks/hrfpn.py +0 -126
- paddlex/ppdet/modeling/necks/ttf_fpn.py +0 -242
- paddlex/ppdet/modeling/necks/yolo_fpn.py +0 -988
- paddlex/ppdet/modeling/ops.py +0 -1611
- paddlex/ppdet/modeling/post_process.py +0 -731
- paddlex/ppdet/modeling/proposal_generator/__init__.py +0 -2
- paddlex/ppdet/modeling/proposal_generator/anchor_generator.py +0 -135
- paddlex/ppdet/modeling/proposal_generator/proposal_generator.py +0 -77
- paddlex/ppdet/modeling/proposal_generator/rpn_head.py +0 -260
- paddlex/ppdet/modeling/proposal_generator/target.py +0 -681
- paddlex/ppdet/modeling/proposal_generator/target_layer.py +0 -491
- paddlex/ppdet/modeling/reid/__init__.py +0 -25
- paddlex/ppdet/modeling/reid/fairmot_embedding_head.py +0 -225
- paddlex/ppdet/modeling/reid/jde_embedding_head.py +0 -214
- paddlex/ppdet/modeling/reid/pplcnet_embedding.py +0 -282
- paddlex/ppdet/modeling/reid/pyramidal_embedding.py +0 -144
- paddlex/ppdet/modeling/reid/resnet.py +0 -310
- paddlex/ppdet/modeling/shape_spec.py +0 -25
- paddlex/ppdet/modeling/transformers/__init__.py +0 -25
- paddlex/ppdet/modeling/transformers/deformable_transformer.py +0 -517
- paddlex/ppdet/modeling/transformers/detr_transformer.py +0 -353
- paddlex/ppdet/modeling/transformers/matchers.py +0 -127
- paddlex/ppdet/modeling/transformers/position_encoding.py +0 -108
- paddlex/ppdet/modeling/transformers/utils.py +0 -110
- paddlex/ppdet/optimizer.py +0 -335
- paddlex/ppdet/slim/__init__.py +0 -82
- paddlex/ppdet/slim/distill.py +0 -110
- paddlex/ppdet/slim/prune.py +0 -85
- paddlex/ppdet/slim/quant.py +0 -84
- paddlex/ppdet/slim/unstructured_prune.py +0 -66
- paddlex/ppdet/utils/__init__.py +0 -13
- paddlex/ppdet/utils/check.py +0 -112
- paddlex/ppdet/utils/checkpoint.py +0 -226
- paddlex/ppdet/utils/cli.py +0 -151
- paddlex/ppdet/utils/colormap.py +0 -58
- paddlex/ppdet/utils/download.py +0 -558
- paddlex/ppdet/utils/logger.py +0 -70
- paddlex/ppdet/utils/profiler.py +0 -111
- paddlex/ppdet/utils/stats.py +0 -94
- paddlex/ppdet/utils/visualizer.py +0 -321
- paddlex/ppdet/utils/voc_utils.py +0 -86
- paddlex/seg.py +0 -41
- paddlex/tools/__init__.py +0 -17
- paddlex/tools/anchor_clustering/__init__.py +0 -15
- paddlex/tools/anchor_clustering/yolo_cluster.py +0 -178
- paddlex/tools/convert.py +0 -52
- paddlex/tools/dataset_conversion/__init__.py +0 -24
- paddlex/tools/dataset_conversion/x2coco.py +0 -379
- paddlex/tools/dataset_conversion/x2imagenet.py +0 -82
- paddlex/tools/dataset_conversion/x2seg.py +0 -343
- paddlex/tools/dataset_conversion/x2voc.py +0 -230
- paddlex/tools/dataset_split/__init__.py +0 -23
- paddlex/tools/dataset_split/coco_split.py +0 -69
- paddlex/tools/dataset_split/imagenet_split.py +0 -75
- paddlex/tools/dataset_split/seg_split.py +0 -96
- paddlex/tools/dataset_split/utils.py +0 -75
- paddlex/tools/dataset_split/voc_split.py +0 -91
- paddlex/tools/split.py +0 -41
- paddlex/utils/checkpoint.py +0 -492
- paddlex/utils/env.py +0 -67
- paddlex/utils/shm.py +0 -67
- paddlex/utils/stats.py +0 -68
- paddlex/utils/utils.py +0 -229
- paddlex-2.1.0.data/data/paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex-2.1.0.dist-info/LICENSE +0 -201
- paddlex-2.1.0.dist-info/METADATA +0 -32
- paddlex-2.1.0.dist-info/RECORD +0 -698
- paddlex-2.1.0.dist-info/WHEEL +0 -5
- paddlex-2.1.0.dist-info/entry_points.txt +0 -4
- paddlex-2.1.0.dist-info/top_level.txt +0 -3
- paddlex_restful/__init__.py +0 -15
- paddlex_restful/command.py +0 -63
- paddlex_restful/restful/__init__.py +0 -15
- paddlex_restful/restful/app.py +0 -969
- paddlex_restful/restful/dataset/__init__.py +0 -13
- paddlex_restful/restful/dataset/cls_dataset.py +0 -159
- paddlex_restful/restful/dataset/dataset.py +0 -266
- paddlex_restful/restful/dataset/datasetbase.py +0 -86
- paddlex_restful/restful/dataset/det_dataset.py +0 -190
- paddlex_restful/restful/dataset/ins_seg_dataset.py +0 -312
- paddlex_restful/restful/dataset/operate.py +0 -155
- paddlex_restful/restful/dataset/seg_dataset.py +0 -222
- paddlex_restful/restful/dataset/utils.py +0 -267
- paddlex_restful/restful/demo.py +0 -202
- paddlex_restful/restful/dir.py +0 -45
- paddlex_restful/restful/model.py +0 -312
- paddlex_restful/restful/project/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/classification.py +0 -126
- paddlex_restful/restful/project/evaluate/detection.py +0 -789
- paddlex_restful/restful/project/evaluate/draw_pred_result.py +0 -181
- paddlex_restful/restful/project/evaluate/segmentation.py +0 -122
- paddlex_restful/restful/project/operate.py +0 -931
- paddlex_restful/restful/project/project.py +0 -143
- paddlex_restful/restful/project/prune/__init__.py +0 -13
- paddlex_restful/restful/project/prune/classification.py +0 -32
- paddlex_restful/restful/project/prune/detection.py +0 -48
- paddlex_restful/restful/project/prune/segmentation.py +0 -34
- paddlex_restful/restful/project/task.py +0 -884
- paddlex_restful/restful/project/train/__init__.py +0 -13
- paddlex_restful/restful/project/train/classification.py +0 -141
- paddlex_restful/restful/project/train/detection.py +0 -263
- paddlex_restful/restful/project/train/params.py +0 -432
- paddlex_restful/restful/project/train/params_v2.py +0 -326
- paddlex_restful/restful/project/train/segmentation.py +0 -191
- paddlex_restful/restful/project/visualize.py +0 -244
- paddlex_restful/restful/system.py +0 -102
- paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex_restful/restful/utils.py +0 -841
- paddlex_restful/restful/workspace.py +0 -343
- paddlex_restful/restful/workspace_pb2.py +0 -1411
@@ -1,353 +0,0 @@
|
|
1
|
-
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
#
|
15
|
-
# Modified from DETR (https://github.com/facebookresearch/detr)
|
16
|
-
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
17
|
-
|
18
|
-
from __future__ import absolute_import
|
19
|
-
from __future__ import division
|
20
|
-
from __future__ import print_function
|
21
|
-
|
22
|
-
import paddle
|
23
|
-
import paddle.nn as nn
|
24
|
-
import paddle.nn.functional as F
|
25
|
-
|
26
|
-
from paddlex.ppdet.core.workspace import register
|
27
|
-
from ..layers import MultiHeadAttention, _convert_attention_mask
|
28
|
-
from .position_encoding import PositionEmbedding
|
29
|
-
from .utils import _get_clones
|
30
|
-
from ..initializer import linear_init_, conv_init_, xavier_uniform_, normal_
|
31
|
-
|
32
|
-
__all__ = ['DETRTransformer']
|
33
|
-
|
34
|
-
|
35
|
-
class TransformerEncoderLayer(nn.Layer):
|
36
|
-
def __init__(self,
|
37
|
-
d_model,
|
38
|
-
nhead,
|
39
|
-
dim_feedforward=2048,
|
40
|
-
dropout=0.1,
|
41
|
-
activation="relu",
|
42
|
-
attn_dropout=None,
|
43
|
-
act_dropout=None,
|
44
|
-
normalize_before=False):
|
45
|
-
super(TransformerEncoderLayer, self).__init__()
|
46
|
-
attn_dropout = dropout if attn_dropout is None else attn_dropout
|
47
|
-
act_dropout = dropout if act_dropout is None else act_dropout
|
48
|
-
self.normalize_before = normalize_before
|
49
|
-
|
50
|
-
self.self_attn = MultiHeadAttention(d_model, nhead, attn_dropout)
|
51
|
-
# Implementation of Feedforward model
|
52
|
-
self.linear1 = nn.Linear(d_model, dim_feedforward)
|
53
|
-
self.dropout = nn.Dropout(act_dropout, mode="upscale_in_train")
|
54
|
-
self.linear2 = nn.Linear(dim_feedforward, d_model)
|
55
|
-
|
56
|
-
self.norm1 = nn.LayerNorm(d_model)
|
57
|
-
self.norm2 = nn.LayerNorm(d_model)
|
58
|
-
self.dropout1 = nn.Dropout(dropout, mode="upscale_in_train")
|
59
|
-
self.dropout2 = nn.Dropout(dropout, mode="upscale_in_train")
|
60
|
-
self.activation = getattr(F, activation)
|
61
|
-
self._reset_parameters()
|
62
|
-
|
63
|
-
def _reset_parameters(self):
|
64
|
-
linear_init_(self.linear1)
|
65
|
-
linear_init_(self.linear2)
|
66
|
-
|
67
|
-
@staticmethod
|
68
|
-
def with_pos_embed(tensor, pos_embed):
|
69
|
-
return tensor if pos_embed is None else tensor + pos_embed
|
70
|
-
|
71
|
-
def forward(self, src, src_mask=None, pos_embed=None):
|
72
|
-
src_mask = _convert_attention_mask(src_mask, src.dtype)
|
73
|
-
|
74
|
-
residual = src
|
75
|
-
if self.normalize_before:
|
76
|
-
src = self.norm1(src)
|
77
|
-
q = k = self.with_pos_embed(src, pos_embed)
|
78
|
-
src = self.self_attn(q, k, value=src, attn_mask=src_mask)
|
79
|
-
|
80
|
-
src = residual + self.dropout1(src)
|
81
|
-
if not self.normalize_before:
|
82
|
-
src = self.norm1(src)
|
83
|
-
|
84
|
-
residual = src
|
85
|
-
if self.normalize_before:
|
86
|
-
src = self.norm2(src)
|
87
|
-
src = self.linear2(self.dropout(self.activation(self.linear1(src))))
|
88
|
-
src = residual + self.dropout2(src)
|
89
|
-
if not self.normalize_before:
|
90
|
-
src = self.norm2(src)
|
91
|
-
return src
|
92
|
-
|
93
|
-
|
94
|
-
class TransformerEncoder(nn.Layer):
|
95
|
-
def __init__(self, encoder_layer, num_layers, norm=None):
|
96
|
-
super(TransformerEncoder, self).__init__()
|
97
|
-
self.layers = _get_clones(encoder_layer, num_layers)
|
98
|
-
self.num_layers = num_layers
|
99
|
-
self.norm = norm
|
100
|
-
|
101
|
-
def forward(self, src, src_mask=None, pos_embed=None):
|
102
|
-
src_mask = _convert_attention_mask(src_mask, src.dtype)
|
103
|
-
|
104
|
-
output = src
|
105
|
-
for layer in self.layers:
|
106
|
-
output = layer(output, src_mask=src_mask, pos_embed=pos_embed)
|
107
|
-
|
108
|
-
if self.norm is not None:
|
109
|
-
output = self.norm(output)
|
110
|
-
|
111
|
-
return output
|
112
|
-
|
113
|
-
|
114
|
-
class TransformerDecoderLayer(nn.Layer):
|
115
|
-
def __init__(self,
|
116
|
-
d_model,
|
117
|
-
nhead,
|
118
|
-
dim_feedforward=2048,
|
119
|
-
dropout=0.1,
|
120
|
-
activation="relu",
|
121
|
-
attn_dropout=None,
|
122
|
-
act_dropout=None,
|
123
|
-
normalize_before=False):
|
124
|
-
super(TransformerDecoderLayer, self).__init__()
|
125
|
-
attn_dropout = dropout if attn_dropout is None else attn_dropout
|
126
|
-
act_dropout = dropout if act_dropout is None else act_dropout
|
127
|
-
self.normalize_before = normalize_before
|
128
|
-
|
129
|
-
self.self_attn = MultiHeadAttention(d_model, nhead, attn_dropout)
|
130
|
-
self.cross_attn = MultiHeadAttention(d_model, nhead, attn_dropout)
|
131
|
-
# Implementation of Feedforward model
|
132
|
-
self.linear1 = nn.Linear(d_model, dim_feedforward)
|
133
|
-
self.dropout = nn.Dropout(act_dropout, mode="upscale_in_train")
|
134
|
-
self.linear2 = nn.Linear(dim_feedforward, d_model)
|
135
|
-
|
136
|
-
self.norm1 = nn.LayerNorm(d_model)
|
137
|
-
self.norm2 = nn.LayerNorm(d_model)
|
138
|
-
self.norm3 = nn.LayerNorm(d_model)
|
139
|
-
self.dropout1 = nn.Dropout(dropout, mode="upscale_in_train")
|
140
|
-
self.dropout2 = nn.Dropout(dropout, mode="upscale_in_train")
|
141
|
-
self.dropout3 = nn.Dropout(dropout, mode="upscale_in_train")
|
142
|
-
self.activation = getattr(F, activation)
|
143
|
-
self._reset_parameters()
|
144
|
-
|
145
|
-
def _reset_parameters(self):
|
146
|
-
linear_init_(self.linear1)
|
147
|
-
linear_init_(self.linear2)
|
148
|
-
|
149
|
-
@staticmethod
|
150
|
-
def with_pos_embed(tensor, pos_embed):
|
151
|
-
return tensor if pos_embed is None else tensor + pos_embed
|
152
|
-
|
153
|
-
def forward(self,
|
154
|
-
tgt,
|
155
|
-
memory,
|
156
|
-
tgt_mask=None,
|
157
|
-
memory_mask=None,
|
158
|
-
pos_embed=None,
|
159
|
-
query_pos_embed=None):
|
160
|
-
tgt_mask = _convert_attention_mask(tgt_mask, tgt.dtype)
|
161
|
-
memory_mask = _convert_attention_mask(memory_mask, memory.dtype)
|
162
|
-
|
163
|
-
residual = tgt
|
164
|
-
if self.normalize_before:
|
165
|
-
tgt = self.norm1(tgt)
|
166
|
-
q = k = self.with_pos_embed(tgt, query_pos_embed)
|
167
|
-
tgt = self.self_attn(q, k, value=tgt, attn_mask=tgt_mask)
|
168
|
-
tgt = residual + self.dropout1(tgt)
|
169
|
-
if not self.normalize_before:
|
170
|
-
tgt = self.norm1(tgt)
|
171
|
-
|
172
|
-
residual = tgt
|
173
|
-
if self.normalize_before:
|
174
|
-
tgt = self.norm2(tgt)
|
175
|
-
q = self.with_pos_embed(tgt, query_pos_embed)
|
176
|
-
k = self.with_pos_embed(memory, pos_embed)
|
177
|
-
tgt = self.cross_attn(q, k, value=memory, attn_mask=memory_mask)
|
178
|
-
tgt = residual + self.dropout2(tgt)
|
179
|
-
if not self.normalize_before:
|
180
|
-
tgt = self.norm2(tgt)
|
181
|
-
|
182
|
-
residual = tgt
|
183
|
-
if self.normalize_before:
|
184
|
-
tgt = self.norm3(tgt)
|
185
|
-
tgt = self.linear2(self.dropout(self.activation(self.linear1(tgt))))
|
186
|
-
tgt = residual + self.dropout3(tgt)
|
187
|
-
if not self.normalize_before:
|
188
|
-
tgt = self.norm3(tgt)
|
189
|
-
return tgt
|
190
|
-
|
191
|
-
|
192
|
-
class TransformerDecoder(nn.Layer):
|
193
|
-
def __init__(self,
|
194
|
-
decoder_layer,
|
195
|
-
num_layers,
|
196
|
-
norm=None,
|
197
|
-
return_intermediate=False):
|
198
|
-
super(TransformerDecoder, self).__init__()
|
199
|
-
self.layers = _get_clones(decoder_layer, num_layers)
|
200
|
-
self.num_layers = num_layers
|
201
|
-
self.norm = norm
|
202
|
-
self.return_intermediate = return_intermediate
|
203
|
-
|
204
|
-
def forward(self,
|
205
|
-
tgt,
|
206
|
-
memory,
|
207
|
-
tgt_mask=None,
|
208
|
-
memory_mask=None,
|
209
|
-
pos_embed=None,
|
210
|
-
query_pos_embed=None):
|
211
|
-
tgt_mask = _convert_attention_mask(tgt_mask, tgt.dtype)
|
212
|
-
memory_mask = _convert_attention_mask(memory_mask, memory.dtype)
|
213
|
-
|
214
|
-
output = tgt
|
215
|
-
intermediate = []
|
216
|
-
for layer in self.layers:
|
217
|
-
output = layer(
|
218
|
-
output,
|
219
|
-
memory,
|
220
|
-
tgt_mask=tgt_mask,
|
221
|
-
memory_mask=memory_mask,
|
222
|
-
pos_embed=pos_embed,
|
223
|
-
query_pos_embed=query_pos_embed)
|
224
|
-
if self.return_intermediate:
|
225
|
-
intermediate.append(self.norm(output))
|
226
|
-
|
227
|
-
if self.norm is not None:
|
228
|
-
output = self.norm(output)
|
229
|
-
|
230
|
-
if self.return_intermediate:
|
231
|
-
return paddle.stack(intermediate)
|
232
|
-
|
233
|
-
return output.unsqueeze(0)
|
234
|
-
|
235
|
-
|
236
|
-
@register
|
237
|
-
class DETRTransformer(nn.Layer):
|
238
|
-
__shared__ = ['hidden_dim']
|
239
|
-
|
240
|
-
def __init__(self,
|
241
|
-
num_queries=100,
|
242
|
-
position_embed_type='sine',
|
243
|
-
return_intermediate_dec=True,
|
244
|
-
backbone_num_channels=2048,
|
245
|
-
hidden_dim=256,
|
246
|
-
nhead=8,
|
247
|
-
num_encoder_layers=6,
|
248
|
-
num_decoder_layers=6,
|
249
|
-
dim_feedforward=2048,
|
250
|
-
dropout=0.1,
|
251
|
-
activation="relu",
|
252
|
-
attn_dropout=None,
|
253
|
-
act_dropout=None,
|
254
|
-
normalize_before=False):
|
255
|
-
super(DETRTransformer, self).__init__()
|
256
|
-
assert position_embed_type in ['sine', 'learned'],\
|
257
|
-
f'ValueError: position_embed_type not supported {position_embed_type}!'
|
258
|
-
self.hidden_dim = hidden_dim
|
259
|
-
self.nhead = nhead
|
260
|
-
|
261
|
-
encoder_layer = TransformerEncoderLayer(
|
262
|
-
hidden_dim, nhead, dim_feedforward, dropout, activation,
|
263
|
-
attn_dropout, act_dropout, normalize_before)
|
264
|
-
encoder_norm = nn.LayerNorm(hidden_dim) if normalize_before else None
|
265
|
-
self.encoder = TransformerEncoder(encoder_layer, num_encoder_layers,
|
266
|
-
encoder_norm)
|
267
|
-
|
268
|
-
decoder_layer = TransformerDecoderLayer(
|
269
|
-
hidden_dim, nhead, dim_feedforward, dropout, activation,
|
270
|
-
attn_dropout, act_dropout, normalize_before)
|
271
|
-
decoder_norm = nn.LayerNorm(hidden_dim)
|
272
|
-
self.decoder = TransformerDecoder(
|
273
|
-
decoder_layer,
|
274
|
-
num_decoder_layers,
|
275
|
-
decoder_norm,
|
276
|
-
return_intermediate=return_intermediate_dec)
|
277
|
-
|
278
|
-
self.input_proj = nn.Conv2D(
|
279
|
-
backbone_num_channels, hidden_dim, kernel_size=1)
|
280
|
-
self.query_pos_embed = nn.Embedding(num_queries, hidden_dim)
|
281
|
-
self.position_embedding = PositionEmbedding(
|
282
|
-
hidden_dim // 2,
|
283
|
-
normalize=True if position_embed_type == 'sine' else False,
|
284
|
-
embed_type=position_embed_type)
|
285
|
-
|
286
|
-
self._reset_parameters()
|
287
|
-
|
288
|
-
def _reset_parameters(self):
|
289
|
-
for p in self.parameters():
|
290
|
-
if p.dim() > 1:
|
291
|
-
xavier_uniform_(p)
|
292
|
-
conv_init_(self.input_proj)
|
293
|
-
normal_(self.query_pos_embed.weight)
|
294
|
-
|
295
|
-
@classmethod
|
296
|
-
def from_config(cls, cfg, input_shape):
|
297
|
-
return {
|
298
|
-
'backbone_num_channels': [i.channels for i in input_shape][-1],
|
299
|
-
}
|
300
|
-
|
301
|
-
def forward(self, src, src_mask=None):
|
302
|
-
r"""
|
303
|
-
Applies a Transformer model on the inputs.
|
304
|
-
|
305
|
-
Parameters:
|
306
|
-
src (List(Tensor)): Backbone feature maps with shape [[bs, c, h, w]].
|
307
|
-
src_mask (Tensor, optional): A tensor used in multi-head attention
|
308
|
-
to prevents attention to some unwanted positions, usually the
|
309
|
-
paddings or the subsequent positions. It is a tensor with shape
|
310
|
-
[bs, H, W]`. When the data type is bool, the unwanted positions
|
311
|
-
have `False` values and the others have `True` values. When the
|
312
|
-
data type is int, the unwanted positions have 0 values and the
|
313
|
-
others have 1 values. When the data type is float, the unwanted
|
314
|
-
positions have `-INF` values and the others have 0 values. It
|
315
|
-
can be None when nothing wanted or needed to be prevented
|
316
|
-
attention to. Default None.
|
317
|
-
|
318
|
-
Returns:
|
319
|
-
output (Tensor): [num_levels, batch_size, num_queries, hidden_dim]
|
320
|
-
memory (Tensor): [batch_size, hidden_dim, h, w]
|
321
|
-
"""
|
322
|
-
# use last level feature map
|
323
|
-
src_proj = self.input_proj(src[-1])
|
324
|
-
bs, c, h, w = src_proj.shape
|
325
|
-
# flatten [B, C, H, W] to [B, HxW, C]
|
326
|
-
src_flatten = src_proj.flatten(2).transpose([0, 2, 1])
|
327
|
-
if src_mask is not None:
|
328
|
-
src_mask = F.interpolate(
|
329
|
-
src_mask.unsqueeze(0).astype(src_flatten.dtype),
|
330
|
-
size=(h, w))[0].astype('bool')
|
331
|
-
else:
|
332
|
-
src_mask = paddle.ones([bs, h, w], dtype='bool')
|
333
|
-
pos_embed = self.position_embedding(src_mask).flatten(2).transpose(
|
334
|
-
[0, 2, 1])
|
335
|
-
|
336
|
-
src_mask = _convert_attention_mask(src_mask, src_flatten.dtype)
|
337
|
-
src_mask = src_mask.reshape([bs, 1, 1, -1])
|
338
|
-
|
339
|
-
memory = self.encoder(
|
340
|
-
src_flatten, src_mask=src_mask, pos_embed=pos_embed)
|
341
|
-
|
342
|
-
query_pos_embed = self.query_pos_embed.weight.unsqueeze(0).tile(
|
343
|
-
[bs, 1, 1])
|
344
|
-
tgt = paddle.zeros_like(query_pos_embed)
|
345
|
-
output = self.decoder(
|
346
|
-
tgt,
|
347
|
-
memory,
|
348
|
-
memory_mask=src_mask,
|
349
|
-
pos_embed=pos_embed,
|
350
|
-
query_pos_embed=query_pos_embed)
|
351
|
-
|
352
|
-
return (output, memory.transpose([0, 2, 1]).reshape([bs, c, h, w]),
|
353
|
-
src_proj, src_mask.reshape([bs, 1, 1, h, w]))
|
@@ -1,127 +0,0 @@
|
|
1
|
-
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
#
|
15
|
-
# Modified from DETR (https://github.com/facebookresearch/detr)
|
16
|
-
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
17
|
-
|
18
|
-
from __future__ import absolute_import
|
19
|
-
from __future__ import division
|
20
|
-
from __future__ import print_function
|
21
|
-
|
22
|
-
import paddle
|
23
|
-
import paddle.nn as nn
|
24
|
-
import paddle.nn.functional as F
|
25
|
-
from scipy.optimize import linear_sum_assignment
|
26
|
-
|
27
|
-
from paddlex.ppdet.core.workspace import register, serializable
|
28
|
-
from ..losses.iou_loss import GIoULoss
|
29
|
-
from .utils import bbox_cxcywh_to_xyxy
|
30
|
-
|
31
|
-
__all__ = ['HungarianMatcher']
|
32
|
-
|
33
|
-
|
34
|
-
@register
|
35
|
-
@serializable
|
36
|
-
class HungarianMatcher(nn.Layer):
|
37
|
-
__shared__ = ['use_focal_loss']
|
38
|
-
|
39
|
-
def __init__(self,
|
40
|
-
matcher_coeff={'class': 1,
|
41
|
-
'bbox': 5,
|
42
|
-
'giou': 2},
|
43
|
-
use_focal_loss=False,
|
44
|
-
alpha=0.25,
|
45
|
-
gamma=2.0):
|
46
|
-
r"""
|
47
|
-
Args:
|
48
|
-
matcher_coeff (dict): The coefficient of hungarian matcher cost.
|
49
|
-
"""
|
50
|
-
super(HungarianMatcher, self).__init__()
|
51
|
-
self.matcher_coeff = matcher_coeff
|
52
|
-
self.use_focal_loss = use_focal_loss
|
53
|
-
self.alpha = alpha
|
54
|
-
self.gamma = gamma
|
55
|
-
|
56
|
-
self.giou_loss = GIoULoss()
|
57
|
-
|
58
|
-
def forward(self, boxes, logits, gt_bbox, gt_class):
|
59
|
-
r"""
|
60
|
-
Args:
|
61
|
-
boxes (Tensor): [b, query, 4]
|
62
|
-
logits (Tensor): [b, query, num_classes]
|
63
|
-
gt_bbox (List(Tensor)): list[[n, 4]]
|
64
|
-
gt_class (List(Tensor)): list[[n, 1]]
|
65
|
-
|
66
|
-
Returns:
|
67
|
-
A list of size batch_size, containing tuples of (index_i, index_j) where:
|
68
|
-
- index_i is the indices of the selected predictions (in order)
|
69
|
-
- index_j is the indices of the corresponding selected targets (in order)
|
70
|
-
For each batch element, it holds:
|
71
|
-
len(index_i) = len(index_j) = min(num_queries, num_target_boxes)
|
72
|
-
"""
|
73
|
-
bs, num_queries = boxes.shape[:2]
|
74
|
-
|
75
|
-
num_gts = sum(len(a) for a in gt_class)
|
76
|
-
if num_gts == 0:
|
77
|
-
return [(paddle.to_tensor(
|
78
|
-
[], dtype=paddle.int64), paddle.to_tensor(
|
79
|
-
[], dtype=paddle.int64)) for _ in range(bs)]
|
80
|
-
|
81
|
-
# We flatten to compute the cost matrices in a batch
|
82
|
-
# [batch_size * num_queries, num_classes]
|
83
|
-
out_prob = F.sigmoid(logits.flatten(
|
84
|
-
0, 1)) if self.use_focal_loss else F.softmax(
|
85
|
-
logits.flatten(0, 1))
|
86
|
-
# [batch_size * num_queries, 4]
|
87
|
-
out_bbox = boxes.flatten(0, 1)
|
88
|
-
|
89
|
-
# Also concat the target labels and boxes
|
90
|
-
tgt_ids = paddle.concat(gt_class).flatten()
|
91
|
-
tgt_bbox = paddle.concat(gt_bbox)
|
92
|
-
|
93
|
-
# Compute the classification cost
|
94
|
-
if self.use_focal_loss:
|
95
|
-
neg_cost_class = (1 - self.alpha) * (out_prob**self.gamma) * (-(
|
96
|
-
1 - out_prob + 1e-8).log())
|
97
|
-
pos_cost_class = self.alpha * (
|
98
|
-
(1 - out_prob)**self.gamma) * (-(out_prob + 1e-8).log())
|
99
|
-
cost_class = paddle.gather(
|
100
|
-
pos_cost_class, tgt_ids, axis=1) - paddle.gather(
|
101
|
-
neg_cost_class, tgt_ids, axis=1)
|
102
|
-
else:
|
103
|
-
cost_class = -paddle.gather(out_prob, tgt_ids, axis=1)
|
104
|
-
|
105
|
-
# Compute the L1 cost between boxes
|
106
|
-
cost_bbox = (
|
107
|
-
out_bbox.unsqueeze(1) - tgt_bbox.unsqueeze(0)).abs().sum(-1)
|
108
|
-
|
109
|
-
# Compute the giou cost betwen boxes
|
110
|
-
cost_giou = self.giou_loss(
|
111
|
-
bbox_cxcywh_to_xyxy(out_bbox.unsqueeze(1)),
|
112
|
-
bbox_cxcywh_to_xyxy(tgt_bbox.unsqueeze(0))).squeeze(-1)
|
113
|
-
|
114
|
-
# Final cost matrix
|
115
|
-
C = self.matcher_coeff['class'] * cost_class + self.matcher_coeff['bbox'] * cost_bbox + \
|
116
|
-
self.matcher_coeff['giou'] * cost_giou
|
117
|
-
C = C.reshape([bs, num_queries, -1])
|
118
|
-
C = [a.squeeze(0) for a in C.chunk(bs)]
|
119
|
-
|
120
|
-
sizes = [a.shape[0] for a in gt_bbox]
|
121
|
-
indices = [
|
122
|
-
linear_sum_assignment(c.split(sizes, -1)[i].numpy())
|
123
|
-
for i, c in enumerate(C)
|
124
|
-
]
|
125
|
-
return [(paddle.to_tensor(
|
126
|
-
i, dtype=paddle.int64), paddle.to_tensor(
|
127
|
-
j, dtype=paddle.int64)) for i, j in indices]
|
@@ -1,108 +0,0 @@
|
|
1
|
-
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
#
|
15
|
-
# Modified from DETR (https://github.com/facebookresearch/detr)
|
16
|
-
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
17
|
-
|
18
|
-
from __future__ import absolute_import
|
19
|
-
from __future__ import division
|
20
|
-
from __future__ import print_function
|
21
|
-
|
22
|
-
import math
|
23
|
-
import paddle
|
24
|
-
import paddle.nn as nn
|
25
|
-
|
26
|
-
from paddlex.ppdet.core.workspace import register, serializable
|
27
|
-
|
28
|
-
|
29
|
-
@register
|
30
|
-
@serializable
|
31
|
-
class PositionEmbedding(nn.Layer):
|
32
|
-
def __init__(self,
|
33
|
-
num_pos_feats=128,
|
34
|
-
temperature=10000,
|
35
|
-
normalize=True,
|
36
|
-
scale=None,
|
37
|
-
embed_type='sine',
|
38
|
-
num_embeddings=50,
|
39
|
-
offset=0.):
|
40
|
-
super(PositionEmbedding, self).__init__()
|
41
|
-
assert embed_type in ['sine', 'learned']
|
42
|
-
|
43
|
-
self.embed_type = embed_type
|
44
|
-
self.offset = offset
|
45
|
-
self.eps = 1e-6
|
46
|
-
if self.embed_type == 'sine':
|
47
|
-
self.num_pos_feats = num_pos_feats
|
48
|
-
self.temperature = temperature
|
49
|
-
self.normalize = normalize
|
50
|
-
if scale is not None and normalize is False:
|
51
|
-
raise ValueError("normalize should be True if scale is passed")
|
52
|
-
if scale is None:
|
53
|
-
scale = 2 * math.pi
|
54
|
-
self.scale = scale
|
55
|
-
elif self.embed_type == 'learned':
|
56
|
-
self.row_embed = nn.Embedding(num_embeddings, num_pos_feats)
|
57
|
-
self.col_embed = nn.Embedding(num_embeddings, num_pos_feats)
|
58
|
-
else:
|
59
|
-
raise ValueError(f"not supported {self.embed_type}")
|
60
|
-
|
61
|
-
def forward(self, mask):
|
62
|
-
"""
|
63
|
-
Args:
|
64
|
-
mask (Tensor): [B, H, W]
|
65
|
-
Returns:
|
66
|
-
pos (Tensor): [B, C, H, W]
|
67
|
-
"""
|
68
|
-
assert mask.dtype == paddle.bool
|
69
|
-
if self.embed_type == 'sine':
|
70
|
-
mask = mask.astype('float32')
|
71
|
-
y_embed = mask.cumsum(1, dtype='float32')
|
72
|
-
x_embed = mask.cumsum(2, dtype='float32')
|
73
|
-
if self.normalize:
|
74
|
-
y_embed = (y_embed + self.offset) / (
|
75
|
-
y_embed[:, -1:, :] + self.eps) * self.scale
|
76
|
-
x_embed = (x_embed + self.offset) / (
|
77
|
-
x_embed[:, :, -1:] + self.eps) * self.scale
|
78
|
-
|
79
|
-
dim_t = 2 * (paddle.arange(self.num_pos_feats) //
|
80
|
-
2).astype('float32')
|
81
|
-
dim_t = self.temperature**(dim_t / self.num_pos_feats)
|
82
|
-
|
83
|
-
pos_x = x_embed.unsqueeze(-1) / dim_t
|
84
|
-
pos_y = y_embed.unsqueeze(-1) / dim_t
|
85
|
-
pos_x = paddle.stack(
|
86
|
-
(pos_x[:, :, :, 0::2].sin(), pos_x[:, :, :, 1::2].cos()),
|
87
|
-
axis=4).flatten(3)
|
88
|
-
pos_y = paddle.stack(
|
89
|
-
(pos_y[:, :, :, 0::2].sin(), pos_y[:, :, :, 1::2].cos()),
|
90
|
-
axis=4).flatten(3)
|
91
|
-
pos = paddle.concat((pos_y, pos_x), axis=3).transpose([0, 3, 1, 2])
|
92
|
-
return pos
|
93
|
-
elif self.embed_type == 'learned':
|
94
|
-
h, w = mask.shape[-2:]
|
95
|
-
i = paddle.arange(w)
|
96
|
-
j = paddle.arange(h)
|
97
|
-
x_emb = self.col_embed(i)
|
98
|
-
y_emb = self.row_embed(j)
|
99
|
-
pos = paddle.concat(
|
100
|
-
[
|
101
|
-
x_emb.unsqueeze(0).repeat(h, 1, 1),
|
102
|
-
y_emb.unsqueeze(1).repeat(1, w, 1),
|
103
|
-
],
|
104
|
-
axis=-1).transpose([2, 0, 1]).unsqueeze(0).tile(mask.shape[0],
|
105
|
-
1, 1, 1)
|
106
|
-
return pos
|
107
|
-
else:
|
108
|
-
raise ValueError(f"not supported {self.embed_type}")
|
@@ -1,110 +0,0 @@
|
|
1
|
-
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
#
|
15
|
-
# Modified from DETR (https://github.com/facebookresearch/detr)
|
16
|
-
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
17
|
-
|
18
|
-
from __future__ import absolute_import
|
19
|
-
from __future__ import division
|
20
|
-
from __future__ import print_function
|
21
|
-
|
22
|
-
import copy
|
23
|
-
import paddle
|
24
|
-
import paddle.nn as nn
|
25
|
-
import paddle.nn.functional as F
|
26
|
-
|
27
|
-
from ..bbox_utils import bbox_overlaps
|
28
|
-
|
29
|
-
__all__ = [
|
30
|
-
'_get_clones', 'bbox_overlaps', 'bbox_cxcywh_to_xyxy',
|
31
|
-
'bbox_xyxy_to_cxcywh', 'sigmoid_focal_loss', 'inverse_sigmoid',
|
32
|
-
'deformable_attention_core_func'
|
33
|
-
]
|
34
|
-
|
35
|
-
|
36
|
-
def _get_clones(module, N):
|
37
|
-
return nn.LayerList([copy.deepcopy(module) for _ in range(N)])
|
38
|
-
|
39
|
-
|
40
|
-
def bbox_cxcywh_to_xyxy(x):
|
41
|
-
x_c, y_c, w, h = x.unbind(-1)
|
42
|
-
b = [(x_c - 0.5 * w), (y_c - 0.5 * h), (x_c + 0.5 * w), (y_c + 0.5 * h)]
|
43
|
-
return paddle.stack(b, axis=-1)
|
44
|
-
|
45
|
-
|
46
|
-
def bbox_xyxy_to_cxcywh(x):
|
47
|
-
x0, y0, x1, y1 = x.unbind(-1)
|
48
|
-
b = [(x0 + x1) / 2, (y0 + y1) / 2, (x1 - x0), (y1 - y0)]
|
49
|
-
return paddle.stack(b, axis=-1)
|
50
|
-
|
51
|
-
|
52
|
-
def sigmoid_focal_loss(logit, label, normalizer=1.0, alpha=0.25, gamma=2.0):
|
53
|
-
prob = F.sigmoid(logit)
|
54
|
-
ce_loss = F.binary_cross_entropy_with_logits(
|
55
|
-
logit, label, reduction="none")
|
56
|
-
p_t = prob * label + (1 - prob) * (1 - label)
|
57
|
-
loss = ce_loss * ((1 - p_t)**gamma)
|
58
|
-
|
59
|
-
if alpha >= 0:
|
60
|
-
alpha_t = alpha * label + (1 - alpha) * (1 - label)
|
61
|
-
loss = alpha_t * loss
|
62
|
-
return loss.mean(1).sum() / normalizer
|
63
|
-
|
64
|
-
|
65
|
-
def inverse_sigmoid(x, eps=1e-6):
|
66
|
-
x = x.clip(min=0., max=1.)
|
67
|
-
return paddle.log(x / (1 - x + eps) + eps)
|
68
|
-
|
69
|
-
|
70
|
-
def deformable_attention_core_func(value, value_spatial_shapes,
|
71
|
-
sampling_locations, attention_weights):
|
72
|
-
"""
|
73
|
-
Args:
|
74
|
-
value (Tensor): [bs, value_length, n_head, c]
|
75
|
-
value_spatial_shapes (Tensor): [n_levels, 2]
|
76
|
-
sampling_locations (Tensor): [bs, query_length, n_head, n_levels, n_points, 2]
|
77
|
-
attention_weights (Tensor): [bs, query_length, n_head, n_levels, n_points]
|
78
|
-
|
79
|
-
Returns:
|
80
|
-
output (Tensor): [bs, Length_{query}, C]
|
81
|
-
"""
|
82
|
-
bs, Len_v, n_head, c = value.shape
|
83
|
-
_, Len_q, n_head, n_levels, n_points, _ = sampling_locations.shape
|
84
|
-
|
85
|
-
value_list = value.split(value_spatial_shapes.prod(1).tolist(), axis=1)
|
86
|
-
sampling_grids = 2 * sampling_locations - 1
|
87
|
-
sampling_value_list = []
|
88
|
-
for level, (h, w) in enumerate(value_spatial_shapes.tolist()):
|
89
|
-
# N_, H_*W_, M_, D_ -> N_, H_*W_, M_*D_ -> N_, M_*D_, H_*W_ -> N_*M_, D_, H_, W_
|
90
|
-
value_l_ = value_list[level].flatten(2).transpose(
|
91
|
-
[0, 2, 1]).reshape([bs * n_head, c, h, w])
|
92
|
-
# N_, Lq_, M_, P_, 2 -> N_, M_, Lq_, P_, 2 -> N_*M_, Lq_, P_, 2
|
93
|
-
sampling_grid_l_ = sampling_grids[:, :, :, level].transpose(
|
94
|
-
[0, 2, 1, 3, 4]).flatten(0, 1)
|
95
|
-
# N_*M_, D_, Lq_, P_
|
96
|
-
sampling_value_l_ = F.grid_sample(
|
97
|
-
value_l_,
|
98
|
-
sampling_grid_l_,
|
99
|
-
mode='bilinear',
|
100
|
-
padding_mode='zeros',
|
101
|
-
align_corners=False)
|
102
|
-
sampling_value_list.append(sampling_value_l_)
|
103
|
-
# (N_, Lq_, M_, L_, P_) -> (N_, M_, Lq_, L_, P_) -> (N_*M_, 1, Lq_, L_*P_)
|
104
|
-
attention_weights = attention_weights.transpose([0, 2, 1, 3, 4]).reshape(
|
105
|
-
[bs * n_head, 1, Len_q, n_levels * n_points])
|
106
|
-
output = (paddle.stack(
|
107
|
-
sampling_value_list, axis=-2).flatten(-2) *
|
108
|
-
attention_weights).sum(-1).reshape([bs, n_head * c, Len_q])
|
109
|
-
|
110
|
-
return output.transpose([0, 2, 1])
|