paddlex 2.1.0__py3-none-any.whl → 3.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- paddlex/.version +1 -0
- paddlex/__init__.py +35 -19
- paddlex/__main__.py +39 -0
- paddlex/configs/modules/3d_bev_detection/BEVFusion.yaml +38 -0
- paddlex/configs/modules/chart_parsing/PP-Chart2Table.yaml +13 -0
- paddlex/configs/modules/doc_text_orientation/PP-LCNet_x1_0_doc_ori.yaml +41 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-2B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-7B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee2-3B.yaml +14 -0
- paddlex/configs/modules/face_detection/BlazeFace-FPN-SSH.yaml +40 -0
- paddlex/configs/modules/face_detection/BlazeFace.yaml +40 -0
- paddlex/configs/modules/face_detection/PP-YOLOE_plus-S_face.yaml +40 -0
- paddlex/configs/modules/face_detection/PicoDet_LCNet_x2_5_face.yaml +40 -0
- paddlex/configs/modules/face_feature/MobileFaceNet.yaml +41 -0
- paddlex/configs/modules/face_feature/ResNet50_face.yaml +41 -0
- paddlex/configs/modules/formula_recognition/LaTeX_OCR_rec.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-M.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/UniMERNet.yaml +40 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-L_human.yaml +42 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-S_human.yaml +42 -0
- paddlex/configs/modules/image_anomaly_detection/STFPM.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_base_patch16_224.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_large_patch14_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_small.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/FasterNet-L.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-M.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-S.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T0.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T1.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T2.yaml +40 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_small.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B1.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B2.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B3.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B4.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B6.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_large.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_5.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet200_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S1.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S2.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S3.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S4.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_small_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_tiny_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_base.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_large.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/CLIP_vit_base_patch16_448_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B4_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B6_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-LCNet_x1_0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/ResNet50_ML.yaml +41 -0
- paddlex/configs/modules/image_unwarping/UVDoc.yaml +12 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-M.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNeXt101-vd-FPN.yaml +39 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/PP-YOLOE_seg-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/SOLOv2.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_128x96.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_256x192.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocBlockLayout.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-M.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-S.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout_plus-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x_table.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_3cls.yaml +40 -0
- paddlex/configs/modules/mainbody_detection/PP-ShiTuV2_det.yaml +41 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_base.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_large.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_medium.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_small.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_tiny.yaml +12 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-FPN.yaml +41 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/CenterNet-DLA-34.yaml +41 -0
- paddlex/configs/modules/object_detection/CenterNet-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/Co-DINO-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-DINO-Swin-L.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-Swin-T.yaml +40 -0
- paddlex/configs/modules/object_detection/DETR-R50.yaml +42 -0
- paddlex/configs/modules/object_detection/FCOS-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNeXt101-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet34-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-Swin-Tiny-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-M.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-X.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-M.yaml +42 -0
- paddlex/configs/modules/object_detection/PicoDet-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-XS.yaml +42 -0
- paddlex/configs/modules/object_detection/RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R18.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-L.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-M.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-N.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-S.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-T.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-DarkNet53.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-MobileNetV3.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-ResNet50_vd_DCN.yaml +40 -0
- paddlex/configs/modules/open_vocabulary_detection/GroundingDINO-T.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_detection/YOLO-Worldv2-L.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_box.yaml +17 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_point.yaml +15 -0
- paddlex/configs/modules/pedestrian_attribute_recognition/PP-LCNet_x1_0_pedestrian_attribute.yaml +41 -0
- paddlex/configs/modules/rotated_object_detection/PP-YOLOE-R-L.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_mobile_seal_det.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_server_seal_det.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_small.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_tiny.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W18.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W48.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-B.yaml +41 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-T.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_base.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_large.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_small.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_tiny.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B0.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B1.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B2.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B3.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B4.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B5.yaml +40 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-L.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-S.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml +42 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wired_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wireless_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_classification/PP-LCNet_x1_0_table_cls.yaml +41 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wired.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wireless.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet_plus.yaml +39 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_server_det.yaml +40 -0
- paddlex/configs/modules/text_recognition/PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec_doc.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/arabic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_RepSVTR_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_SVTRv2_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/chinese_cht_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/cyrillic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/devanagari_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/japan_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ka_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/korean_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/latin_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ta_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/te_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/textline_orientation/PP-LCNet_x0_25_textline_ori.yaml +41 -0
- paddlex/configs/modules/ts_anomaly_detection/AutoEncoder_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/DLinear_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/Nonstationary_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/PatchTST_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/TimesNet_ad.yaml +37 -0
- paddlex/configs/modules/ts_classification/TimesNet_cls.yaml +37 -0
- paddlex/configs/modules/ts_forecast/DLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/NLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/Nonstationary.yaml +38 -0
- paddlex/configs/modules/ts_forecast/PatchTST.yaml +38 -0
- paddlex/configs/modules/ts_forecast/RLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TiDE.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TimesNet.yaml +38 -0
- paddlex/configs/modules/vehicle_attribute_recognition/PP-LCNet_x1_0_vehicle_attribute.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-L_vehicle.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-S_vehicle.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSM-R50_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_16frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_detection/YOWO.yaml +40 -0
- paddlex/configs/pipelines/3d_bev_detection.yaml +9 -0
- paddlex/configs/pipelines/OCR.yaml +45 -0
- paddlex/configs/pipelines/PP-ChatOCRv3-doc.yaml +151 -0
- paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml +237 -0
- paddlex/configs/pipelines/PP-ShiTuV2.yaml +18 -0
- paddlex/configs/pipelines/PP-StructureV3.yaml +226 -0
- paddlex/configs/pipelines/anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/doc_preprocessor.yaml +15 -0
- paddlex/configs/pipelines/doc_understanding.yaml +9 -0
- paddlex/configs/pipelines/face_recognition.yaml +18 -0
- paddlex/configs/pipelines/formula_recognition.yaml +39 -0
- paddlex/configs/pipelines/human_keypoint_detection.yaml +17 -0
- paddlex/configs/pipelines/image_classification.yaml +10 -0
- paddlex/configs/pipelines/image_multilabel_classification.yaml +9 -0
- paddlex/configs/pipelines/instance_segmentation.yaml +10 -0
- paddlex/configs/pipelines/layout_parsing.yaml +102 -0
- paddlex/configs/pipelines/multilingual_speech_recognition.yaml +9 -0
- paddlex/configs/pipelines/object_detection.yaml +10 -0
- paddlex/configs/pipelines/open_vocabulary_detection.yaml +12 -0
- paddlex/configs/pipelines/open_vocabulary_segmentation.yaml +13 -0
- paddlex/configs/pipelines/pedestrian_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/rotated_object_detection.yaml +10 -0
- paddlex/configs/pipelines/seal_recognition.yaml +52 -0
- paddlex/configs/pipelines/semantic_segmentation.yaml +10 -0
- paddlex/configs/pipelines/small_object_detection.yaml +10 -0
- paddlex/configs/pipelines/table_recognition.yaml +57 -0
- paddlex/configs/pipelines/table_recognition_v2.yaml +82 -0
- paddlex/configs/pipelines/ts_anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/ts_classification.yaml +8 -0
- paddlex/configs/pipelines/ts_forecast.yaml +8 -0
- paddlex/configs/pipelines/vehicle_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/video_classification.yaml +9 -0
- paddlex/configs/pipelines/video_detection.yaml +10 -0
- paddlex/constants.py +17 -0
- paddlex/engine.py +56 -0
- paddlex/hpip_links.html +31 -0
- paddlex/inference/__init__.py +19 -0
- paddlex/inference/common/__init__.py +13 -0
- paddlex/inference/common/batch_sampler/__init__.py +21 -0
- paddlex/inference/common/batch_sampler/audio_batch_sampler.py +83 -0
- paddlex/inference/common/batch_sampler/base_batch_sampler.py +94 -0
- paddlex/inference/common/batch_sampler/det_3d_batch_sampler.py +144 -0
- paddlex/inference/common/batch_sampler/doc_vlm_batch_sampler.py +87 -0
- paddlex/inference/common/batch_sampler/image_batch_sampler.py +121 -0
- paddlex/inference/common/batch_sampler/ts_batch_sampler.py +109 -0
- paddlex/inference/common/batch_sampler/video_batch_sampler.py +74 -0
- paddlex/inference/common/reader/__init__.py +19 -0
- paddlex/inference/common/reader/audio_reader.py +46 -0
- paddlex/inference/common/reader/det_3d_reader.py +241 -0
- paddlex/inference/common/reader/image_reader.py +73 -0
- paddlex/inference/common/reader/ts_reader.py +46 -0
- paddlex/inference/common/reader/video_reader.py +42 -0
- paddlex/inference/common/result/__init__.py +29 -0
- paddlex/inference/common/result/base_cv_result.py +41 -0
- paddlex/inference/common/result/base_result.py +72 -0
- paddlex/inference/common/result/base_ts_result.py +41 -0
- paddlex/inference/common/result/base_video_result.py +36 -0
- paddlex/inference/common/result/mixin.py +709 -0
- paddlex/inference/models/__init__.py +86 -0
- paddlex/inference/models/anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/anomaly_detection/predictor.py +135 -0
- paddlex/inference/models/anomaly_detection/processors.py +53 -0
- paddlex/inference/models/anomaly_detection/result.py +71 -0
- paddlex/inference/models/base/__init__.py +15 -0
- paddlex/inference/models/base/predictor/__init__.py +15 -0
- paddlex/inference/models/base/predictor/base_predictor.py +414 -0
- paddlex/inference/models/common/__init__.py +26 -0
- paddlex/inference/models/common/static_infer.py +801 -0
- paddlex/inference/models/common/tokenizer/__init__.py +21 -0
- paddlex/inference/models/common/tokenizer/bert_tokenizer.py +655 -0
- paddlex/inference/models/common/tokenizer/clip_tokenizer.py +609 -0
- paddlex/inference/models/common/tokenizer/gpt_tokenizer.py +453 -0
- paddlex/inference/models/common/tokenizer/qwen2_5_tokenizer.py +112 -0
- paddlex/inference/models/common/tokenizer/qwen2_tokenizer.py +438 -0
- paddlex/inference/models/common/tokenizer/qwen_tokenizer.py +288 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils.py +2149 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils_base.py +3720 -0
- paddlex/inference/models/common/tokenizer/utils.py +66 -0
- paddlex/inference/models/common/tokenizer/vocab.py +647 -0
- paddlex/inference/models/common/ts/__init__.py +15 -0
- paddlex/inference/models/common/ts/funcs.py +540 -0
- paddlex/inference/models/common/ts/processors.py +322 -0
- paddlex/inference/models/common/vision/__init__.py +23 -0
- paddlex/inference/models/common/vision/funcs.py +98 -0
- paddlex/inference/models/common/vision/processors.py +285 -0
- paddlex/inference/models/common/vlm/__init__.py +13 -0
- paddlex/inference/models/common/vlm/activations.py +189 -0
- paddlex/inference/models/common/vlm/bert_padding.py +127 -0
- paddlex/inference/models/common/vlm/conversion_utils.py +99 -0
- paddlex/inference/models/common/vlm/distributed.py +229 -0
- paddlex/inference/models/common/vlm/flash_attn_utils.py +119 -0
- paddlex/inference/models/common/vlm/fusion_ops.py +205 -0
- paddlex/inference/models/common/vlm/generation/__init__.py +34 -0
- paddlex/inference/models/common/vlm/generation/configuration_utils.py +533 -0
- paddlex/inference/models/common/vlm/generation/logits_process.py +730 -0
- paddlex/inference/models/common/vlm/generation/stopping_criteria.py +106 -0
- paddlex/inference/models/common/vlm/generation/utils.py +2162 -0
- paddlex/inference/models/common/vlm/transformers/__init__.py +16 -0
- paddlex/inference/models/common/vlm/transformers/configuration_utils.py +1037 -0
- paddlex/inference/models/common/vlm/transformers/conversion_utils.py +408 -0
- paddlex/inference/models/common/vlm/transformers/model_outputs.py +1612 -0
- paddlex/inference/models/common/vlm/transformers/model_utils.py +2014 -0
- paddlex/inference/models/common/vlm/transformers/utils.py +178 -0
- paddlex/inference/models/common/vlm/utils.py +109 -0
- paddlex/inference/models/doc_vlm/__init__.py +15 -0
- paddlex/inference/models/doc_vlm/modeling/GOT_ocr_2_0.py +830 -0
- paddlex/inference/models/doc_vlm/modeling/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2.py +1606 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_5_vl.py +3006 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_vl.py +2495 -0
- paddlex/inference/models/doc_vlm/predictor.py +253 -0
- paddlex/inference/models/doc_vlm/processors/GOT_ocr_2_0.py +97 -0
- paddlex/inference/models/doc_vlm/processors/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/processors/common.py +561 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_5_vl.py +548 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_vl.py +543 -0
- paddlex/inference/models/doc_vlm/result.py +21 -0
- paddlex/inference/models/face_feature/__init__.py +15 -0
- paddlex/inference/models/face_feature/predictor.py +66 -0
- paddlex/inference/models/formula_recognition/__init__.py +15 -0
- paddlex/inference/models/formula_recognition/predictor.py +193 -0
- paddlex/inference/models/formula_recognition/processors.py +1015 -0
- paddlex/inference/models/formula_recognition/result.py +411 -0
- paddlex/inference/models/image_classification/__init__.py +15 -0
- paddlex/inference/models/image_classification/predictor.py +172 -0
- paddlex/inference/models/image_classification/processors.py +89 -0
- paddlex/inference/models/image_classification/result.py +93 -0
- paddlex/inference/models/image_feature/__init__.py +15 -0
- paddlex/inference/models/image_feature/predictor.py +146 -0
- paddlex/inference/models/image_feature/processors.py +31 -0
- paddlex/inference/models/image_feature/result.py +32 -0
- paddlex/inference/models/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/models/image_multilabel_classification/predictor.py +95 -0
- paddlex/inference/models/image_multilabel_classification/processors.py +89 -0
- paddlex/inference/models/image_multilabel_classification/result.py +96 -0
- paddlex/inference/models/image_unwarping/__init__.py +15 -0
- paddlex/inference/models/image_unwarping/predictor.py +97 -0
- paddlex/inference/models/image_unwarping/processors.py +92 -0
- paddlex/inference/models/image_unwarping/result.py +47 -0
- paddlex/inference/models/instance_segmentation/__init__.py +15 -0
- paddlex/inference/models/instance_segmentation/predictor.py +202 -0
- paddlex/inference/models/instance_segmentation/processors.py +102 -0
- paddlex/inference/models/instance_segmentation/result.py +162 -0
- paddlex/inference/models/keypoint_detection/__init__.py +15 -0
- paddlex/inference/models/keypoint_detection/predictor.py +190 -0
- paddlex/inference/models/keypoint_detection/processors.py +367 -0
- paddlex/inference/models/keypoint_detection/result.py +197 -0
- paddlex/inference/models/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/models/m_3d_bev_detection/predictor.py +303 -0
- paddlex/inference/models/m_3d_bev_detection/processors.py +990 -0
- paddlex/inference/models/m_3d_bev_detection/result.py +68 -0
- paddlex/inference/models/m_3d_bev_detection/visualizer_3d.py +169 -0
- paddlex/inference/models/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/models/multilingual_speech_recognition/predictor.py +137 -0
- paddlex/inference/models/multilingual_speech_recognition/processors.py +1933 -0
- paddlex/inference/models/multilingual_speech_recognition/result.py +21 -0
- paddlex/inference/models/object_detection/__init__.py +15 -0
- paddlex/inference/models/object_detection/predictor.py +344 -0
- paddlex/inference/models/object_detection/processors.py +885 -0
- paddlex/inference/models/object_detection/result.py +114 -0
- paddlex/inference/models/object_detection/utils.py +70 -0
- paddlex/inference/models/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_detection/predictor.py +172 -0
- paddlex/inference/models/open_vocabulary_detection/processors/__init__.py +16 -0
- paddlex/inference/models/open_vocabulary_detection/processors/common.py +114 -0
- paddlex/inference/models/open_vocabulary_detection/processors/groundingdino_processors.py +496 -0
- paddlex/inference/models/open_vocabulary_detection/processors/yoloworld_processors.py +209 -0
- paddlex/inference/models/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/predictor.py +113 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/sam_processer.py +249 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/sam_result.py +149 -0
- paddlex/inference/models/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/models/semantic_segmentation/predictor.py +158 -0
- paddlex/inference/models/semantic_segmentation/processors.py +117 -0
- paddlex/inference/models/semantic_segmentation/result.py +73 -0
- paddlex/inference/models/table_structure_recognition/__init__.py +15 -0
- paddlex/inference/models/table_structure_recognition/predictor.py +161 -0
- paddlex/inference/models/table_structure_recognition/processors.py +229 -0
- paddlex/inference/models/table_structure_recognition/result.py +63 -0
- paddlex/inference/models/text_detection/__init__.py +15 -0
- paddlex/inference/models/text_detection/predictor.py +191 -0
- paddlex/inference/models/text_detection/processors.py +538 -0
- paddlex/inference/models/text_detection/result.py +46 -0
- paddlex/inference/models/text_recognition/__init__.py +15 -0
- paddlex/inference/models/text_recognition/predictor.py +98 -0
- paddlex/inference/models/text_recognition/processors.py +245 -0
- paddlex/inference/models/text_recognition/result.py +76 -0
- paddlex/inference/models/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/ts_anomaly_detection/predictor.py +141 -0
- paddlex/inference/models/ts_anomaly_detection/processors.py +98 -0
- paddlex/inference/models/ts_anomaly_detection/result.py +83 -0
- paddlex/inference/models/ts_classification/__init__.py +15 -0
- paddlex/inference/models/ts_classification/predictor.py +122 -0
- paddlex/inference/models/ts_classification/processors.py +122 -0
- paddlex/inference/models/ts_classification/result.py +87 -0
- paddlex/inference/models/ts_forecasting/__init__.py +15 -0
- paddlex/inference/models/ts_forecasting/predictor.py +154 -0
- paddlex/inference/models/ts_forecasting/processors.py +158 -0
- paddlex/inference/models/ts_forecasting/result.py +96 -0
- paddlex/inference/models/video_classification/__init__.py +15 -0
- paddlex/inference/models/video_classification/predictor.py +141 -0
- paddlex/inference/models/video_classification/processors.py +409 -0
- paddlex/inference/models/video_classification/result.py +96 -0
- paddlex/inference/models/video_detection/__init__.py +15 -0
- paddlex/inference/models/video_detection/predictor.py +129 -0
- paddlex/inference/models/video_detection/processors.py +463 -0
- paddlex/inference/models/video_detection/result.py +109 -0
- paddlex/inference/pipelines/__init__.py +239 -0
- paddlex/inference/pipelines/_parallel.py +172 -0
- paddlex/inference/pipelines/anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/anomaly_detection/pipeline.py +82 -0
- paddlex/inference/pipelines/attribute_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/attribute_recognition/pipeline.py +120 -0
- paddlex/inference/pipelines/attribute_recognition/result.py +102 -0
- paddlex/inference/pipelines/base.py +156 -0
- paddlex/inference/pipelines/components/__init__.py +29 -0
- paddlex/inference/pipelines/components/chat_server/__init__.py +16 -0
- paddlex/inference/pipelines/components/chat_server/base.py +39 -0
- paddlex/inference/pipelines/components/chat_server/openai_bot_chat.py +236 -0
- paddlex/inference/pipelines/components/common/__init__.py +19 -0
- paddlex/inference/pipelines/components/common/base_operator.py +37 -0
- paddlex/inference/pipelines/components/common/base_result.py +66 -0
- paddlex/inference/pipelines/components/common/convert_points_and_boxes.py +45 -0
- paddlex/inference/pipelines/components/common/crop_image_regions.py +556 -0
- paddlex/inference/pipelines/components/common/seal_det_warp.py +972 -0
- paddlex/inference/pipelines/components/common/sort_boxes.py +85 -0
- paddlex/inference/pipelines/components/common/warp_image.py +50 -0
- paddlex/inference/pipelines/components/faisser.py +357 -0
- paddlex/inference/pipelines/components/prompt_engineering/__init__.py +16 -0
- paddlex/inference/pipelines/components/prompt_engineering/base.py +35 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_ensemble_prompt.py +128 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_kie_prompt.py +148 -0
- paddlex/inference/pipelines/components/retriever/__init__.py +16 -0
- paddlex/inference/pipelines/components/retriever/base.py +228 -0
- paddlex/inference/pipelines/components/retriever/openai_bot_retriever.py +70 -0
- paddlex/inference/pipelines/components/retriever/qianfan_bot_retriever.py +166 -0
- paddlex/inference/pipelines/components/utils/__init__.py +13 -0
- paddlex/inference/pipelines/components/utils/mixin.py +206 -0
- paddlex/inference/pipelines/doc_preprocessor/__init__.py +15 -0
- paddlex/inference/pipelines/doc_preprocessor/pipeline.py +209 -0
- paddlex/inference/pipelines/doc_preprocessor/result.py +98 -0
- paddlex/inference/pipelines/doc_understanding/__init__.py +15 -0
- paddlex/inference/pipelines/doc_understanding/pipeline.py +71 -0
- paddlex/inference/pipelines/face_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/face_recognition/pipeline.py +63 -0
- paddlex/inference/pipelines/face_recognition/result.py +44 -0
- paddlex/inference/pipelines/formula_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/formula_recognition/pipeline.py +347 -0
- paddlex/inference/pipelines/formula_recognition/result.py +282 -0
- paddlex/inference/pipelines/image_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_classification/pipeline.py +90 -0
- paddlex/inference/pipelines/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_multilabel_classification/pipeline.py +97 -0
- paddlex/inference/pipelines/instance_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/instance_segmentation/pipeline.py +91 -0
- paddlex/inference/pipelines/keypoint_detection/__init__.py +15 -0
- paddlex/inference/pipelines/keypoint_detection/pipeline.py +158 -0
- paddlex/inference/pipelines/layout_parsing/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/pipeline.py +568 -0
- paddlex/inference/pipelines/layout_parsing/pipeline_v2.py +1382 -0
- paddlex/inference/pipelines/layout_parsing/result.py +191 -0
- paddlex/inference/pipelines/layout_parsing/result_v2.py +745 -0
- paddlex/inference/pipelines/layout_parsing/setting.py +87 -0
- paddlex/inference/pipelines/layout_parsing/utils.py +951 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/utils.py +1143 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/xycuts.py +562 -0
- paddlex/inference/pipelines/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/pipelines/m_3d_bev_detection/pipeline.py +74 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/pipeline.py +78 -0
- paddlex/inference/pipelines/object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/object_detection/pipeline.py +115 -0
- paddlex/inference/pipelines/ocr/__init__.py +15 -0
- paddlex/inference/pipelines/ocr/pipeline.py +463 -0
- paddlex/inference/pipelines/ocr/result.py +255 -0
- paddlex/inference/pipelines/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_detection/pipeline.py +86 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/pipeline.py +100 -0
- paddlex/inference/pipelines/pp_chatocr/__init__.py +16 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_base.py +111 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v3.py +781 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v4.py +992 -0
- paddlex/inference/pipelines/pp_shitu_v2/__init__.py +15 -0
- paddlex/inference/pipelines/pp_shitu_v2/pipeline.py +156 -0
- paddlex/inference/pipelines/pp_shitu_v2/result.py +126 -0
- paddlex/inference/pipelines/rotated_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/rotated_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/seal_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/seal_recognition/pipeline.py +335 -0
- paddlex/inference/pipelines/seal_recognition/result.py +89 -0
- paddlex/inference/pipelines/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/semantic_segmentation/pipeline.py +95 -0
- paddlex/inference/pipelines/small_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/small_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/table_recognition/__init__.py +16 -0
- paddlex/inference/pipelines/table_recognition/pipeline.py +486 -0
- paddlex/inference/pipelines/table_recognition/pipeline_v2.py +1395 -0
- paddlex/inference/pipelines/table_recognition/result.py +218 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing.py +366 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing_v2.py +488 -0
- paddlex/inference/pipelines/table_recognition/utils.py +44 -0
- paddlex/inference/pipelines/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/ts_anomaly_detection/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_classification/__init__.py +15 -0
- paddlex/inference/pipelines/ts_classification/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_forecasting/__init__.py +15 -0
- paddlex/inference/pipelines/ts_forecasting/pipeline.py +72 -0
- paddlex/inference/pipelines/video_classification/__init__.py +15 -0
- paddlex/inference/pipelines/video_classification/pipeline.py +79 -0
- paddlex/inference/pipelines/video_detection/__init__.py +15 -0
- paddlex/inference/pipelines/video_detection/pipeline.py +86 -0
- paddlex/inference/serving/__init__.py +17 -0
- paddlex/inference/serving/basic_serving/__init__.py +18 -0
- paddlex/inference/serving/basic_serving/_app.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/__init__.py +44 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/__init__.py +13 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/common.py +104 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/image_recognition.py +36 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/ocr.py +95 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/anomaly_detection.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_preprocessor.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_understanding.py +153 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/face_recognition.py +226 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/formula_recognition.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/human_keypoint_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_classification.py +69 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_multilabel_classification.py +73 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/instance_segmentation.py +87 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/layout_parsing.py +117 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/m_3d_bev_detection.py +79 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/multilingual_speech_recognition.py +92 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/object_detection.py +77 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ocr.py +102 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_segmentation.py +91 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pedestrian_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv3_doc.py +193 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv4_doc.py +223 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_shituv2.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_structurev3.py +143 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/rotated_object_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/seal_recognition.py +106 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/semantic_segmentation.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/small_object_detection.py +72 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition.py +108 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition_v2.py +113 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_anomaly_detection.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_classification.py +64 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_forecast.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/vehicle_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_classification.py +76 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_detection.py +92 -0
- paddlex/inference/serving/basic_serving/_server.py +40 -0
- paddlex/inference/serving/infra/__init__.py +13 -0
- paddlex/inference/serving/infra/config.py +36 -0
- paddlex/inference/serving/infra/models.py +79 -0
- paddlex/inference/serving/infra/storage.py +180 -0
- paddlex/inference/serving/infra/utils.py +285 -0
- paddlex/inference/serving/schemas/__init__.py +13 -0
- paddlex/inference/serving/schemas/anomaly_detection.py +39 -0
- paddlex/inference/serving/schemas/doc_preprocessor.py +54 -0
- paddlex/inference/serving/schemas/doc_understanding.py +78 -0
- paddlex/inference/serving/schemas/face_recognition.py +124 -0
- paddlex/inference/serving/schemas/formula_recognition.py +56 -0
- paddlex/inference/serving/schemas/human_keypoint_detection.py +55 -0
- paddlex/inference/serving/schemas/image_classification.py +45 -0
- paddlex/inference/serving/schemas/image_multilabel_classification.py +47 -0
- paddlex/inference/serving/schemas/instance_segmentation.py +53 -0
- paddlex/inference/serving/schemas/layout_parsing.py +71 -0
- paddlex/inference/serving/schemas/m_3d_bev_detection.py +48 -0
- paddlex/inference/serving/schemas/multilingual_speech_recognition.py +57 -0
- paddlex/inference/serving/schemas/object_detection.py +52 -0
- paddlex/inference/serving/schemas/ocr.py +60 -0
- paddlex/inference/serving/schemas/open_vocabulary_detection.py +52 -0
- paddlex/inference/serving/schemas/open_vocabulary_segmentation.py +52 -0
- paddlex/inference/serving/schemas/pedestrian_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/pp_chatocrv3_doc.py +133 -0
- paddlex/inference/serving/schemas/pp_chatocrv4_doc.py +150 -0
- paddlex/inference/serving/schemas/pp_shituv2.py +124 -0
- paddlex/inference/serving/schemas/pp_structurev3.py +88 -0
- paddlex/inference/serving/schemas/rotated_object_detection.py +52 -0
- paddlex/inference/serving/schemas/seal_recognition.py +62 -0
- paddlex/inference/serving/schemas/semantic_segmentation.py +45 -0
- paddlex/inference/serving/schemas/shared/__init__.py +13 -0
- paddlex/inference/serving/schemas/shared/classification.py +23 -0
- paddlex/inference/serving/schemas/shared/image_segmentation.py +28 -0
- paddlex/inference/serving/schemas/shared/object_detection.py +24 -0
- paddlex/inference/serving/schemas/shared/ocr.py +25 -0
- paddlex/inference/serving/schemas/small_object_detection.py +52 -0
- paddlex/inference/serving/schemas/table_recognition.py +64 -0
- paddlex/inference/serving/schemas/table_recognition_v2.py +69 -0
- paddlex/inference/serving/schemas/ts_anomaly_detection.py +37 -0
- paddlex/inference/serving/schemas/ts_classification.py +38 -0
- paddlex/inference/serving/schemas/ts_forecast.py +37 -0
- paddlex/inference/serving/schemas/vehicle_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/video_classification.py +44 -0
- paddlex/inference/serving/schemas/video_detection.py +56 -0
- paddlex/inference/utils/__init__.py +13 -0
- paddlex/inference/utils/benchmark.py +379 -0
- paddlex/inference/utils/color_map.py +123 -0
- paddlex/inference/utils/get_pipeline_path.py +27 -0
- paddlex/inference/utils/hpi.py +254 -0
- paddlex/inference/utils/hpi_model_info_collection.json +2331 -0
- paddlex/inference/utils/io/__init__.py +36 -0
- paddlex/inference/utils/io/readers.py +504 -0
- paddlex/inference/utils/io/style.py +381 -0
- paddlex/inference/utils/io/tablepyxl.py +157 -0
- paddlex/inference/utils/io/writers.py +458 -0
- paddlex/inference/utils/model_paths.py +48 -0
- paddlex/inference/utils/new_ir_blocklist.py +27 -0
- paddlex/inference/utils/official_models.py +367 -0
- paddlex/inference/utils/pp_option.py +339 -0
- paddlex/inference/utils/trt_blocklist.py +43 -0
- paddlex/inference/utils/trt_config.py +420 -0
- paddlex/model.py +131 -0
- paddlex/modules/__init__.py +115 -0
- paddlex/modules/anomaly_detection/__init__.py +18 -0
- paddlex/modules/anomaly_detection/dataset_checker/__init__.py +94 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +233 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/visualizer.py +76 -0
- paddlex/modules/anomaly_detection/evaluator.py +58 -0
- paddlex/modules/anomaly_detection/exportor.py +22 -0
- paddlex/modules/anomaly_detection/model_list.py +16 -0
- paddlex/modules/anomaly_detection/trainer.py +70 -0
- paddlex/modules/base/__init__.py +18 -0
- paddlex/modules/base/build_model.py +33 -0
- paddlex/modules/base/dataset_checker/__init__.py +16 -0
- paddlex/modules/base/dataset_checker/dataset_checker.py +169 -0
- paddlex/modules/base/dataset_checker/utils.py +108 -0
- paddlex/modules/base/evaluator.py +170 -0
- paddlex/modules/base/exportor.py +145 -0
- paddlex/modules/base/trainer.py +144 -0
- paddlex/modules/base/utils/__init__.py +13 -0
- paddlex/modules/base/utils/cinn_setting.py +89 -0
- paddlex/modules/base/utils/coco_eval.py +94 -0
- paddlex/modules/base/utils/topk_eval.py +118 -0
- paddlex/modules/doc_vlm/__init__.py +18 -0
- paddlex/modules/doc_vlm/dataset_checker.py +29 -0
- paddlex/modules/doc_vlm/evaluator.py +29 -0
- paddlex/modules/doc_vlm/exportor.py +29 -0
- paddlex/modules/doc_vlm/model_list.py +16 -0
- paddlex/modules/doc_vlm/trainer.py +41 -0
- paddlex/modules/face_recognition/__init__.py +18 -0
- paddlex/modules/face_recognition/dataset_checker/__init__.py +71 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/__init__.py +16 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/check_dataset.py +172 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/face_recognition/evaluator.py +52 -0
- paddlex/modules/face_recognition/exportor.py +22 -0
- paddlex/modules/face_recognition/model_list.py +15 -0
- paddlex/modules/face_recognition/trainer.py +75 -0
- paddlex/modules/formula_recognition/__init__.py +18 -0
- paddlex/modules/formula_recognition/dataset_checker/__init__.py +113 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/analyse_dataset.py +158 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/check_dataset.py +76 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/formula_recognition/evaluator.py +80 -0
- paddlex/modules/formula_recognition/exportor.py +22 -0
- paddlex/modules/formula_recognition/model_list.py +23 -0
- paddlex/modules/formula_recognition/trainer.py +123 -0
- paddlex/modules/general_recognition/__init__.py +18 -0
- paddlex/modules/general_recognition/dataset_checker/__init__.py +107 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/analyse_dataset.py +96 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/check_dataset.py +99 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/convert_dataset.py +100 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/visualizer.py +147 -0
- paddlex/modules/general_recognition/evaluator.py +31 -0
- paddlex/modules/general_recognition/exportor.py +22 -0
- paddlex/modules/general_recognition/model_list.py +19 -0
- paddlex/modules/general_recognition/trainer.py +52 -0
- paddlex/modules/image_classification/__init__.py +18 -0
- paddlex/modules/image_classification/dataset_checker/__init__.py +104 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/analyse_dataset.py +92 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/convert_dataset.py +51 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/image_classification/evaluator.py +43 -0
- paddlex/modules/image_classification/exportor.py +22 -0
- paddlex/modules/image_classification/model_list.py +99 -0
- paddlex/modules/image_classification/trainer.py +82 -0
- paddlex/modules/image_unwarping/__init__.py +13 -0
- paddlex/modules/image_unwarping/model_list.py +17 -0
- paddlex/modules/instance_segmentation/__init__.py +18 -0
- paddlex/modules/instance_segmentation/dataset_checker/__init__.py +107 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/check_dataset.py +95 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/convert_dataset.py +241 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/split_dataset.py +122 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/visualizer.py +223 -0
- paddlex/modules/instance_segmentation/evaluator.py +32 -0
- paddlex/modules/instance_segmentation/exportor.py +22 -0
- paddlex/modules/instance_segmentation/model_list.py +33 -0
- paddlex/modules/instance_segmentation/trainer.py +31 -0
- paddlex/modules/keypoint_detection/__init__.py +18 -0
- paddlex/modules/keypoint_detection/dataset_checker/__init__.py +56 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/__init__.py +15 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/visualizer.py +124 -0
- paddlex/modules/keypoint_detection/evaluator.py +41 -0
- paddlex/modules/keypoint_detection/exportor.py +22 -0
- paddlex/modules/keypoint_detection/model_list.py +16 -0
- paddlex/modules/keypoint_detection/trainer.py +39 -0
- paddlex/modules/m_3d_bev_detection/__init__.py +18 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/__init__.py +95 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/analyse_dataset.py +106 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/check_dataset.py +101 -0
- paddlex/modules/m_3d_bev_detection/evaluator.py +46 -0
- paddlex/modules/m_3d_bev_detection/exportor.py +22 -0
- paddlex/modules/m_3d_bev_detection/model_list.py +18 -0
- paddlex/modules/m_3d_bev_detection/trainer.py +68 -0
- paddlex/modules/multilabel_classification/__init__.py +18 -0
- paddlex/modules/multilabel_classification/dataset_checker/__init__.py +106 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/analyse_dataset.py +94 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/convert_dataset.py +120 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/visualizer.py +149 -0
- paddlex/modules/multilabel_classification/evaluator.py +43 -0
- paddlex/modules/multilabel_classification/exportor.py +22 -0
- paddlex/modules/multilabel_classification/model_list.py +24 -0
- paddlex/modules/multilabel_classification/trainer.py +85 -0
- paddlex/modules/multilingual_speech_recognition/__init__.py +18 -0
- paddlex/modules/multilingual_speech_recognition/dataset_checker.py +27 -0
- paddlex/modules/multilingual_speech_recognition/evaluator.py +27 -0
- paddlex/modules/multilingual_speech_recognition/exportor.py +27 -0
- paddlex/modules/multilingual_speech_recognition/model_list.py +22 -0
- paddlex/modules/multilingual_speech_recognition/trainer.py +42 -0
- paddlex/modules/object_detection/__init__.py +18 -0
- paddlex/modules/object_detection/dataset_checker/__init__.py +106 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/convert_dataset.py +438 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/split_dataset.py +123 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/visualizer.py +193 -0
- paddlex/modules/object_detection/evaluator.py +57 -0
- paddlex/modules/object_detection/exportor.py +22 -0
- paddlex/modules/object_detection/model_list.py +86 -0
- paddlex/modules/object_detection/trainer.py +98 -0
- paddlex/modules/open_vocabulary_detection/__init__.py +18 -0
- paddlex/modules/open_vocabulary_detection/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_detection/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_detection/exportor.py +29 -0
- paddlex/modules/open_vocabulary_detection/model_list.py +16 -0
- paddlex/modules/open_vocabulary_detection/trainer.py +44 -0
- paddlex/modules/open_vocabulary_segmentation/__init__.py +18 -0
- paddlex/modules/open_vocabulary_segmentation/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/exportor.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/model_list.py +19 -0
- paddlex/modules/open_vocabulary_segmentation/trainer.py +44 -0
- paddlex/modules/semantic_segmentation/__init__.py +18 -0
- paddlex/modules/semantic_segmentation/dataset_checker/__init__.py +109 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/analyse_dataset.py +76 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/check_dataset.py +80 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/convert_dataset.py +165 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/visualizer.py +75 -0
- paddlex/modules/semantic_segmentation/evaluator.py +58 -0
- paddlex/modules/semantic_segmentation/exportor.py +31 -0
- paddlex/modules/semantic_segmentation/model_list.py +37 -0
- paddlex/modules/semantic_segmentation/trainer.py +72 -0
- paddlex/modules/table_recognition/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/__init__.py +98 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/analyse_dataset.py +59 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py +87 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/table_recognition/evaluator.py +43 -0
- paddlex/modules/table_recognition/exportor.py +22 -0
- paddlex/modules/table_recognition/model_list.py +21 -0
- paddlex/modules/table_recognition/trainer.py +67 -0
- paddlex/modules/text_detection/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/__init__.py +107 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/analyse_dataset.py +220 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py +106 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/split_dataset.py +140 -0
- paddlex/modules/text_detection/evaluator.py +41 -0
- paddlex/modules/text_detection/exportor.py +22 -0
- paddlex/modules/text_detection/model_list.py +26 -0
- paddlex/modules/text_detection/trainer.py +65 -0
- paddlex/modules/text_recognition/__init__.py +18 -0
- paddlex/modules/text_recognition/dataset_checker/__init__.py +125 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/analyse_dataset.py +162 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py +104 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/text_recognition/evaluator.py +64 -0
- paddlex/modules/text_recognition/exportor.py +22 -0
- paddlex/modules/text_recognition/model_list.py +36 -0
- paddlex/modules/text_recognition/trainer.py +105 -0
- paddlex/modules/ts_anomaly_detection/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_anomaly_detection/evaluator.py +67 -0
- paddlex/modules/ts_anomaly_detection/exportor.py +44 -0
- paddlex/modules/ts_anomaly_detection/model_list.py +22 -0
- paddlex/modules/ts_anomaly_detection/trainer.py +113 -0
- paddlex/modules/ts_classification/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/analyse_dataset.py +77 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/split_dataset.py +88 -0
- paddlex/modules/ts_classification/evaluator.py +66 -0
- paddlex/modules/ts_classification/exportor.py +44 -0
- paddlex/modules/ts_classification/model_list.py +18 -0
- paddlex/modules/ts_classification/trainer.py +108 -0
- paddlex/modules/ts_forecast/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/convert_dataset.py +73 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_forecast/evaluator.py +66 -0
- paddlex/modules/ts_forecast/exportor.py +44 -0
- paddlex/modules/ts_forecast/model_list.py +24 -0
- paddlex/modules/ts_forecast/trainer.py +108 -0
- paddlex/modules/video_classification/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/__init__.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/analyse_dataset.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/check_dataset.py +120 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/video_classification/evaluator.py +44 -0
- paddlex/modules/video_classification/exportor.py +22 -0
- paddlex/modules/video_classification/model_list.py +19 -0
- paddlex/modules/video_classification/trainer.py +88 -0
- paddlex/modules/video_detection/__init__.py +18 -0
- paddlex/modules/video_detection/dataset_checker/__init__.py +86 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/analyse_dataset.py +100 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/video_detection/evaluator.py +42 -0
- paddlex/modules/video_detection/exportor.py +22 -0
- paddlex/modules/video_detection/model_list.py +15 -0
- paddlex/modules/video_detection/trainer.py +82 -0
- paddlex/ops/__init__.py +152 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.cpp +266 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.h +28 -0
- paddlex/ops/iou3d_nms/iou3d_nms.cpp +206 -0
- paddlex/ops/iou3d_nms/iou3d_nms.h +35 -0
- paddlex/ops/iou3d_nms/iou3d_nms_api.cpp +114 -0
- paddlex/ops/iou3d_nms/iou3d_nms_kernel.cu +484 -0
- paddlex/ops/setup.py +37 -0
- paddlex/ops/voxel/voxelize_op.cc +194 -0
- paddlex/ops/voxel/voxelize_op.cu +346 -0
- paddlex/paddlex_cli.py +476 -0
- paddlex/repo_apis/Paddle3D_api/__init__.py +17 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/__init__.py +18 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/config.py +118 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py +238 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/register.py +55 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/runner.py +104 -0
- paddlex/repo_apis/Paddle3D_api/pp3d_config.py +145 -0
- paddlex/repo_apis/PaddleClas_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleClas_api/cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleClas_api/cls/config.py +595 -0
- paddlex/repo_apis/PaddleClas_api/cls/model.py +355 -0
- paddlex/repo_apis/PaddleClas_api/cls/register.py +907 -0
- paddlex/repo_apis/PaddleClas_api/cls/runner.py +218 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/__init__.py +18 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/config.py +141 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/model.py +20 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/register.py +68 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/runner.py +50 -0
- paddlex/repo_apis/PaddleDetection_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleDetection_api/config_helper.py +280 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/__init__.py +18 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/config.py +457 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py +403 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/register.py +262 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/runner.py +225 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/config.py +540 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/model.py +429 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py +245 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/register.py +1135 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/runner.py +225 -0
- paddlex/repo_apis/PaddleNLP_api/__init__.py +13 -0
- paddlex/repo_apis/PaddleOCR_api/__init__.py +22 -0
- paddlex/repo_apis/PaddleOCR_api/config_utils.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/config.py +571 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/register.py +99 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/config.py +64 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/model.py +126 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/register.py +70 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/runner.py +51 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/config.py +62 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/model.py +72 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/register.py +107 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/runner.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/config.py +564 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/register.py +216 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleSeg_api/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/base_seg_config.py +134 -0
- paddlex/repo_apis/PaddleSeg_api/seg/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/seg/config.py +183 -0
- paddlex/repo_apis/PaddleSeg_api/seg/model.py +491 -0
- paddlex/repo_apis/PaddleSeg_api/seg/register.py +272 -0
- paddlex/repo_apis/PaddleSeg_api/seg/runner.py +261 -0
- paddlex/repo_apis/PaddleTS_api/__init__.py +20 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/config.py +88 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/register.py +146 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/__init__.py +13 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/config.py +244 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/model.py +276 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/config.py +72 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/register.py +59 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/config.py +136 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/register.py +186 -0
- paddlex/repo_apis/PaddleVideo_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleVideo_api/config_utils.py +51 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/config.py +548 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/model.py +346 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/register.py +70 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/runner.py +204 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/config.py +549 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/model.py +298 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/register.py +44 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/runner.py +199 -0
- paddlex/repo_apis/__init__.py +13 -0
- paddlex/repo_apis/base/__init__.py +22 -0
- paddlex/repo_apis/base/config.py +237 -0
- paddlex/repo_apis/base/model.py +563 -0
- paddlex/repo_apis/base/register.py +135 -0
- paddlex/repo_apis/base/runner.py +390 -0
- paddlex/repo_apis/base/utils/__init__.py +13 -0
- paddlex/repo_apis/base/utils/arg.py +64 -0
- paddlex/repo_apis/base/utils/subprocess.py +107 -0
- paddlex/repo_manager/__init__.py +17 -0
- paddlex/repo_manager/core.py +253 -0
- paddlex/repo_manager/meta.py +180 -0
- paddlex/repo_manager/repo.py +425 -0
- paddlex/repo_manager/utils.py +148 -0
- paddlex/utils/__init__.py +1 -12
- paddlex/utils/cache.py +146 -0
- paddlex/utils/config.py +216 -0
- paddlex/utils/custom_device_list.py +311 -0
- paddlex/utils/deps.py +249 -0
- paddlex/utils/device.py +195 -0
- paddlex/utils/download.py +168 -182
- paddlex/utils/env.py +32 -45
- paddlex/utils/errors/__init__.py +17 -0
- paddlex/utils/errors/dataset_checker.py +78 -0
- paddlex/utils/errors/others.py +138 -0
- paddlex/utils/file_interface.py +211 -0
- paddlex/utils/flags.py +70 -0
- paddlex/utils/fonts/__init__.py +97 -0
- paddlex/utils/func_register.py +41 -0
- paddlex/utils/install.py +87 -0
- paddlex/utils/interactive_get_pipeline.py +55 -0
- paddlex/utils/lazy_loader.py +68 -0
- paddlex/utils/logging.py +140 -33
- paddlex/utils/misc.py +201 -0
- paddlex/utils/pipeline_arguments.py +719 -0
- paddlex/utils/result_saver.py +58 -0
- paddlex/utils/subclass_register.py +99 -0
- paddlex/version.py +55 -0
- paddlex-3.0.0.dist-info/METADATA +1168 -0
- paddlex-3.0.0.dist-info/RECORD +1093 -0
- paddlex-3.0.0.dist-info/WHEEL +5 -0
- paddlex-3.0.0.dist-info/entry_points.txt +2 -0
- paddlex-3.0.0.dist-info/licenses/LICENSE +169 -0
- paddlex-3.0.0.dist-info/top_level.txt +1 -0
- PaddleClas/__init__.py +0 -16
- PaddleClas/deploy/__init__.py +0 -1
- PaddleClas/deploy/paddleserving/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/classification_web_service.py +0 -74
- PaddleClas/deploy/paddleserving/cpu_utilization.py +0 -4
- PaddleClas/deploy/paddleserving/pipeline_http_client.py +0 -20
- PaddleClas/deploy/paddleserving/pipeline_rpc_client.py +0 -33
- PaddleClas/deploy/paddleserving/recognition/__init__.py +0 -0
- PaddleClas/deploy/paddleserving/recognition/pipeline_http_client.py +0 -21
- PaddleClas/deploy/paddleserving/recognition/pipeline_rpc_client.py +0 -34
- PaddleClas/deploy/paddleserving/recognition/recognition_web_service.py +0 -209
- PaddleClas/deploy/python/__init__.py +0 -0
- PaddleClas/deploy/python/build_gallery.py +0 -214
- PaddleClas/deploy/python/det_preprocess.py +0 -205
- PaddleClas/deploy/python/postprocess.py +0 -161
- PaddleClas/deploy/python/predict_cls.py +0 -142
- PaddleClas/deploy/python/predict_det.py +0 -158
- PaddleClas/deploy/python/predict_rec.py +0 -138
- PaddleClas/deploy/python/predict_system.py +0 -144
- PaddleClas/deploy/python/preprocess.py +0 -337
- PaddleClas/deploy/utils/__init__.py +0 -5
- PaddleClas/deploy/utils/config.py +0 -197
- PaddleClas/deploy/utils/draw_bbox.py +0 -61
- PaddleClas/deploy/utils/encode_decode.py +0 -31
- PaddleClas/deploy/utils/get_image_list.py +0 -49
- PaddleClas/deploy/utils/logger.py +0 -120
- PaddleClas/deploy/utils/predictor.py +0 -71
- PaddleClas/deploy/vector_search/__init__.py +0 -1
- PaddleClas/deploy/vector_search/interface.py +0 -272
- PaddleClas/deploy/vector_search/test.py +0 -34
- PaddleClas/hubconf.py +0 -788
- PaddleClas/paddleclas.py +0 -552
- PaddleClas/ppcls/__init__.py +0 -20
- PaddleClas/ppcls/arch/__init__.py +0 -127
- PaddleClas/ppcls/arch/backbone/__init__.py +0 -80
- PaddleClas/ppcls/arch/backbone/base/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/base/theseus_layer.py +0 -126
- PaddleClas/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- PaddleClas/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- PaddleClas/ppcls/arch/backbone/legendary_models/hrnet.py +0 -744
- PaddleClas/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- PaddleClas/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- PaddleClas/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- PaddleClas/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- PaddleClas/ppcls/arch/backbone/legendary_models/vgg.py +0 -231
- PaddleClas/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- PaddleClas/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- PaddleClas/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- PaddleClas/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- PaddleClas/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- PaddleClas/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- PaddleClas/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- PaddleClas/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- PaddleClas/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- PaddleClas/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- PaddleClas/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- PaddleClas/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- PaddleClas/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- PaddleClas/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- PaddleClas/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- PaddleClas/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- PaddleClas/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- PaddleClas/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- PaddleClas/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- PaddleClas/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- PaddleClas/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- PaddleClas/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- PaddleClas/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- PaddleClas/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- PaddleClas/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- PaddleClas/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- PaddleClas/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- PaddleClas/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- PaddleClas/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- PaddleClas/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- PaddleClas/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- PaddleClas/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- PaddleClas/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- PaddleClas/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- PaddleClas/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- PaddleClas/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- PaddleClas/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- PaddleClas/ppcls/arch/gears/__init__.py +0 -32
- PaddleClas/ppcls/arch/gears/arcmargin.py +0 -72
- PaddleClas/ppcls/arch/gears/circlemargin.py +0 -59
- PaddleClas/ppcls/arch/gears/cosmargin.py +0 -55
- PaddleClas/ppcls/arch/gears/fc.py +0 -35
- PaddleClas/ppcls/arch/gears/identity_head.py +0 -9
- PaddleClas/ppcls/arch/gears/vehicle_neck.py +0 -52
- PaddleClas/ppcls/arch/utils.py +0 -53
- PaddleClas/ppcls/data/__init__.py +0 -144
- PaddleClas/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- PaddleClas/ppcls/data/dataloader/__init__.py +0 -9
- PaddleClas/ppcls/data/dataloader/common_dataset.py +0 -84
- PaddleClas/ppcls/data/dataloader/dali.py +0 -319
- PaddleClas/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- PaddleClas/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- PaddleClas/ppcls/data/dataloader/logo_dataset.py +0 -46
- PaddleClas/ppcls/data/dataloader/mix_dataset.py +0 -49
- PaddleClas/ppcls/data/dataloader/mix_sampler.py +0 -79
- PaddleClas/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- PaddleClas/ppcls/data/dataloader/pk_sampler.py +0 -105
- PaddleClas/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- PaddleClas/ppcls/data/postprocess/__init__.py +0 -41
- PaddleClas/ppcls/data/postprocess/topk.py +0 -85
- PaddleClas/ppcls/data/preprocess/__init__.py +0 -100
- PaddleClas/ppcls/data/preprocess/batch_ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- PaddleClas/ppcls/data/preprocess/ops/__init__.py +0 -1
- PaddleClas/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- PaddleClas/ppcls/data/preprocess/ops/cutout.py +0 -41
- PaddleClas/ppcls/data/preprocess/ops/fmix.py +0 -217
- PaddleClas/ppcls/data/preprocess/ops/functional.py +0 -138
- PaddleClas/ppcls/data/preprocess/ops/grid.py +0 -89
- PaddleClas/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- PaddleClas/ppcls/data/preprocess/ops/operators.py +0 -384
- PaddleClas/ppcls/data/preprocess/ops/randaugment.py +0 -106
- PaddleClas/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- PaddleClas/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- PaddleClas/ppcls/data/utils/__init__.py +0 -13
- PaddleClas/ppcls/data/utils/get_image_list.py +0 -49
- PaddleClas/ppcls/engine/__init__.py +0 -0
- PaddleClas/ppcls/engine/engine.py +0 -436
- PaddleClas/ppcls/engine/evaluation/__init__.py +0 -16
- PaddleClas/ppcls/engine/evaluation/classification.py +0 -143
- PaddleClas/ppcls/engine/evaluation/retrieval.py +0 -169
- PaddleClas/ppcls/engine/slim/__init__.py +0 -16
- PaddleClas/ppcls/engine/slim/prune.py +0 -66
- PaddleClas/ppcls/engine/slim/quant.py +0 -55
- PaddleClas/ppcls/engine/train/__init__.py +0 -14
- PaddleClas/ppcls/engine/train/train.py +0 -79
- PaddleClas/ppcls/engine/train/utils.py +0 -72
- PaddleClas/ppcls/loss/__init__.py +0 -65
- PaddleClas/ppcls/loss/celoss.py +0 -67
- PaddleClas/ppcls/loss/centerloss.py +0 -54
- PaddleClas/ppcls/loss/comfunc.py +0 -45
- PaddleClas/ppcls/loss/deephashloss.py +0 -92
- PaddleClas/ppcls/loss/distanceloss.py +0 -43
- PaddleClas/ppcls/loss/distillationloss.py +0 -141
- PaddleClas/ppcls/loss/dmlloss.py +0 -46
- PaddleClas/ppcls/loss/emlloss.py +0 -97
- PaddleClas/ppcls/loss/googlenetloss.py +0 -41
- PaddleClas/ppcls/loss/msmloss.py +0 -78
- PaddleClas/ppcls/loss/multilabelloss.py +0 -43
- PaddleClas/ppcls/loss/npairsloss.py +0 -38
- PaddleClas/ppcls/loss/pairwisecosface.py +0 -55
- PaddleClas/ppcls/loss/supconloss.py +0 -108
- PaddleClas/ppcls/loss/trihardloss.py +0 -82
- PaddleClas/ppcls/loss/triplet.py +0 -137
- PaddleClas/ppcls/metric/__init__.py +0 -51
- PaddleClas/ppcls/metric/metrics.py +0 -308
- PaddleClas/ppcls/optimizer/__init__.py +0 -72
- PaddleClas/ppcls/optimizer/learning_rate.py +0 -326
- PaddleClas/ppcls/optimizer/optimizer.py +0 -207
- PaddleClas/ppcls/utils/__init__.py +0 -27
- PaddleClas/ppcls/utils/check.py +0 -151
- PaddleClas/ppcls/utils/config.py +0 -210
- PaddleClas/ppcls/utils/download.py +0 -319
- PaddleClas/ppcls/utils/ema.py +0 -63
- PaddleClas/ppcls/utils/logger.py +0 -137
- PaddleClas/ppcls/utils/metrics.py +0 -107
- PaddleClas/ppcls/utils/misc.py +0 -63
- PaddleClas/ppcls/utils/model_zoo.py +0 -213
- PaddleClas/ppcls/utils/profiler.py +0 -111
- PaddleClas/ppcls/utils/save_load.py +0 -136
- PaddleClas/setup.py +0 -58
- PaddleClas/tools/__init__.py +0 -15
- PaddleClas/tools/eval.py +0 -31
- PaddleClas/tools/export_model.py +0 -34
- PaddleClas/tools/infer.py +0 -31
- PaddleClas/tools/train.py +0 -32
- paddlex/cls.py +0 -82
- paddlex/command.py +0 -215
- paddlex/cv/__init__.py +0 -17
- paddlex/cv/datasets/__init__.py +0 -18
- paddlex/cv/datasets/coco.py +0 -208
- paddlex/cv/datasets/imagenet.py +0 -88
- paddlex/cv/datasets/seg_dataset.py +0 -91
- paddlex/cv/datasets/voc.py +0 -445
- paddlex/cv/models/__init__.py +0 -18
- paddlex/cv/models/base.py +0 -631
- paddlex/cv/models/classifier.py +0 -989
- paddlex/cv/models/detector.py +0 -2292
- paddlex/cv/models/load_model.py +0 -148
- paddlex/cv/models/segmenter.py +0 -768
- paddlex/cv/models/slim/__init__.py +0 -13
- paddlex/cv/models/slim/prune.py +0 -55
- paddlex/cv/models/utils/__init__.py +0 -13
- paddlex/cv/models/utils/det_metrics/__init__.py +0 -15
- paddlex/cv/models/utils/det_metrics/coco_utils.py +0 -476
- paddlex/cv/models/utils/det_metrics/metrics.py +0 -220
- paddlex/cv/models/utils/infer_nets.py +0 -45
- paddlex/cv/models/utils/seg_metrics.py +0 -62
- paddlex/cv/models/utils/visualize.py +0 -399
- paddlex/cv/transforms/__init__.py +0 -46
- paddlex/cv/transforms/batch_operators.py +0 -286
- paddlex/cv/transforms/box_utils.py +0 -41
- paddlex/cv/transforms/functions.py +0 -193
- paddlex/cv/transforms/operators.py +0 -1402
- paddlex/deploy.py +0 -268
- paddlex/det.py +0 -49
- paddlex/paddleseg/__init__.py +0 -17
- paddlex/paddleseg/core/__init__.py +0 -20
- paddlex/paddleseg/core/infer.py +0 -289
- paddlex/paddleseg/core/predict.py +0 -145
- paddlex/paddleseg/core/train.py +0 -258
- paddlex/paddleseg/core/val.py +0 -172
- paddlex/paddleseg/cvlibs/__init__.py +0 -17
- paddlex/paddleseg/cvlibs/callbacks.py +0 -279
- paddlex/paddleseg/cvlibs/config.py +0 -359
- paddlex/paddleseg/cvlibs/manager.py +0 -142
- paddlex/paddleseg/cvlibs/param_init.py +0 -91
- paddlex/paddleseg/datasets/__init__.py +0 -21
- paddlex/paddleseg/datasets/ade.py +0 -112
- paddlex/paddleseg/datasets/cityscapes.py +0 -86
- paddlex/paddleseg/datasets/cocostuff.py +0 -79
- paddlex/paddleseg/datasets/dataset.py +0 -164
- paddlex/paddleseg/datasets/mini_deep_globe_road_extraction.py +0 -95
- paddlex/paddleseg/datasets/optic_disc_seg.py +0 -97
- paddlex/paddleseg/datasets/pascal_context.py +0 -80
- paddlex/paddleseg/datasets/voc.py +0 -113
- paddlex/paddleseg/models/__init__.py +0 -39
- paddlex/paddleseg/models/ann.py +0 -436
- paddlex/paddleseg/models/attention_unet.py +0 -189
- paddlex/paddleseg/models/backbones/__init__.py +0 -18
- paddlex/paddleseg/models/backbones/hrnet.py +0 -815
- paddlex/paddleseg/models/backbones/mobilenetv3.py +0 -365
- paddlex/paddleseg/models/backbones/resnet_vd.py +0 -364
- paddlex/paddleseg/models/backbones/xception_deeplab.py +0 -415
- paddlex/paddleseg/models/bisenet.py +0 -311
- paddlex/paddleseg/models/danet.py +0 -220
- paddlex/paddleseg/models/decoupled_segnet.py +0 -233
- paddlex/paddleseg/models/deeplab.py +0 -258
- paddlex/paddleseg/models/dnlnet.py +0 -231
- paddlex/paddleseg/models/emanet.py +0 -219
- paddlex/paddleseg/models/fast_scnn.py +0 -318
- paddlex/paddleseg/models/fcn.py +0 -135
- paddlex/paddleseg/models/gcnet.py +0 -223
- paddlex/paddleseg/models/gscnn.py +0 -357
- paddlex/paddleseg/models/hardnet.py +0 -309
- paddlex/paddleseg/models/isanet.py +0 -202
- paddlex/paddleseg/models/layers/__init__.py +0 -19
- paddlex/paddleseg/models/layers/activation.py +0 -73
- paddlex/paddleseg/models/layers/attention.py +0 -146
- paddlex/paddleseg/models/layers/layer_libs.py +0 -168
- paddlex/paddleseg/models/layers/nonlocal2d.py +0 -155
- paddlex/paddleseg/models/layers/pyramid_pool.py +0 -182
- paddlex/paddleseg/models/losses/__init__.py +0 -27
- paddlex/paddleseg/models/losses/binary_cross_entropy_loss.py +0 -174
- paddlex/paddleseg/models/losses/bootstrapped_cross_entropy.py +0 -73
- paddlex/paddleseg/models/losses/cross_entropy_loss.py +0 -94
- paddlex/paddleseg/models/losses/decoupledsegnet_relax_boundary_loss.py +0 -129
- paddlex/paddleseg/models/losses/dice_loss.py +0 -61
- paddlex/paddleseg/models/losses/edge_attention_loss.py +0 -78
- paddlex/paddleseg/models/losses/gscnn_dual_task_loss.py +0 -141
- paddlex/paddleseg/models/losses/l1_loss.py +0 -76
- paddlex/paddleseg/models/losses/lovasz_loss.py +0 -222
- paddlex/paddleseg/models/losses/mean_square_error_loss.py +0 -65
- paddlex/paddleseg/models/losses/mixed_loss.py +0 -58
- paddlex/paddleseg/models/losses/ohem_cross_entropy_loss.py +0 -99
- paddlex/paddleseg/models/losses/ohem_edge_attention_loss.py +0 -114
- paddlex/paddleseg/models/ocrnet.py +0 -248
- paddlex/paddleseg/models/pspnet.py +0 -147
- paddlex/paddleseg/models/sfnet.py +0 -236
- paddlex/paddleseg/models/shufflenet_slim.py +0 -268
- paddlex/paddleseg/models/u2net.py +0 -574
- paddlex/paddleseg/models/unet.py +0 -155
- paddlex/paddleseg/models/unet_3plus.py +0 -316
- paddlex/paddleseg/models/unet_plusplus.py +0 -237
- paddlex/paddleseg/transforms/__init__.py +0 -16
- paddlex/paddleseg/transforms/functional.py +0 -161
- paddlex/paddleseg/transforms/transforms.py +0 -937
- paddlex/paddleseg/utils/__init__.py +0 -22
- paddlex/paddleseg/utils/config_check.py +0 -60
- paddlex/paddleseg/utils/download.py +0 -163
- paddlex/paddleseg/utils/env/__init__.py +0 -16
- paddlex/paddleseg/utils/env/seg_env.py +0 -56
- paddlex/paddleseg/utils/env/sys_env.py +0 -122
- paddlex/paddleseg/utils/logger.py +0 -48
- paddlex/paddleseg/utils/metrics.py +0 -146
- paddlex/paddleseg/utils/progbar.py +0 -212
- paddlex/paddleseg/utils/timer.py +0 -53
- paddlex/paddleseg/utils/utils.py +0 -120
- paddlex/paddleseg/utils/visualize.py +0 -90
- paddlex/ppcls/__init__.py +0 -20
- paddlex/ppcls/arch/__init__.py +0 -127
- paddlex/ppcls/arch/backbone/__init__.py +0 -80
- paddlex/ppcls/arch/backbone/base/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/base/theseus_layer.py +0 -130
- paddlex/ppcls/arch/backbone/legendary_models/__init__.py +0 -6
- paddlex/ppcls/arch/backbone/legendary_models/esnet.py +0 -355
- paddlex/ppcls/arch/backbone/legendary_models/hrnet.py +0 -748
- paddlex/ppcls/arch/backbone/legendary_models/inception_v3.py +0 -539
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +0 -234
- paddlex/ppcls/arch/backbone/legendary_models/mobilenet_v3.py +0 -561
- paddlex/ppcls/arch/backbone/legendary_models/pp_lcnet.py +0 -399
- paddlex/ppcls/arch/backbone/legendary_models/resnet.py +0 -534
- paddlex/ppcls/arch/backbone/legendary_models/vgg.py +0 -235
- paddlex/ppcls/arch/backbone/model_zoo/__init__.py +0 -0
- paddlex/ppcls/arch/backbone/model_zoo/alexnet.py +0 -168
- paddlex/ppcls/arch/backbone/model_zoo/cspnet.py +0 -376
- paddlex/ppcls/arch/backbone/model_zoo/darknet.py +0 -197
- paddlex/ppcls/arch/backbone/model_zoo/densenet.py +0 -344
- paddlex/ppcls/arch/backbone/model_zoo/distilled_vision_transformer.py +0 -272
- paddlex/ppcls/arch/backbone/model_zoo/dla.py +0 -528
- paddlex/ppcls/arch/backbone/model_zoo/dpn.py +0 -451
- paddlex/ppcls/arch/backbone/model_zoo/efficientnet.py +0 -976
- paddlex/ppcls/arch/backbone/model_zoo/ghostnet.py +0 -363
- paddlex/ppcls/arch/backbone/model_zoo/googlenet.py +0 -229
- paddlex/ppcls/arch/backbone/model_zoo/gvt.py +0 -693
- paddlex/ppcls/arch/backbone/model_zoo/hardnet.py +0 -293
- paddlex/ppcls/arch/backbone/model_zoo/inception_v4.py +0 -477
- paddlex/ppcls/arch/backbone/model_zoo/levit.py +0 -589
- paddlex/ppcls/arch/backbone/model_zoo/mixnet.py +0 -815
- paddlex/ppcls/arch/backbone/model_zoo/mobilenet_v2.py +0 -287
- paddlex/ppcls/arch/backbone/model_zoo/rednet.py +0 -203
- paddlex/ppcls/arch/backbone/model_zoo/regnet.py +0 -431
- paddlex/ppcls/arch/backbone/model_zoo/repvgg.py +0 -422
- paddlex/ppcls/arch/backbone/model_zoo/res2net.py +0 -264
- paddlex/ppcls/arch/backbone/model_zoo/res2net_vd.py +0 -305
- paddlex/ppcls/arch/backbone/model_zoo/resnest.py +0 -740
- paddlex/ppcls/arch/backbone/model_zoo/resnet_vc.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/resnext.py +0 -298
- paddlex/ppcls/arch/backbone/model_zoo/resnext101_wsl.py +0 -490
- paddlex/ppcls/arch/backbone/model_zoo/resnext_vd.py +0 -317
- paddlex/ppcls/arch/backbone/model_zoo/rexnet.py +0 -281
- paddlex/ppcls/arch/backbone/model_zoo/se_resnet_vd.py +0 -390
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext.py +0 -364
- paddlex/ppcls/arch/backbone/model_zoo/se_resnext_vd.py +0 -309
- paddlex/ppcls/arch/backbone/model_zoo/shufflenet_v2.py +0 -362
- paddlex/ppcls/arch/backbone/model_zoo/squeezenet.py +0 -194
- paddlex/ppcls/arch/backbone/model_zoo/swin_transformer.py +0 -857
- paddlex/ppcls/arch/backbone/model_zoo/tnt.py +0 -385
- paddlex/ppcls/arch/backbone/model_zoo/vision_transformer.py +0 -495
- paddlex/ppcls/arch/backbone/model_zoo/xception.py +0 -377
- paddlex/ppcls/arch/backbone/model_zoo/xception_deeplab.py +0 -421
- paddlex/ppcls/arch/backbone/variant_models/__init__.py +0 -3
- paddlex/ppcls/arch/backbone/variant_models/pp_lcnet_variant.py +0 -29
- paddlex/ppcls/arch/backbone/variant_models/resnet_variant.py +0 -23
- paddlex/ppcls/arch/backbone/variant_models/vgg_variant.py +0 -28
- paddlex/ppcls/arch/gears/__init__.py +0 -32
- paddlex/ppcls/arch/gears/arcmargin.py +0 -72
- paddlex/ppcls/arch/gears/circlemargin.py +0 -59
- paddlex/ppcls/arch/gears/cosmargin.py +0 -55
- paddlex/ppcls/arch/gears/fc.py +0 -35
- paddlex/ppcls/arch/gears/identity_head.py +0 -9
- paddlex/ppcls/arch/gears/vehicle_neck.py +0 -52
- paddlex/ppcls/arch/utils.py +0 -53
- paddlex/ppcls/data/__init__.py +0 -144
- paddlex/ppcls/data/dataloader/DistributedRandomIdentitySampler.py +0 -90
- paddlex/ppcls/data/dataloader/__init__.py +0 -9
- paddlex/ppcls/data/dataloader/common_dataset.py +0 -84
- paddlex/ppcls/data/dataloader/dali.py +0 -319
- paddlex/ppcls/data/dataloader/icartoon_dataset.py +0 -36
- paddlex/ppcls/data/dataloader/imagenet_dataset.py +0 -38
- paddlex/ppcls/data/dataloader/logo_dataset.py +0 -46
- paddlex/ppcls/data/dataloader/mix_dataset.py +0 -49
- paddlex/ppcls/data/dataloader/mix_sampler.py +0 -79
- paddlex/ppcls/data/dataloader/multilabel_dataset.py +0 -59
- paddlex/ppcls/data/dataloader/pk_sampler.py +0 -105
- paddlex/ppcls/data/dataloader/vehicle_dataset.py +0 -138
- paddlex/ppcls/data/postprocess/__init__.py +0 -41
- paddlex/ppcls/data/postprocess/topk.py +0 -85
- paddlex/ppcls/data/preprocess/__init__.py +0 -100
- paddlex/ppcls/data/preprocess/batch_ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/batch_ops/batch_operators.py +0 -231
- paddlex/ppcls/data/preprocess/ops/__init__.py +0 -0
- paddlex/ppcls/data/preprocess/ops/autoaugment.py +0 -264
- paddlex/ppcls/data/preprocess/ops/cutout.py +0 -41
- paddlex/ppcls/data/preprocess/ops/fmix.py +0 -217
- paddlex/ppcls/data/preprocess/ops/functional.py +0 -141
- paddlex/ppcls/data/preprocess/ops/grid.py +0 -89
- paddlex/ppcls/data/preprocess/ops/hide_and_seek.py +0 -44
- paddlex/ppcls/data/preprocess/ops/operators.py +0 -384
- paddlex/ppcls/data/preprocess/ops/randaugment.py +0 -106
- paddlex/ppcls/data/preprocess/ops/random_erasing.py +0 -90
- paddlex/ppcls/data/preprocess/ops/timm_autoaugment.py +0 -877
- paddlex/ppcls/data/utils/__init__.py +0 -13
- paddlex/ppcls/data/utils/get_image_list.py +0 -49
- paddlex/ppcls/engine/__init__.py +0 -0
- paddlex/ppcls/engine/engine.py +0 -436
- paddlex/ppcls/engine/evaluation/__init__.py +0 -16
- paddlex/ppcls/engine/evaluation/classification.py +0 -143
- paddlex/ppcls/engine/evaluation/retrieval.py +0 -169
- paddlex/ppcls/engine/slim/__init__.py +0 -16
- paddlex/ppcls/engine/slim/prune.py +0 -66
- paddlex/ppcls/engine/slim/quant.py +0 -55
- paddlex/ppcls/engine/train/__init__.py +0 -14
- paddlex/ppcls/engine/train/train.py +0 -79
- paddlex/ppcls/engine/train/utils.py +0 -72
- paddlex/ppcls/loss/__init__.py +0 -65
- paddlex/ppcls/loss/celoss.py +0 -67
- paddlex/ppcls/loss/centerloss.py +0 -54
- paddlex/ppcls/loss/comfunc.py +0 -45
- paddlex/ppcls/loss/deephashloss.py +0 -96
- paddlex/ppcls/loss/distanceloss.py +0 -43
- paddlex/ppcls/loss/distillationloss.py +0 -141
- paddlex/ppcls/loss/dmlloss.py +0 -46
- paddlex/ppcls/loss/emlloss.py +0 -97
- paddlex/ppcls/loss/googlenetloss.py +0 -42
- paddlex/ppcls/loss/msmloss.py +0 -78
- paddlex/ppcls/loss/multilabelloss.py +0 -43
- paddlex/ppcls/loss/npairsloss.py +0 -38
- paddlex/ppcls/loss/pairwisecosface.py +0 -59
- paddlex/ppcls/loss/supconloss.py +0 -108
- paddlex/ppcls/loss/trihardloss.py +0 -82
- paddlex/ppcls/loss/triplet.py +0 -137
- paddlex/ppcls/metric/__init__.py +0 -51
- paddlex/ppcls/metric/metrics.py +0 -308
- paddlex/ppcls/optimizer/__init__.py +0 -72
- paddlex/ppcls/optimizer/learning_rate.py +0 -326
- paddlex/ppcls/optimizer/optimizer.py +0 -208
- paddlex/ppcls/utils/__init__.py +0 -27
- paddlex/ppcls/utils/check.py +0 -151
- paddlex/ppcls/utils/config.py +0 -210
- paddlex/ppcls/utils/download.py +0 -319
- paddlex/ppcls/utils/ema.py +0 -63
- paddlex/ppcls/utils/logger.py +0 -137
- paddlex/ppcls/utils/metrics.py +0 -112
- paddlex/ppcls/utils/misc.py +0 -63
- paddlex/ppcls/utils/model_zoo.py +0 -213
- paddlex/ppcls/utils/profiler.py +0 -111
- paddlex/ppcls/utils/save_load.py +0 -136
- paddlex/ppdet/__init__.py +0 -16
- paddlex/ppdet/core/__init__.py +0 -15
- paddlex/ppdet/core/config/__init__.py +0 -13
- paddlex/ppdet/core/config/schema.py +0 -248
- paddlex/ppdet/core/config/yaml_helpers.py +0 -118
- paddlex/ppdet/core/workspace.py +0 -278
- paddlex/ppdet/data/__init__.py +0 -21
- paddlex/ppdet/data/crop_utils/__init__.py +0 -13
- paddlex/ppdet/data/crop_utils/annotation_cropper.py +0 -585
- paddlex/ppdet/data/crop_utils/chip_box_utils.py +0 -170
- paddlex/ppdet/data/reader.py +0 -302
- paddlex/ppdet/data/shm_utils.py +0 -67
- paddlex/ppdet/data/source/__init__.py +0 -29
- paddlex/ppdet/data/source/category.py +0 -904
- paddlex/ppdet/data/source/coco.py +0 -251
- paddlex/ppdet/data/source/dataset.py +0 -197
- paddlex/ppdet/data/source/keypoint_coco.py +0 -669
- paddlex/ppdet/data/source/mot.py +0 -636
- paddlex/ppdet/data/source/sniper_coco.py +0 -191
- paddlex/ppdet/data/source/voc.py +0 -231
- paddlex/ppdet/data/source/widerface.py +0 -180
- paddlex/ppdet/data/transform/__init__.py +0 -28
- paddlex/ppdet/data/transform/atss_assigner.py +0 -270
- paddlex/ppdet/data/transform/autoaugment_utils.py +0 -1591
- paddlex/ppdet/data/transform/batch_operators.py +0 -1080
- paddlex/ppdet/data/transform/gridmask_utils.py +0 -86
- paddlex/ppdet/data/transform/keypoint_operators.py +0 -868
- paddlex/ppdet/data/transform/mot_operators.py +0 -628
- paddlex/ppdet/data/transform/op_helper.py +0 -498
- paddlex/ppdet/data/transform/operators.py +0 -3025
- paddlex/ppdet/engine/__init__.py +0 -30
- paddlex/ppdet/engine/callbacks.py +0 -340
- paddlex/ppdet/engine/env.py +0 -50
- paddlex/ppdet/engine/export_utils.py +0 -177
- paddlex/ppdet/engine/tracker.py +0 -538
- paddlex/ppdet/engine/trainer.py +0 -723
- paddlex/ppdet/metrics/__init__.py +0 -29
- paddlex/ppdet/metrics/coco_utils.py +0 -184
- paddlex/ppdet/metrics/json_results.py +0 -149
- paddlex/ppdet/metrics/keypoint_metrics.py +0 -401
- paddlex/ppdet/metrics/map_utils.py +0 -444
- paddlex/ppdet/metrics/mcmot_metrics.py +0 -470
- paddlex/ppdet/metrics/metrics.py +0 -434
- paddlex/ppdet/metrics/mot_metrics.py +0 -1236
- paddlex/ppdet/metrics/munkres.py +0 -428
- paddlex/ppdet/metrics/widerface_utils.py +0 -393
- paddlex/ppdet/model_zoo/__init__.py +0 -18
- paddlex/ppdet/model_zoo/model_zoo.py +0 -84
- paddlex/ppdet/modeling/__init__.py +0 -45
- paddlex/ppdet/modeling/architectures/__init__.py +0 -51
- paddlex/ppdet/modeling/architectures/blazeface.py +0 -91
- paddlex/ppdet/modeling/architectures/cascade_rcnn.py +0 -144
- paddlex/ppdet/modeling/architectures/centernet.py +0 -108
- paddlex/ppdet/modeling/architectures/deepsort.py +0 -69
- paddlex/ppdet/modeling/architectures/detr.py +0 -93
- paddlex/ppdet/modeling/architectures/fairmot.py +0 -100
- paddlex/ppdet/modeling/architectures/faster_rcnn.py +0 -106
- paddlex/ppdet/modeling/architectures/fcos.py +0 -105
- paddlex/ppdet/modeling/architectures/gfl.py +0 -87
- paddlex/ppdet/modeling/architectures/jde.py +0 -111
- paddlex/ppdet/modeling/architectures/keypoint_hrhrnet.py +0 -287
- paddlex/ppdet/modeling/architectures/keypoint_hrnet.py +0 -267
- paddlex/ppdet/modeling/architectures/mask_rcnn.py +0 -135
- paddlex/ppdet/modeling/architectures/meta_arch.py +0 -128
- paddlex/ppdet/modeling/architectures/picodet.py +0 -91
- paddlex/ppdet/modeling/architectures/s2anet.py +0 -102
- paddlex/ppdet/modeling/architectures/solov2.py +0 -110
- paddlex/ppdet/modeling/architectures/sparse_rcnn.py +0 -99
- paddlex/ppdet/modeling/architectures/ssd.py +0 -93
- paddlex/ppdet/modeling/architectures/tood.py +0 -78
- paddlex/ppdet/modeling/architectures/ttfnet.py +0 -98
- paddlex/ppdet/modeling/architectures/yolo.py +0 -124
- paddlex/ppdet/modeling/assigners/__init__.py +0 -23
- paddlex/ppdet/modeling/assigners/atss_assigner.py +0 -211
- paddlex/ppdet/modeling/assigners/simota_assigner.py +0 -262
- paddlex/ppdet/modeling/assigners/task_aligned_assigner.py +0 -158
- paddlex/ppdet/modeling/assigners/utils.py +0 -195
- paddlex/ppdet/modeling/backbones/__init__.py +0 -49
- paddlex/ppdet/modeling/backbones/blazenet.py +0 -323
- paddlex/ppdet/modeling/backbones/darknet.py +0 -340
- paddlex/ppdet/modeling/backbones/dla.py +0 -244
- paddlex/ppdet/modeling/backbones/esnet.py +0 -290
- paddlex/ppdet/modeling/backbones/ghostnet.py +0 -470
- paddlex/ppdet/modeling/backbones/hardnet.py +0 -224
- paddlex/ppdet/modeling/backbones/hrnet.py +0 -727
- paddlex/ppdet/modeling/backbones/lcnet.py +0 -259
- paddlex/ppdet/modeling/backbones/lite_hrnet.py +0 -886
- paddlex/ppdet/modeling/backbones/mobilenet_v1.py +0 -418
- paddlex/ppdet/modeling/backbones/mobilenet_v3.py +0 -483
- paddlex/ppdet/modeling/backbones/name_adapter.py +0 -69
- paddlex/ppdet/modeling/backbones/res2net.py +0 -358
- paddlex/ppdet/modeling/backbones/resnet.py +0 -613
- paddlex/ppdet/modeling/backbones/senet.py +0 -139
- paddlex/ppdet/modeling/backbones/shufflenet_v2.py +0 -246
- paddlex/ppdet/modeling/backbones/swin_transformer.py +0 -743
- paddlex/ppdet/modeling/backbones/vgg.py +0 -210
- paddlex/ppdet/modeling/bbox_utils.py +0 -778
- paddlex/ppdet/modeling/heads/__init__.py +0 -53
- paddlex/ppdet/modeling/heads/bbox_head.py +0 -377
- paddlex/ppdet/modeling/heads/cascade_head.py +0 -284
- paddlex/ppdet/modeling/heads/centernet_head.py +0 -292
- paddlex/ppdet/modeling/heads/detr_head.py +0 -368
- paddlex/ppdet/modeling/heads/face_head.py +0 -110
- paddlex/ppdet/modeling/heads/fcos_head.py +0 -259
- paddlex/ppdet/modeling/heads/gfl_head.py +0 -487
- paddlex/ppdet/modeling/heads/keypoint_hrhrnet_head.py +0 -108
- paddlex/ppdet/modeling/heads/mask_head.py +0 -250
- paddlex/ppdet/modeling/heads/pico_head.py +0 -278
- paddlex/ppdet/modeling/heads/roi_extractor.py +0 -111
- paddlex/ppdet/modeling/heads/s2anet_head.py +0 -1056
- paddlex/ppdet/modeling/heads/simota_head.py +0 -506
- paddlex/ppdet/modeling/heads/solov2_head.py +0 -560
- paddlex/ppdet/modeling/heads/sparsercnn_head.py +0 -375
- paddlex/ppdet/modeling/heads/ssd_head.py +0 -215
- paddlex/ppdet/modeling/heads/tood_head.py +0 -366
- paddlex/ppdet/modeling/heads/ttf_head.py +0 -316
- paddlex/ppdet/modeling/heads/yolo_head.py +0 -124
- paddlex/ppdet/modeling/initializer.py +0 -317
- paddlex/ppdet/modeling/keypoint_utils.py +0 -342
- paddlex/ppdet/modeling/layers.py +0 -1430
- paddlex/ppdet/modeling/losses/__init__.py +0 -43
- paddlex/ppdet/modeling/losses/ctfocal_loss.py +0 -68
- paddlex/ppdet/modeling/losses/detr_loss.py +0 -233
- paddlex/ppdet/modeling/losses/fairmot_loss.py +0 -41
- paddlex/ppdet/modeling/losses/fcos_loss.py +0 -225
- paddlex/ppdet/modeling/losses/gfocal_loss.py +0 -217
- paddlex/ppdet/modeling/losses/iou_aware_loss.py +0 -47
- paddlex/ppdet/modeling/losses/iou_loss.py +0 -210
- paddlex/ppdet/modeling/losses/jde_loss.py +0 -193
- paddlex/ppdet/modeling/losses/keypoint_loss.py +0 -229
- paddlex/ppdet/modeling/losses/solov2_loss.py +0 -101
- paddlex/ppdet/modeling/losses/sparsercnn_loss.py +0 -425
- paddlex/ppdet/modeling/losses/ssd_loss.py +0 -170
- paddlex/ppdet/modeling/losses/varifocal_loss.py +0 -152
- paddlex/ppdet/modeling/losses/yolo_loss.py +0 -212
- paddlex/ppdet/modeling/mot/__init__.py +0 -25
- paddlex/ppdet/modeling/mot/matching/__init__.py +0 -19
- paddlex/ppdet/modeling/mot/matching/deepsort_matching.py +0 -382
- paddlex/ppdet/modeling/mot/matching/jde_matching.py +0 -144
- paddlex/ppdet/modeling/mot/motion/__init__.py +0 -17
- paddlex/ppdet/modeling/mot/motion/kalman_filter.py +0 -270
- paddlex/ppdet/modeling/mot/tracker/__init__.py +0 -23
- paddlex/ppdet/modeling/mot/tracker/base_jde_tracker.py +0 -297
- paddlex/ppdet/modeling/mot/tracker/base_sde_tracker.py +0 -156
- paddlex/ppdet/modeling/mot/tracker/deepsort_tracker.py +0 -188
- paddlex/ppdet/modeling/mot/tracker/jde_tracker.py +0 -277
- paddlex/ppdet/modeling/mot/utils.py +0 -263
- paddlex/ppdet/modeling/mot/visualization.py +0 -150
- paddlex/ppdet/modeling/necks/__init__.py +0 -30
- paddlex/ppdet/modeling/necks/bifpn.py +0 -302
- paddlex/ppdet/modeling/necks/blazeface_fpn.py +0 -216
- paddlex/ppdet/modeling/necks/centernet_fpn.py +0 -426
- paddlex/ppdet/modeling/necks/csp_pan.py +0 -364
- paddlex/ppdet/modeling/necks/fpn.py +0 -231
- paddlex/ppdet/modeling/necks/hrfpn.py +0 -126
- paddlex/ppdet/modeling/necks/ttf_fpn.py +0 -242
- paddlex/ppdet/modeling/necks/yolo_fpn.py +0 -988
- paddlex/ppdet/modeling/ops.py +0 -1611
- paddlex/ppdet/modeling/post_process.py +0 -731
- paddlex/ppdet/modeling/proposal_generator/__init__.py +0 -2
- paddlex/ppdet/modeling/proposal_generator/anchor_generator.py +0 -135
- paddlex/ppdet/modeling/proposal_generator/proposal_generator.py +0 -77
- paddlex/ppdet/modeling/proposal_generator/rpn_head.py +0 -260
- paddlex/ppdet/modeling/proposal_generator/target.py +0 -681
- paddlex/ppdet/modeling/proposal_generator/target_layer.py +0 -491
- paddlex/ppdet/modeling/reid/__init__.py +0 -25
- paddlex/ppdet/modeling/reid/fairmot_embedding_head.py +0 -225
- paddlex/ppdet/modeling/reid/jde_embedding_head.py +0 -214
- paddlex/ppdet/modeling/reid/pplcnet_embedding.py +0 -282
- paddlex/ppdet/modeling/reid/pyramidal_embedding.py +0 -144
- paddlex/ppdet/modeling/reid/resnet.py +0 -310
- paddlex/ppdet/modeling/shape_spec.py +0 -25
- paddlex/ppdet/modeling/transformers/__init__.py +0 -25
- paddlex/ppdet/modeling/transformers/deformable_transformer.py +0 -517
- paddlex/ppdet/modeling/transformers/detr_transformer.py +0 -353
- paddlex/ppdet/modeling/transformers/matchers.py +0 -127
- paddlex/ppdet/modeling/transformers/position_encoding.py +0 -108
- paddlex/ppdet/modeling/transformers/utils.py +0 -110
- paddlex/ppdet/optimizer.py +0 -335
- paddlex/ppdet/slim/__init__.py +0 -82
- paddlex/ppdet/slim/distill.py +0 -110
- paddlex/ppdet/slim/prune.py +0 -85
- paddlex/ppdet/slim/quant.py +0 -84
- paddlex/ppdet/slim/unstructured_prune.py +0 -66
- paddlex/ppdet/utils/__init__.py +0 -13
- paddlex/ppdet/utils/check.py +0 -112
- paddlex/ppdet/utils/checkpoint.py +0 -226
- paddlex/ppdet/utils/cli.py +0 -151
- paddlex/ppdet/utils/colormap.py +0 -58
- paddlex/ppdet/utils/download.py +0 -558
- paddlex/ppdet/utils/logger.py +0 -70
- paddlex/ppdet/utils/profiler.py +0 -111
- paddlex/ppdet/utils/stats.py +0 -94
- paddlex/ppdet/utils/visualizer.py +0 -321
- paddlex/ppdet/utils/voc_utils.py +0 -86
- paddlex/seg.py +0 -41
- paddlex/tools/__init__.py +0 -17
- paddlex/tools/anchor_clustering/__init__.py +0 -15
- paddlex/tools/anchor_clustering/yolo_cluster.py +0 -178
- paddlex/tools/convert.py +0 -52
- paddlex/tools/dataset_conversion/__init__.py +0 -24
- paddlex/tools/dataset_conversion/x2coco.py +0 -379
- paddlex/tools/dataset_conversion/x2imagenet.py +0 -82
- paddlex/tools/dataset_conversion/x2seg.py +0 -343
- paddlex/tools/dataset_conversion/x2voc.py +0 -230
- paddlex/tools/dataset_split/__init__.py +0 -23
- paddlex/tools/dataset_split/coco_split.py +0 -69
- paddlex/tools/dataset_split/imagenet_split.py +0 -75
- paddlex/tools/dataset_split/seg_split.py +0 -96
- paddlex/tools/dataset_split/utils.py +0 -75
- paddlex/tools/dataset_split/voc_split.py +0 -91
- paddlex/tools/split.py +0 -41
- paddlex/utils/checkpoint.py +0 -492
- paddlex/utils/shm.py +0 -67
- paddlex/utils/stats.py +0 -68
- paddlex/utils/utils.py +0 -229
- paddlex-2.1.0.data/data/paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex-2.1.0.dist-info/LICENSE +0 -201
- paddlex-2.1.0.dist-info/METADATA +0 -32
- paddlex-2.1.0.dist-info/RECORD +0 -698
- paddlex-2.1.0.dist-info/WHEEL +0 -5
- paddlex-2.1.0.dist-info/entry_points.txt +0 -4
- paddlex-2.1.0.dist-info/top_level.txt +0 -3
- paddlex_restful/__init__.py +0 -15
- paddlex_restful/command.py +0 -63
- paddlex_restful/restful/__init__.py +0 -15
- paddlex_restful/restful/app.py +0 -969
- paddlex_restful/restful/dataset/__init__.py +0 -13
- paddlex_restful/restful/dataset/cls_dataset.py +0 -159
- paddlex_restful/restful/dataset/dataset.py +0 -266
- paddlex_restful/restful/dataset/datasetbase.py +0 -86
- paddlex_restful/restful/dataset/det_dataset.py +0 -190
- paddlex_restful/restful/dataset/ins_seg_dataset.py +0 -312
- paddlex_restful/restful/dataset/operate.py +0 -155
- paddlex_restful/restful/dataset/seg_dataset.py +0 -222
- paddlex_restful/restful/dataset/utils.py +0 -267
- paddlex_restful/restful/demo.py +0 -202
- paddlex_restful/restful/dir.py +0 -45
- paddlex_restful/restful/model.py +0 -312
- paddlex_restful/restful/project/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/__init__.py +0 -13
- paddlex_restful/restful/project/evaluate/classification.py +0 -126
- paddlex_restful/restful/project/evaluate/detection.py +0 -789
- paddlex_restful/restful/project/evaluate/draw_pred_result.py +0 -181
- paddlex_restful/restful/project/evaluate/segmentation.py +0 -122
- paddlex_restful/restful/project/operate.py +0 -931
- paddlex_restful/restful/project/project.py +0 -143
- paddlex_restful/restful/project/prune/__init__.py +0 -13
- paddlex_restful/restful/project/prune/classification.py +0 -32
- paddlex_restful/restful/project/prune/detection.py +0 -48
- paddlex_restful/restful/project/prune/segmentation.py +0 -34
- paddlex_restful/restful/project/task.py +0 -884
- paddlex_restful/restful/project/train/__init__.py +0 -13
- paddlex_restful/restful/project/train/classification.py +0 -141
- paddlex_restful/restful/project/train/detection.py +0 -263
- paddlex_restful/restful/project/train/params.py +0 -432
- paddlex_restful/restful/project/train/params_v2.py +0 -326
- paddlex_restful/restful/project/train/segmentation.py +0 -191
- paddlex_restful/restful/project/visualize.py +0 -244
- paddlex_restful/restful/system.py +0 -102
- paddlex_restful/restful/templates/paddlex_restful_demo.html +0 -5205
- paddlex_restful/restful/utils.py +0 -841
- paddlex_restful/restful/workspace.py +0 -343
- paddlex_restful/restful/workspace_pb2.py +0 -1411
@@ -1,778 +0,0 @@
|
|
1
|
-
# Copyright (c) 2020 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
|
-
import math
|
16
|
-
import paddle
|
17
|
-
import numpy as np
|
18
|
-
|
19
|
-
|
20
|
-
def bbox2delta(src_boxes, tgt_boxes, weights):
|
21
|
-
src_w = src_boxes[:, 2] - src_boxes[:, 0]
|
22
|
-
src_h = src_boxes[:, 3] - src_boxes[:, 1]
|
23
|
-
src_ctr_x = src_boxes[:, 0] + 0.5 * src_w
|
24
|
-
src_ctr_y = src_boxes[:, 1] + 0.5 * src_h
|
25
|
-
|
26
|
-
tgt_w = tgt_boxes[:, 2] - tgt_boxes[:, 0]
|
27
|
-
tgt_h = tgt_boxes[:, 3] - tgt_boxes[:, 1]
|
28
|
-
tgt_ctr_x = tgt_boxes[:, 0] + 0.5 * tgt_w
|
29
|
-
tgt_ctr_y = tgt_boxes[:, 1] + 0.5 * tgt_h
|
30
|
-
|
31
|
-
wx, wy, ww, wh = weights
|
32
|
-
dx = wx * (tgt_ctr_x - src_ctr_x) / src_w
|
33
|
-
dy = wy * (tgt_ctr_y - src_ctr_y) / src_h
|
34
|
-
dw = ww * paddle.log(tgt_w / src_w)
|
35
|
-
dh = wh * paddle.log(tgt_h / src_h)
|
36
|
-
|
37
|
-
deltas = paddle.stack((dx, dy, dw, dh), axis=1)
|
38
|
-
return deltas
|
39
|
-
|
40
|
-
|
41
|
-
def delta2bbox(deltas, boxes, weights):
|
42
|
-
clip_scale = math.log(1000.0 / 16)
|
43
|
-
|
44
|
-
widths = boxes[:, 2] - boxes[:, 0]
|
45
|
-
heights = boxes[:, 3] - boxes[:, 1]
|
46
|
-
ctr_x = boxes[:, 0] + 0.5 * widths
|
47
|
-
ctr_y = boxes[:, 1] + 0.5 * heights
|
48
|
-
|
49
|
-
wx, wy, ww, wh = weights
|
50
|
-
dx = deltas[:, 0::4] / wx
|
51
|
-
dy = deltas[:, 1::4] / wy
|
52
|
-
dw = deltas[:, 2::4] / ww
|
53
|
-
dh = deltas[:, 3::4] / wh
|
54
|
-
# Prevent sending too large values into paddle.exp()
|
55
|
-
dw = paddle.clip(dw, max=clip_scale)
|
56
|
-
dh = paddle.clip(dh, max=clip_scale)
|
57
|
-
|
58
|
-
pred_ctr_x = dx * widths.unsqueeze(1) + ctr_x.unsqueeze(1)
|
59
|
-
pred_ctr_y = dy * heights.unsqueeze(1) + ctr_y.unsqueeze(1)
|
60
|
-
pred_w = paddle.exp(dw) * widths.unsqueeze(1)
|
61
|
-
pred_h = paddle.exp(dh) * heights.unsqueeze(1)
|
62
|
-
|
63
|
-
pred_boxes = []
|
64
|
-
pred_boxes.append(pred_ctr_x - 0.5 * pred_w)
|
65
|
-
pred_boxes.append(pred_ctr_y - 0.5 * pred_h)
|
66
|
-
pred_boxes.append(pred_ctr_x + 0.5 * pred_w)
|
67
|
-
pred_boxes.append(pred_ctr_y + 0.5 * pred_h)
|
68
|
-
pred_boxes = paddle.stack(pred_boxes, axis=-1)
|
69
|
-
|
70
|
-
return pred_boxes
|
71
|
-
|
72
|
-
|
73
|
-
def expand_bbox(bboxes, scale):
|
74
|
-
w_half = (bboxes[:, 2] - bboxes[:, 0]) * .5
|
75
|
-
h_half = (bboxes[:, 3] - bboxes[:, 1]) * .5
|
76
|
-
x_c = (bboxes[:, 2] + bboxes[:, 0]) * .5
|
77
|
-
y_c = (bboxes[:, 3] + bboxes[:, 1]) * .5
|
78
|
-
|
79
|
-
w_half *= scale
|
80
|
-
h_half *= scale
|
81
|
-
|
82
|
-
bboxes_exp = np.zeros(bboxes.shape, dtype=np.float32)
|
83
|
-
bboxes_exp[:, 0] = x_c - w_half
|
84
|
-
bboxes_exp[:, 2] = x_c + w_half
|
85
|
-
bboxes_exp[:, 1] = y_c - h_half
|
86
|
-
bboxes_exp[:, 3] = y_c + h_half
|
87
|
-
|
88
|
-
return bboxes_exp
|
89
|
-
|
90
|
-
|
91
|
-
def clip_bbox(boxes, im_shape):
|
92
|
-
h, w = im_shape[0], im_shape[1]
|
93
|
-
x1 = boxes[:, 0].clip(0, w)
|
94
|
-
y1 = boxes[:, 1].clip(0, h)
|
95
|
-
x2 = boxes[:, 2].clip(0, w)
|
96
|
-
y2 = boxes[:, 3].clip(0, h)
|
97
|
-
return paddle.stack([x1, y1, x2, y2], axis=1)
|
98
|
-
|
99
|
-
|
100
|
-
def nonempty_bbox(boxes, min_size=0, return_mask=False):
|
101
|
-
w = boxes[:, 2] - boxes[:, 0]
|
102
|
-
h = boxes[:, 3] - boxes[:, 1]
|
103
|
-
mask = paddle.logical_and(h > min_size, w > min_size)
|
104
|
-
if return_mask:
|
105
|
-
return mask
|
106
|
-
keep = paddle.nonzero(mask).flatten()
|
107
|
-
return keep
|
108
|
-
|
109
|
-
|
110
|
-
def bbox_area(boxes):
|
111
|
-
return (boxes[:, 2] - boxes[:, 0]) * (boxes[:, 3] - boxes[:, 1])
|
112
|
-
|
113
|
-
|
114
|
-
def bbox_overlaps(boxes1, boxes2):
|
115
|
-
"""
|
116
|
-
Calculate overlaps between boxes1 and boxes2
|
117
|
-
|
118
|
-
Args:
|
119
|
-
boxes1 (Tensor): boxes with shape [M, 4]
|
120
|
-
boxes2 (Tensor): boxes with shape [N, 4]
|
121
|
-
|
122
|
-
Return:
|
123
|
-
overlaps (Tensor): overlaps between boxes1 and boxes2 with shape [M, N]
|
124
|
-
"""
|
125
|
-
M = boxes1.shape[0]
|
126
|
-
N = boxes2.shape[0]
|
127
|
-
if M * N == 0:
|
128
|
-
return paddle.zeros([M, N], dtype='float32')
|
129
|
-
area1 = bbox_area(boxes1)
|
130
|
-
area2 = bbox_area(boxes2)
|
131
|
-
|
132
|
-
xy_max = paddle.minimum(
|
133
|
-
paddle.unsqueeze(boxes1, 1)[:, :, 2:], boxes2[:, 2:])
|
134
|
-
xy_min = paddle.maximum(
|
135
|
-
paddle.unsqueeze(boxes1, 1)[:, :, :2], boxes2[:, :2])
|
136
|
-
width_height = xy_max - xy_min
|
137
|
-
width_height = width_height.clip(min=0)
|
138
|
-
inter = width_height.prod(axis=2)
|
139
|
-
|
140
|
-
overlaps = paddle.where(inter > 0, inter /
|
141
|
-
(paddle.unsqueeze(area1, 1) + area2 - inter),
|
142
|
-
paddle.zeros_like(inter))
|
143
|
-
return overlaps
|
144
|
-
|
145
|
-
|
146
|
-
def batch_bbox_overlaps(bboxes1,
|
147
|
-
bboxes2,
|
148
|
-
mode='iou',
|
149
|
-
is_aligned=False,
|
150
|
-
eps=1e-6):
|
151
|
-
"""Calculate overlap between two set of bboxes.
|
152
|
-
If ``is_aligned `` is ``False``, then calculate the overlaps between each
|
153
|
-
bbox of bboxes1 and bboxes2, otherwise the overlaps between each aligned
|
154
|
-
pair of bboxes1 and bboxes2.
|
155
|
-
Args:
|
156
|
-
bboxes1 (Tensor): shape (B, m, 4) in <x1, y1, x2, y2> format or empty.
|
157
|
-
bboxes2 (Tensor): shape (B, n, 4) in <x1, y1, x2, y2> format or empty.
|
158
|
-
B indicates the batch dim, in shape (B1, B2, ..., Bn).
|
159
|
-
If ``is_aligned `` is ``True``, then m and n must be equal.
|
160
|
-
mode (str): "iou" (intersection over union) or "iof" (intersection over
|
161
|
-
foreground).
|
162
|
-
is_aligned (bool, optional): If True, then m and n must be equal.
|
163
|
-
Default False.
|
164
|
-
eps (float, optional): A value added to the denominator for numerical
|
165
|
-
stability. Default 1e-6.
|
166
|
-
Returns:
|
167
|
-
Tensor: shape (m, n) if ``is_aligned `` is False else shape (m,)
|
168
|
-
"""
|
169
|
-
assert mode in ['iou', 'iof', 'giou'], 'Unsupported mode {}'.format(mode)
|
170
|
-
# Either the boxes are empty or the length of boxes's last dimenstion is 4
|
171
|
-
assert (bboxes1.shape[-1] == 4 or bboxes1.shape[0] == 0)
|
172
|
-
assert (bboxes2.shape[-1] == 4 or bboxes2.shape[0] == 0)
|
173
|
-
|
174
|
-
# Batch dim must be the same
|
175
|
-
# Batch dim: (B1, B2, ... Bn)
|
176
|
-
assert bboxes1.shape[:-2] == bboxes2.shape[:-2]
|
177
|
-
batch_shape = bboxes1.shape[:-2]
|
178
|
-
|
179
|
-
rows = bboxes1.shape[-2] if bboxes1.shape[0] > 0 else 0
|
180
|
-
cols = bboxes2.shape[-2] if bboxes2.shape[0] > 0 else 0
|
181
|
-
if is_aligned:
|
182
|
-
assert rows == cols
|
183
|
-
|
184
|
-
if rows * cols == 0:
|
185
|
-
if is_aligned:
|
186
|
-
return paddle.full(batch_shape + (rows, ), 1)
|
187
|
-
else:
|
188
|
-
return paddle.full(batch_shape + (rows, cols), 1)
|
189
|
-
|
190
|
-
area1 = (bboxes1[:, 2] - bboxes1[:, 0]) * (bboxes1[:, 3] - bboxes1[:, 1])
|
191
|
-
area2 = (bboxes2[:, 2] - bboxes2[:, 0]) * (bboxes2[:, 3] - bboxes2[:, 1])
|
192
|
-
|
193
|
-
if is_aligned:
|
194
|
-
lt = paddle.maximum(bboxes1[:, :2], bboxes2[:, :2]) # [B, rows, 2]
|
195
|
-
rb = paddle.minimum(bboxes1[:, 2:], bboxes2[:, 2:]) # [B, rows, 2]
|
196
|
-
|
197
|
-
wh = (rb - lt).clip(min=0) # [B, rows, 2]
|
198
|
-
overlap = wh[:, 0] * wh[:, 1]
|
199
|
-
|
200
|
-
if mode in ['iou', 'giou']:
|
201
|
-
union = area1 + area2 - overlap
|
202
|
-
else:
|
203
|
-
union = area1
|
204
|
-
if mode == 'giou':
|
205
|
-
enclosed_lt = paddle.minimum(bboxes1[:, :2], bboxes2[:, :2])
|
206
|
-
enclosed_rb = paddle.maximum(bboxes1[:, 2:], bboxes2[:, 2:])
|
207
|
-
else:
|
208
|
-
lt = paddle.maximum(bboxes1[:, :2].reshape([rows, 1, 2]),
|
209
|
-
bboxes2[:, :2]) # [B, rows, cols, 2]
|
210
|
-
rb = paddle.minimum(bboxes1[:, 2:].reshape([rows, 1, 2]),
|
211
|
-
bboxes2[:, 2:]) # [B, rows, cols, 2]
|
212
|
-
|
213
|
-
wh = (rb - lt).clip(min=0) # [B, rows, cols, 2]
|
214
|
-
overlap = wh[:, :, 0] * wh[:, :, 1]
|
215
|
-
|
216
|
-
if mode in ['iou', 'giou']:
|
217
|
-
union = area1.reshape([rows,1]) \
|
218
|
-
+ area2.reshape([1,cols]) - overlap
|
219
|
-
else:
|
220
|
-
union = area1[:, None]
|
221
|
-
if mode == 'giou':
|
222
|
-
enclosed_lt = paddle.minimum(bboxes1[:, :2].reshape([rows, 1, 2]),
|
223
|
-
bboxes2[:, :2])
|
224
|
-
enclosed_rb = paddle.maximum(bboxes1[:, 2:].reshape([rows, 1, 2]),
|
225
|
-
bboxes2[:, 2:])
|
226
|
-
|
227
|
-
eps = paddle.to_tensor([eps])
|
228
|
-
union = paddle.maximum(union, eps)
|
229
|
-
ious = overlap / union
|
230
|
-
if mode in ['iou', 'iof']:
|
231
|
-
return ious
|
232
|
-
# calculate gious
|
233
|
-
enclose_wh = (enclosed_rb - enclosed_lt).clip(min=0)
|
234
|
-
enclose_area = enclose_wh[:, :, 0] * enclose_wh[:, :, 1]
|
235
|
-
enclose_area = paddle.maximum(enclose_area, eps)
|
236
|
-
gious = ious - (enclose_area - union) / enclose_area
|
237
|
-
return 1 - gious
|
238
|
-
|
239
|
-
|
240
|
-
def xywh2xyxy(box):
|
241
|
-
x, y, w, h = box
|
242
|
-
x1 = x - w * 0.5
|
243
|
-
y1 = y - h * 0.5
|
244
|
-
x2 = x + w * 0.5
|
245
|
-
y2 = y + h * 0.5
|
246
|
-
return [x1, y1, x2, y2]
|
247
|
-
|
248
|
-
|
249
|
-
def make_grid(h, w, dtype):
|
250
|
-
yv, xv = paddle.meshgrid([paddle.arange(h), paddle.arange(w)])
|
251
|
-
return paddle.stack((xv, yv), 2).cast(dtype=dtype)
|
252
|
-
|
253
|
-
|
254
|
-
def decode_yolo(box, anchor, downsample_ratio):
|
255
|
-
"""decode yolo box
|
256
|
-
|
257
|
-
Args:
|
258
|
-
box (list): [x, y, w, h], all have the shape [b, na, h, w, 1]
|
259
|
-
anchor (list): anchor with the shape [na, 2]
|
260
|
-
downsample_ratio (int): downsample ratio, default 32
|
261
|
-
scale (float): scale, default 1.
|
262
|
-
|
263
|
-
Return:
|
264
|
-
box (list): decoded box, [x, y, w, h], all have the shape [b, na, h, w, 1]
|
265
|
-
"""
|
266
|
-
x, y, w, h = box
|
267
|
-
na, grid_h, grid_w = x.shape[1:4]
|
268
|
-
grid = make_grid(grid_h, grid_w, x.dtype).reshape(
|
269
|
-
(1, 1, grid_h, grid_w, 2))
|
270
|
-
x1 = (x + grid[:, :, :, :, 0:1]) / grid_w
|
271
|
-
y1 = (y + grid[:, :, :, :, 1:2]) / grid_h
|
272
|
-
|
273
|
-
anchor = paddle.to_tensor(anchor)
|
274
|
-
anchor = paddle.cast(anchor, x.dtype)
|
275
|
-
anchor = anchor.reshape((1, na, 1, 1, 2))
|
276
|
-
w1 = paddle.exp(w) * anchor[:, :, :, :, 0:1] / (downsample_ratio * grid_w)
|
277
|
-
h1 = paddle.exp(h) * anchor[:, :, :, :, 1:2] / (downsample_ratio * grid_h)
|
278
|
-
|
279
|
-
return [x1, y1, w1, h1]
|
280
|
-
|
281
|
-
|
282
|
-
def iou_similarity(box1, box2, eps=1e-9):
|
283
|
-
"""Calculate iou of box1 and box2
|
284
|
-
|
285
|
-
Args:
|
286
|
-
box1 (Tensor): box with the shape [N, M1, 4]
|
287
|
-
box2 (Tensor): box with the shape [N, M2, 4]
|
288
|
-
|
289
|
-
Return:
|
290
|
-
iou (Tensor): iou between box1 and box2 with the shape [N, M1, M2]
|
291
|
-
"""
|
292
|
-
box1 = box1.unsqueeze(2) # [N, M1, 4] -> [N, M1, 1, 4]
|
293
|
-
box2 = box2.unsqueeze(1) # [N, M2, 4] -> [N, 1, M2, 4]
|
294
|
-
px1y1, px2y2 = box1[:, :, :, 0:2], box1[:, :, :, 2:4]
|
295
|
-
gx1y1, gx2y2 = box2[:, :, :, 0:2], box2[:, :, :, 2:4]
|
296
|
-
x1y1 = paddle.maximum(px1y1, gx1y1)
|
297
|
-
x2y2 = paddle.minimum(px2y2, gx2y2)
|
298
|
-
overlap = (x2y2 - x1y1).clip(0).prod(-1)
|
299
|
-
area1 = (px2y2 - px1y1).clip(0).prod(-1)
|
300
|
-
area2 = (gx2y2 - gx1y1).clip(0).prod(-1)
|
301
|
-
union = area1 + area2 - overlap + eps
|
302
|
-
return overlap / union
|
303
|
-
|
304
|
-
|
305
|
-
def bbox_iou(box1, box2, giou=False, diou=False, ciou=False, eps=1e-9):
|
306
|
-
"""calculate the iou of box1 and box2
|
307
|
-
|
308
|
-
Args:
|
309
|
-
box1 (list): [x, y, w, h], all have the shape [b, na, h, w, 1]
|
310
|
-
box2 (list): [x, y, w, h], all have the shape [b, na, h, w, 1]
|
311
|
-
giou (bool): whether use giou or not, default False
|
312
|
-
diou (bool): whether use diou or not, default False
|
313
|
-
ciou (bool): whether use ciou or not, default False
|
314
|
-
eps (float): epsilon to avoid divide by zero
|
315
|
-
|
316
|
-
Return:
|
317
|
-
iou (Tensor): iou of box1 and box1, with the shape [b, na, h, w, 1]
|
318
|
-
"""
|
319
|
-
px1, py1, px2, py2 = box1
|
320
|
-
gx1, gy1, gx2, gy2 = box2
|
321
|
-
x1 = paddle.maximum(px1, gx1)
|
322
|
-
y1 = paddle.maximum(py1, gy1)
|
323
|
-
x2 = paddle.minimum(px2, gx2)
|
324
|
-
y2 = paddle.minimum(py2, gy2)
|
325
|
-
|
326
|
-
overlap = ((x2 - x1).clip(0)) * ((y2 - y1).clip(0))
|
327
|
-
|
328
|
-
area1 = (px2 - px1) * (py2 - py1)
|
329
|
-
area1 = area1.clip(0)
|
330
|
-
|
331
|
-
area2 = (gx2 - gx1) * (gy2 - gy1)
|
332
|
-
area2 = area2.clip(0)
|
333
|
-
|
334
|
-
union = area1 + area2 - overlap + eps
|
335
|
-
iou = overlap / union
|
336
|
-
|
337
|
-
if giou or ciou or diou:
|
338
|
-
# convex w, h
|
339
|
-
cw = paddle.maximum(px2, gx2) - paddle.minimum(px1, gx1)
|
340
|
-
ch = paddle.maximum(py2, gy2) - paddle.minimum(py1, gy1)
|
341
|
-
if giou:
|
342
|
-
c_area = cw * ch + eps
|
343
|
-
return iou - (c_area - union) / c_area
|
344
|
-
else:
|
345
|
-
# convex diagonal squared
|
346
|
-
c2 = cw**2 + ch**2 + eps
|
347
|
-
# center distance
|
348
|
-
rho2 = (
|
349
|
-
(px1 + px2 - gx1 - gx2)**2 + (py1 + py2 - gy1 - gy2)**2) / 4
|
350
|
-
if diou:
|
351
|
-
return iou - rho2 / c2
|
352
|
-
else:
|
353
|
-
w1, h1 = px2 - px1, py2 - py1 + eps
|
354
|
-
w2, h2 = gx2 - gx1, gy2 - gy1 + eps
|
355
|
-
delta = paddle.atan(w1 / h1) - paddle.atan(w2 / h2)
|
356
|
-
v = (4 / math.pi**2) * paddle.pow(delta, 2)
|
357
|
-
alpha = v / (1 + eps - iou + v)
|
358
|
-
alpha.stop_gradient = True
|
359
|
-
return iou - (rho2 / c2 + v * alpha)
|
360
|
-
else:
|
361
|
-
return iou
|
362
|
-
|
363
|
-
|
364
|
-
def rect2rbox(bboxes):
|
365
|
-
"""
|
366
|
-
:param bboxes: shape (n, 4) (xmin, ymin, xmax, ymax)
|
367
|
-
:return: dbboxes: shape (n, 5) (x_ctr, y_ctr, w, h, angle)
|
368
|
-
"""
|
369
|
-
bboxes = bboxes.reshape(-1, 4)
|
370
|
-
num_boxes = bboxes.shape[0]
|
371
|
-
|
372
|
-
x_ctr = (bboxes[:, 2] + bboxes[:, 0]) / 2.0
|
373
|
-
y_ctr = (bboxes[:, 3] + bboxes[:, 1]) / 2.0
|
374
|
-
edges1 = np.abs(bboxes[:, 2] - bboxes[:, 0])
|
375
|
-
edges2 = np.abs(bboxes[:, 3] - bboxes[:, 1])
|
376
|
-
angles = np.zeros([num_boxes], dtype=bboxes.dtype)
|
377
|
-
|
378
|
-
inds = edges1 < edges2
|
379
|
-
|
380
|
-
rboxes = np.stack((x_ctr, y_ctr, edges1, edges2, angles), axis=1)
|
381
|
-
rboxes[inds, 2] = edges2[inds]
|
382
|
-
rboxes[inds, 3] = edges1[inds]
|
383
|
-
rboxes[inds, 4] = np.pi / 2.0
|
384
|
-
return rboxes
|
385
|
-
|
386
|
-
|
387
|
-
def delta2rbox(rrois,
|
388
|
-
deltas,
|
389
|
-
means=[0, 0, 0, 0, 0],
|
390
|
-
stds=[1, 1, 1, 1, 1],
|
391
|
-
wh_ratio_clip=1e-6):
|
392
|
-
"""
|
393
|
-
:param rrois: (cx, cy, w, h, theta)
|
394
|
-
:param deltas: (dx, dy, dw, dh, dtheta)
|
395
|
-
:param means:
|
396
|
-
:param stds:
|
397
|
-
:param wh_ratio_clip:
|
398
|
-
:return:
|
399
|
-
"""
|
400
|
-
means = paddle.to_tensor(means)
|
401
|
-
stds = paddle.to_tensor(stds)
|
402
|
-
deltas = paddle.reshape(deltas, [-1, deltas.shape[-1]])
|
403
|
-
denorm_deltas = deltas * stds + means
|
404
|
-
|
405
|
-
dx = denorm_deltas[:, 0]
|
406
|
-
dy = denorm_deltas[:, 1]
|
407
|
-
dw = denorm_deltas[:, 2]
|
408
|
-
dh = denorm_deltas[:, 3]
|
409
|
-
dangle = denorm_deltas[:, 4]
|
410
|
-
|
411
|
-
max_ratio = np.abs(np.log(wh_ratio_clip))
|
412
|
-
dw = paddle.clip(dw, min=-max_ratio, max=max_ratio)
|
413
|
-
dh = paddle.clip(dh, min=-max_ratio, max=max_ratio)
|
414
|
-
|
415
|
-
rroi_x = rrois[:, 0]
|
416
|
-
rroi_y = rrois[:, 1]
|
417
|
-
rroi_w = rrois[:, 2]
|
418
|
-
rroi_h = rrois[:, 3]
|
419
|
-
rroi_angle = rrois[:, 4]
|
420
|
-
|
421
|
-
gx = dx * rroi_w * paddle.cos(rroi_angle) - dy * rroi_h * paddle.sin(
|
422
|
-
rroi_angle) + rroi_x
|
423
|
-
gy = dx * rroi_w * paddle.sin(rroi_angle) + dy * rroi_h * paddle.cos(
|
424
|
-
rroi_angle) + rroi_y
|
425
|
-
gw = rroi_w * dw.exp()
|
426
|
-
gh = rroi_h * dh.exp()
|
427
|
-
ga = np.pi * dangle + rroi_angle
|
428
|
-
ga = (ga + np.pi / 4) % np.pi - np.pi / 4
|
429
|
-
ga = paddle.to_tensor(ga)
|
430
|
-
|
431
|
-
gw = paddle.to_tensor(gw, dtype='float32')
|
432
|
-
gh = paddle.to_tensor(gh, dtype='float32')
|
433
|
-
bboxes = paddle.stack([gx, gy, gw, gh, ga], axis=-1)
|
434
|
-
return bboxes
|
435
|
-
|
436
|
-
|
437
|
-
def rbox2delta(proposals, gt, means=[0, 0, 0, 0, 0], stds=[1, 1, 1, 1, 1]):
|
438
|
-
"""
|
439
|
-
|
440
|
-
Args:
|
441
|
-
proposals:
|
442
|
-
gt:
|
443
|
-
means: 1x5
|
444
|
-
stds: 1x5
|
445
|
-
|
446
|
-
Returns:
|
447
|
-
|
448
|
-
"""
|
449
|
-
proposals = proposals.astype(np.float64)
|
450
|
-
|
451
|
-
PI = np.pi
|
452
|
-
|
453
|
-
gt_widths = gt[..., 2]
|
454
|
-
gt_heights = gt[..., 3]
|
455
|
-
gt_angle = gt[..., 4]
|
456
|
-
|
457
|
-
proposals_widths = proposals[..., 2]
|
458
|
-
proposals_heights = proposals[..., 3]
|
459
|
-
proposals_angle = proposals[..., 4]
|
460
|
-
|
461
|
-
coord = gt[..., 0:2] - proposals[..., 0:2]
|
462
|
-
dx = (np.cos(proposals[..., 4]) * coord[..., 0] + np.sin(proposals[..., 4])
|
463
|
-
* coord[..., 1]) / proposals_widths
|
464
|
-
dy = (-np.sin(proposals[..., 4]) * coord[..., 0] +
|
465
|
-
np.cos(proposals[..., 4]) * coord[..., 1]) / proposals_heights
|
466
|
-
dw = np.log(gt_widths / proposals_widths)
|
467
|
-
dh = np.log(gt_heights / proposals_heights)
|
468
|
-
da = (gt_angle - proposals_angle)
|
469
|
-
|
470
|
-
da = (da + PI / 4) % PI - PI / 4
|
471
|
-
da /= PI
|
472
|
-
|
473
|
-
deltas = np.stack([dx, dy, dw, dh, da], axis=-1)
|
474
|
-
means = np.array(means, dtype=deltas.dtype)
|
475
|
-
stds = np.array(stds, dtype=deltas.dtype)
|
476
|
-
deltas = (deltas - means) / stds
|
477
|
-
deltas = deltas.astype(np.float32)
|
478
|
-
return deltas
|
479
|
-
|
480
|
-
|
481
|
-
def bbox_decode(bbox_preds,
|
482
|
-
anchors,
|
483
|
-
means=[0, 0, 0, 0, 0],
|
484
|
-
stds=[1, 1, 1, 1, 1]):
|
485
|
-
"""decode bbox from deltas
|
486
|
-
Args:
|
487
|
-
bbox_preds: [N,H,W,5]
|
488
|
-
anchors: [H*W,5]
|
489
|
-
return:
|
490
|
-
bboxes: [N,H,W,5]
|
491
|
-
"""
|
492
|
-
means = paddle.to_tensor(means)
|
493
|
-
stds = paddle.to_tensor(stds)
|
494
|
-
num_imgs, H, W, _ = bbox_preds.shape
|
495
|
-
bboxes_list = []
|
496
|
-
for img_id in range(num_imgs):
|
497
|
-
bbox_pred = bbox_preds[img_id]
|
498
|
-
# bbox_pred.shape=[5,H,W]
|
499
|
-
bbox_delta = bbox_pred
|
500
|
-
anchors = paddle.to_tensor(anchors)
|
501
|
-
bboxes = delta2rbox(
|
502
|
-
anchors, bbox_delta, means, stds, wh_ratio_clip=1e-6)
|
503
|
-
bboxes = paddle.reshape(bboxes, [H, W, 5])
|
504
|
-
bboxes_list.append(bboxes)
|
505
|
-
return paddle.stack(bboxes_list, axis=0)
|
506
|
-
|
507
|
-
|
508
|
-
def poly2rbox(polys):
|
509
|
-
"""
|
510
|
-
poly:[x0,y0,x1,y1,x2,y2,x3,y3]
|
511
|
-
to
|
512
|
-
rotated_boxes:[x_ctr,y_ctr,w,h,angle]
|
513
|
-
"""
|
514
|
-
rotated_boxes = []
|
515
|
-
for poly in polys:
|
516
|
-
poly = np.array(poly[:8], dtype=np.float32)
|
517
|
-
|
518
|
-
pt1 = (poly[0], poly[1])
|
519
|
-
pt2 = (poly[2], poly[3])
|
520
|
-
pt3 = (poly[4], poly[5])
|
521
|
-
pt4 = (poly[6], poly[7])
|
522
|
-
|
523
|
-
edge1 = np.sqrt((pt1[0] - pt2[0]) * (pt1[0] - pt2[0]) + (pt1[1] - pt2[
|
524
|
-
1]) * (pt1[1] - pt2[1]))
|
525
|
-
edge2 = np.sqrt((pt2[0] - pt3[0]) * (pt2[0] - pt3[0]) + (pt2[1] - pt3[
|
526
|
-
1]) * (pt2[1] - pt3[1]))
|
527
|
-
|
528
|
-
width = max(edge1, edge2)
|
529
|
-
height = min(edge1, edge2)
|
530
|
-
|
531
|
-
rbox_angle = 0
|
532
|
-
if edge1 > edge2:
|
533
|
-
rbox_angle = np.arctan2(
|
534
|
-
float(pt2[1] - pt1[1]), float(pt2[0] - pt1[0]))
|
535
|
-
elif edge2 >= edge1:
|
536
|
-
rbox_angle = np.arctan2(
|
537
|
-
float(pt4[1] - pt1[1]), float(pt4[0] - pt1[0]))
|
538
|
-
|
539
|
-
def norm_angle(angle, range=[-np.pi / 4, np.pi]):
|
540
|
-
return (angle - range[0]) % range[1] + range[0]
|
541
|
-
|
542
|
-
rbox_angle = norm_angle(rbox_angle)
|
543
|
-
|
544
|
-
x_ctr = float(pt1[0] + pt3[0]) / 2
|
545
|
-
y_ctr = float(pt1[1] + pt3[1]) / 2
|
546
|
-
rotated_box = np.array([x_ctr, y_ctr, width, height, rbox_angle])
|
547
|
-
rotated_boxes.append(rotated_box)
|
548
|
-
ret_rotated_boxes = np.array(rotated_boxes)
|
549
|
-
assert ret_rotated_boxes.shape[1] == 5
|
550
|
-
return ret_rotated_boxes
|
551
|
-
|
552
|
-
|
553
|
-
def cal_line_length(point1, point2):
|
554
|
-
import math
|
555
|
-
return math.sqrt(
|
556
|
-
math.pow(point1[0] - point2[0], 2) + math.pow(point1[1] - point2[1],
|
557
|
-
2))
|
558
|
-
|
559
|
-
|
560
|
-
def get_best_begin_point_single(coordinate):
|
561
|
-
x1, y1, x2, y2, x3, y3, x4, y4 = coordinate
|
562
|
-
xmin = min(x1, x2, x3, x4)
|
563
|
-
ymin = min(y1, y2, y3, y4)
|
564
|
-
xmax = max(x1, x2, x3, x4)
|
565
|
-
ymax = max(y1, y2, y3, y4)
|
566
|
-
combinate = [[[x1, y1], [x2, y2], [x3, y3], [x4, y4]],
|
567
|
-
[[x4, y4], [x1, y1], [x2, y2], [x3, y3]],
|
568
|
-
[[x3, y3], [x4, y4], [x1, y1], [x2, y2]],
|
569
|
-
[[x2, y2], [x3, y3], [x4, y4], [x1, y1]]]
|
570
|
-
dst_coordinate = [[xmin, ymin], [xmax, ymin], [xmax, ymax], [xmin, ymax]]
|
571
|
-
force = 100000000.0
|
572
|
-
force_flag = 0
|
573
|
-
for i in range(4):
|
574
|
-
temp_force = cal_line_length(combinate[i][0], dst_coordinate[0]) \
|
575
|
-
+ cal_line_length(combinate[i][1], dst_coordinate[1]) \
|
576
|
-
+ cal_line_length(combinate[i][2], dst_coordinate[2]) \
|
577
|
-
+ cal_line_length(combinate[i][3], dst_coordinate[3])
|
578
|
-
if temp_force < force:
|
579
|
-
force = temp_force
|
580
|
-
force_flag = i
|
581
|
-
if force_flag != 0:
|
582
|
-
pass
|
583
|
-
return np.array(combinate[force_flag]).reshape(8)
|
584
|
-
|
585
|
-
|
586
|
-
def rbox2poly_np(rrects):
|
587
|
-
"""
|
588
|
-
rrect:[x_ctr,y_ctr,w,h,angle]
|
589
|
-
to
|
590
|
-
poly:[x0,y0,x1,y1,x2,y2,x3,y3]
|
591
|
-
"""
|
592
|
-
polys = []
|
593
|
-
for i in range(rrects.shape[0]):
|
594
|
-
rrect = rrects[i]
|
595
|
-
# x_ctr, y_ctr, width, height, angle = rrect[:5]
|
596
|
-
x_ctr = rrect[0]
|
597
|
-
y_ctr = rrect[1]
|
598
|
-
width = rrect[2]
|
599
|
-
height = rrect[3]
|
600
|
-
angle = rrect[4]
|
601
|
-
tl_x, tl_y, br_x, br_y = -width / 2, -height / 2, width / 2, height / 2
|
602
|
-
rect = np.array([[tl_x, br_x, br_x, tl_x], [tl_y, tl_y, br_y, br_y]])
|
603
|
-
R = np.array([[np.cos(angle), -np.sin(angle)],
|
604
|
-
[np.sin(angle), np.cos(angle)]])
|
605
|
-
poly = R.dot(rect)
|
606
|
-
x0, x1, x2, x3 = poly[0, :4] + x_ctr
|
607
|
-
y0, y1, y2, y3 = poly[1, :4] + y_ctr
|
608
|
-
poly = np.array([x0, y0, x1, y1, x2, y2, x3, y3], dtype=np.float32)
|
609
|
-
poly = get_best_begin_point_single(poly)
|
610
|
-
polys.append(poly)
|
611
|
-
polys = np.array(polys)
|
612
|
-
return polys
|
613
|
-
|
614
|
-
|
615
|
-
def rbox2poly(rrects):
|
616
|
-
"""
|
617
|
-
rrect:[x_ctr,y_ctr,w,h,angle]
|
618
|
-
to
|
619
|
-
poly:[x0,y0,x1,y1,x2,y2,x3,y3]
|
620
|
-
"""
|
621
|
-
N = paddle.shape(rrects)[0]
|
622
|
-
|
623
|
-
x_ctr = rrects[:, 0]
|
624
|
-
y_ctr = rrects[:, 1]
|
625
|
-
width = rrects[:, 2]
|
626
|
-
height = rrects[:, 3]
|
627
|
-
angle = rrects[:, 4]
|
628
|
-
|
629
|
-
tl_x, tl_y, br_x, br_y = -width * 0.5, -height * 0.5, width * 0.5, height * 0.5
|
630
|
-
|
631
|
-
normal_rects = paddle.stack(
|
632
|
-
[tl_x, br_x, br_x, tl_x, tl_y, tl_y, br_y, br_y], axis=0)
|
633
|
-
normal_rects = paddle.reshape(normal_rects, [2, 4, N])
|
634
|
-
normal_rects = paddle.transpose(normal_rects, [2, 0, 1])
|
635
|
-
|
636
|
-
sin, cos = paddle.sin(angle), paddle.cos(angle)
|
637
|
-
# M.shape=[N,2,2]
|
638
|
-
M = paddle.stack([cos, -sin, sin, cos], axis=0)
|
639
|
-
M = paddle.reshape(M, [2, 2, N])
|
640
|
-
M = paddle.transpose(M, [2, 0, 1])
|
641
|
-
|
642
|
-
# polys:[N,8]
|
643
|
-
polys = paddle.matmul(M, normal_rects)
|
644
|
-
polys = paddle.transpose(polys, [2, 1, 0])
|
645
|
-
polys = paddle.reshape(polys, [-1, N])
|
646
|
-
polys = paddle.transpose(polys, [1, 0])
|
647
|
-
|
648
|
-
tmp = paddle.stack(
|
649
|
-
[x_ctr, y_ctr, x_ctr, y_ctr, x_ctr, y_ctr, x_ctr, y_ctr], axis=1)
|
650
|
-
polys = polys + tmp
|
651
|
-
return polys
|
652
|
-
|
653
|
-
|
654
|
-
def bbox_iou_np_expand(box1, box2, x1y1x2y2=True, eps=1e-16):
|
655
|
-
"""
|
656
|
-
Calculate the iou of box1 and box2 with numpy.
|
657
|
-
|
658
|
-
Args:
|
659
|
-
box1 (ndarray): [N, 4]
|
660
|
-
box2 (ndarray): [M, 4], usually N != M
|
661
|
-
x1y1x2y2 (bool): whether in x1y1x2y2 stype, default True
|
662
|
-
eps (float): epsilon to avoid divide by zero
|
663
|
-
Return:
|
664
|
-
iou (ndarray): iou of box1 and box2, [N, M]
|
665
|
-
"""
|
666
|
-
N, M = len(box1), len(box2) # usually N != M
|
667
|
-
if x1y1x2y2:
|
668
|
-
b1_x1, b1_y1 = box1[:, 0], box1[:, 1]
|
669
|
-
b1_x2, b1_y2 = box1[:, 2], box1[:, 3]
|
670
|
-
b2_x1, b2_y1 = box2[:, 0], box2[:, 1]
|
671
|
-
b2_x2, b2_y2 = box2[:, 2], box2[:, 3]
|
672
|
-
else:
|
673
|
-
# cxcywh style
|
674
|
-
# Transform from center and width to exact coordinates
|
675
|
-
b1_x1, b1_x2 = box1[:, 0] - box1[:, 2] / 2, box1[:, 0] + box1[:, 2] / 2
|
676
|
-
b1_y1, b1_y2 = box1[:, 1] - box1[:, 3] / 2, box1[:, 1] + box1[:, 3] / 2
|
677
|
-
b2_x1, b2_x2 = box2[:, 0] - box2[:, 2] / 2, box2[:, 0] + box2[:, 2] / 2
|
678
|
-
b2_y1, b2_y2 = box2[:, 1] - box2[:, 3] / 2, box2[:, 1] + box2[:, 3] / 2
|
679
|
-
|
680
|
-
# get the coordinates of the intersection rectangle
|
681
|
-
inter_rect_x1 = np.zeros((N, M), dtype=np.float32)
|
682
|
-
inter_rect_y1 = np.zeros((N, M), dtype=np.float32)
|
683
|
-
inter_rect_x2 = np.zeros((N, M), dtype=np.float32)
|
684
|
-
inter_rect_y2 = np.zeros((N, M), dtype=np.float32)
|
685
|
-
for i in range(len(box2)):
|
686
|
-
inter_rect_x1[:, i] = np.maximum(b1_x1, b2_x1[i])
|
687
|
-
inter_rect_y1[:, i] = np.maximum(b1_y1, b2_y1[i])
|
688
|
-
inter_rect_x2[:, i] = np.minimum(b1_x2, b2_x2[i])
|
689
|
-
inter_rect_y2[:, i] = np.minimum(b1_y2, b2_y2[i])
|
690
|
-
# Intersection area
|
691
|
-
inter_area = np.maximum(inter_rect_x2 - inter_rect_x1, 0) * np.maximum(
|
692
|
-
inter_rect_y2 - inter_rect_y1, 0)
|
693
|
-
# Union Area
|
694
|
-
b1_area = np.repeat(
|
695
|
-
((b1_x2 - b1_x1) * (b1_y2 - b1_y1)).reshape(-1, 1), M, axis=-1)
|
696
|
-
b2_area = np.repeat(
|
697
|
-
((b2_x2 - b2_x1) * (b2_y2 - b2_y1)).reshape(1, -1), N, axis=0)
|
698
|
-
|
699
|
-
ious = inter_area / (b1_area + b2_area - inter_area + eps)
|
700
|
-
return ious
|
701
|
-
|
702
|
-
|
703
|
-
def bbox2distance(points, bbox, max_dis=None, eps=0.1):
|
704
|
-
"""Decode bounding box based on distances.
|
705
|
-
Args:
|
706
|
-
points (Tensor): Shape (n, 2), [x, y].
|
707
|
-
bbox (Tensor): Shape (n, 4), "xyxy" format
|
708
|
-
max_dis (float): Upper bound of the distance.
|
709
|
-
eps (float): a small value to ensure target < max_dis, instead <=
|
710
|
-
Returns:
|
711
|
-
Tensor: Decoded distances.
|
712
|
-
"""
|
713
|
-
left = points[:, 0] - bbox[:, 0]
|
714
|
-
top = points[:, 1] - bbox[:, 1]
|
715
|
-
right = bbox[:, 2] - points[:, 0]
|
716
|
-
bottom = bbox[:, 3] - points[:, 1]
|
717
|
-
if max_dis is not None:
|
718
|
-
left = left.clip(min=0, max=max_dis - eps)
|
719
|
-
top = top.clip(min=0, max=max_dis - eps)
|
720
|
-
right = right.clip(min=0, max=max_dis - eps)
|
721
|
-
bottom = bottom.clip(min=0, max=max_dis - eps)
|
722
|
-
return paddle.stack([left, top, right, bottom], -1)
|
723
|
-
|
724
|
-
|
725
|
-
def distance2bbox(points, distance, max_shape=None):
|
726
|
-
"""Decode distance prediction to bounding box.
|
727
|
-
Args:
|
728
|
-
points (Tensor): Shape (n, 2), [x, y].
|
729
|
-
distance (Tensor): Distance from the given point to 4
|
730
|
-
boundaries (left, top, right, bottom).
|
731
|
-
max_shape (tuple): Shape of the image.
|
732
|
-
Returns:
|
733
|
-
Tensor: Decoded bboxes.
|
734
|
-
"""
|
735
|
-
x1 = points[:, 0] - distance[:, 0]
|
736
|
-
y1 = points[:, 1] - distance[:, 1]
|
737
|
-
x2 = points[:, 0] + distance[:, 2]
|
738
|
-
y2 = points[:, 1] + distance[:, 3]
|
739
|
-
if max_shape is not None:
|
740
|
-
x1 = x1.clip(min=0, max=max_shape[1])
|
741
|
-
y1 = y1.clip(min=0, max=max_shape[0])
|
742
|
-
x2 = x2.clip(min=0, max=max_shape[1])
|
743
|
-
y2 = y2.clip(min=0, max=max_shape[0])
|
744
|
-
return paddle.stack([x1, y1, x2, y2], -1)
|
745
|
-
|
746
|
-
|
747
|
-
def bbox_center(boxes):
|
748
|
-
"""Get bbox centers from boxes.
|
749
|
-
Args:
|
750
|
-
boxes (Tensor): boxes with shape (N, 4), "xmin, ymin, xmax, ymax" format.
|
751
|
-
Returns:
|
752
|
-
Tensor: boxes centers with shape (N, 2), "cx, cy" format.
|
753
|
-
"""
|
754
|
-
boxes_cx = (boxes[..., 0] + boxes[..., 2]) / 2
|
755
|
-
boxes_cy = (boxes[..., 1] + boxes[..., 3]) / 2
|
756
|
-
return paddle.stack([boxes_cx, boxes_cy], axis=-1)
|
757
|
-
|
758
|
-
|
759
|
-
def batch_distance2bbox(points, distance, max_shapes=None):
|
760
|
-
"""Decode distance prediction to bounding box for batch.
|
761
|
-
Args:
|
762
|
-
points (Tensor): [B, ..., 2]
|
763
|
-
distance (Tensor): [B, ..., 4]
|
764
|
-
max_shapes (tuple): [B, 2], "h,w" format, Shape of the image.
|
765
|
-
Returns:
|
766
|
-
Tensor: Decoded bboxes.
|
767
|
-
"""
|
768
|
-
x1 = points[..., 0] - distance[..., 0]
|
769
|
-
y1 = points[..., 1] - distance[..., 1]
|
770
|
-
x2 = points[..., 0] + distance[..., 2]
|
771
|
-
y2 = points[..., 1] + distance[..., 3]
|
772
|
-
if max_shapes is not None:
|
773
|
-
for i, max_shape in enumerate(max_shapes):
|
774
|
-
x1[i] = x1[i].clip(min=0, max=max_shape[1])
|
775
|
-
y1[i] = y1[i].clip(min=0, max=max_shape[0])
|
776
|
-
x2[i] = x2[i].clip(min=0, max=max_shape[1])
|
777
|
-
y2[i] = y2[i].clip(min=0, max=max_shape[0])
|
778
|
-
return paddle.stack([x1, y1, x2, y2], -1)
|