paddlex 2.0.0rc4__py3-none-any.whl → 3.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- paddlex/.version +1 -0
- paddlex/__init__.py +35 -18
- paddlex/__main__.py +39 -0
- paddlex/configs/modules/3d_bev_detection/BEVFusion.yaml +38 -0
- paddlex/configs/modules/chart_parsing/PP-Chart2Table.yaml +13 -0
- paddlex/configs/modules/doc_text_orientation/PP-LCNet_x1_0_doc_ori.yaml +41 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-2B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee-7B.yaml +14 -0
- paddlex/configs/modules/doc_vlm/PP-DocBee2-3B.yaml +14 -0
- paddlex/configs/modules/face_detection/BlazeFace-FPN-SSH.yaml +40 -0
- paddlex/configs/modules/face_detection/BlazeFace.yaml +40 -0
- paddlex/configs/modules/face_detection/PP-YOLOE_plus-S_face.yaml +40 -0
- paddlex/configs/modules/face_detection/PicoDet_LCNet_x2_5_face.yaml +40 -0
- paddlex/configs/modules/face_feature/MobileFaceNet.yaml +41 -0
- paddlex/configs/modules/face_feature/ResNet50_face.yaml +41 -0
- paddlex/configs/modules/formula_recognition/LaTeX_OCR_rec.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-L.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-M.yaml +40 -0
- paddlex/configs/modules/formula_recognition/PP-FormulaNet_plus-S.yaml +40 -0
- paddlex/configs/modules/formula_recognition/UniMERNet.yaml +40 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-L_human.yaml +42 -0
- paddlex/configs/modules/human_detection/PP-YOLOE-S_human.yaml +42 -0
- paddlex/configs/modules/image_anomaly_detection/STFPM.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_base_patch16_224.yaml +41 -0
- paddlex/configs/modules/image_classification/CLIP_vit_large_patch14_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_base_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_224.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_large_384.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_small.yaml +41 -0
- paddlex/configs/modules/image_classification/ConvNeXt_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/FasterNet-L.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-M.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-S.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T0.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T1.yaml +40 -0
- paddlex/configs/modules/image_classification/FasterNet-T2.yaml +40 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV1_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV2_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_large_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV3_small_x1_25.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_conv_small.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_large.yaml +41 -0
- paddlex/configs/modules/image_classification/MobileNetV4_hybrid_medium.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B1.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B2.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B3.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B4.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNetV2-B6.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-HGNet_tiny.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_base.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_large.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNetV2_small.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_25.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_35.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x0_75.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x1_5.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_0.yaml +41 -0
- paddlex/configs/modules/image_classification/PP-LCNet_x2_5.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet101_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet152_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet18_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet200_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet34_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50.yaml +41 -0
- paddlex/configs/modules/image_classification/ResNet50_vd.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S1.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S2.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S3.yaml +41 -0
- paddlex/configs/modules/image_classification/StarNet-S4.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_base_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window12_384.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_large_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_small_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_classification/SwinTransformer_tiny_patch4_window7_224.yaml +41 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_base.yaml +42 -0
- paddlex/configs/modules/image_feature/PP-ShiTuV2_rec_CLIP_vit_large.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/CLIP_vit_base_patch16_448_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B4_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-HGNetV2-B6_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/PP-LCNet_x1_0_ML.yaml +41 -0
- paddlex/configs/modules/image_multilabel_classification/ResNet50_ML.yaml +41 -0
- paddlex/configs/modules/image_unwarping/UVDoc.yaml +12 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-M.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/Mask-RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNeXt101-vd-FPN.yaml +39 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet101-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50-vd-FPN.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/MaskRCNN-ResNet50.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/PP-YOLOE_seg-S.yaml +40 -0
- paddlex/configs/modules/instance_segmentation/SOLOv2.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_128x96.yaml +40 -0
- paddlex/configs/modules/keypoint_detection/PP-TinyPose_256x192.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocBlockLayout.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-M.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout-S.yaml +40 -0
- paddlex/configs/modules/layout_detection/PP-DocLayout_plus-L.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-L_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet-S_layout_3cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x.yaml +40 -0
- paddlex/configs/modules/layout_detection/PicoDet_layout_1x_table.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_17cls.yaml +40 -0
- paddlex/configs/modules/layout_detection/RT-DETR-H_layout_3cls.yaml +40 -0
- paddlex/configs/modules/mainbody_detection/PP-ShiTuV2_det.yaml +41 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_base.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_large.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_medium.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_small.yaml +12 -0
- paddlex/configs/modules/multilingual_speech_recognition/whisper_tiny.yaml +12 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-FPN.yaml +41 -0
- paddlex/configs/modules/object_detection/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/CenterNet-DLA-34.yaml +41 -0
- paddlex/configs/modules/object_detection/CenterNet-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/Co-DINO-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-DINO-Swin-L.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/Co-Deformable-DETR-Swin-T.yaml +40 -0
- paddlex/configs/modules/object_detection/DETR-R50.yaml +42 -0
- paddlex/configs/modules/object_detection/FCOS-ResNet50.yaml +41 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNeXt101-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet101.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet34-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-ResNet50.yaml +42 -0
- paddlex/configs/modules/object_detection/FasterRCNN-Swin-Tiny-FPN.yaml +42 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-M.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PP-YOLOE_plus-X.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-L.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-M.yaml +42 -0
- paddlex/configs/modules/object_detection/PicoDet-S.yaml +40 -0
- paddlex/configs/modules/object_detection/PicoDet-XS.yaml +42 -0
- paddlex/configs/modules/object_detection/RT-DETR-H.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-L.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R18.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-R50.yaml +40 -0
- paddlex/configs/modules/object_detection/RT-DETR-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-L.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-M.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-N.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-S.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-T.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOX-X.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-DarkNet53.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-MobileNetV3.yaml +40 -0
- paddlex/configs/modules/object_detection/YOLOv3-ResNet50_vd_DCN.yaml +40 -0
- paddlex/configs/modules/open_vocabulary_detection/GroundingDINO-T.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_detection/YOLO-Worldv2-L.yaml +13 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_box.yaml +17 -0
- paddlex/configs/modules/open_vocabulary_segmentation/SAM-H_point.yaml +15 -0
- paddlex/configs/modules/pedestrian_attribute_recognition/PP-LCNet_x1_0_pedestrian_attribute.yaml +41 -0
- paddlex/configs/modules/rotated_object_detection/PP-YOLOE-R-L.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_mobile_seal_det.yaml +40 -0
- paddlex/configs/modules/seal_text_detection/PP-OCRv4_server_seal_det.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R101.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/Deeplabv3_Plus-R50.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_small.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/MaskFormer_tiny.yaml +42 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W18.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/OCRNet_HRNet-W48.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-B.yaml +41 -0
- paddlex/configs/modules/semantic_segmentation/PP-LiteSeg-T.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_base.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_large.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_small.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SeaFormer_tiny.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B0.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B1.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B2.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B3.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B4.yaml +40 -0
- paddlex/configs/modules/semantic_segmentation/SegFormer-B5.yaml +40 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-L.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-S.yaml +42 -0
- paddlex/configs/modules/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml +42 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wired_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_cells_detection/RT-DETR-L_wireless_table_cell_det.yaml +40 -0
- paddlex/configs/modules/table_classification/PP-LCNet_x1_0_table_cls.yaml +41 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wired.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANeXt_wireless.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet.yaml +39 -0
- paddlex/configs/modules/table_structure_recognition/SLANet_plus.yaml +39 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv3_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv4_server_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_mobile_det.yaml +40 -0
- paddlex/configs/modules/text_detection/PP-OCRv5_server_det.yaml +40 -0
- paddlex/configs/modules/text_recognition/PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv4_server_rec_doc.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/PP-OCRv5_server_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/arabic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_RepSVTR_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ch_SVTRv2_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/chinese_cht_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/cyrillic_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/devanagari_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/en_PP-OCRv4_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/japan_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ka_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/korean_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/latin_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/ta_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/text_recognition/te_PP-OCRv3_mobile_rec.yaml +39 -0
- paddlex/configs/modules/textline_orientation/PP-LCNet_x0_25_textline_ori.yaml +41 -0
- paddlex/configs/modules/ts_anomaly_detection/AutoEncoder_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/DLinear_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/Nonstationary_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/PatchTST_ad.yaml +37 -0
- paddlex/configs/modules/ts_anomaly_detection/TimesNet_ad.yaml +37 -0
- paddlex/configs/modules/ts_classification/TimesNet_cls.yaml +37 -0
- paddlex/configs/modules/ts_forecast/DLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/NLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/Nonstationary.yaml +38 -0
- paddlex/configs/modules/ts_forecast/PatchTST.yaml +38 -0
- paddlex/configs/modules/ts_forecast/RLinear.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TiDE.yaml +38 -0
- paddlex/configs/modules/ts_forecast/TimesNet.yaml +38 -0
- paddlex/configs/modules/vehicle_attribute_recognition/PP-LCNet_x1_0_vehicle_attribute.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-L_vehicle.yaml +41 -0
- paddlex/configs/modules/vehicle_detection/PP-YOLOE-S_vehicle.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSM-R50_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_16frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_classification/PP-TSMv2-LCNetV2_8frames_uniform.yaml +42 -0
- paddlex/configs/modules/video_detection/YOWO.yaml +40 -0
- paddlex/configs/pipelines/3d_bev_detection.yaml +9 -0
- paddlex/configs/pipelines/OCR.yaml +45 -0
- paddlex/configs/pipelines/PP-ChatOCRv3-doc.yaml +151 -0
- paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml +237 -0
- paddlex/configs/pipelines/PP-ShiTuV2.yaml +18 -0
- paddlex/configs/pipelines/PP-StructureV3.yaml +226 -0
- paddlex/configs/pipelines/anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/doc_preprocessor.yaml +15 -0
- paddlex/configs/pipelines/doc_understanding.yaml +9 -0
- paddlex/configs/pipelines/face_recognition.yaml +18 -0
- paddlex/configs/pipelines/formula_recognition.yaml +39 -0
- paddlex/configs/pipelines/human_keypoint_detection.yaml +17 -0
- paddlex/configs/pipelines/image_classification.yaml +10 -0
- paddlex/configs/pipelines/image_multilabel_classification.yaml +9 -0
- paddlex/configs/pipelines/instance_segmentation.yaml +10 -0
- paddlex/configs/pipelines/layout_parsing.yaml +102 -0
- paddlex/configs/pipelines/multilingual_speech_recognition.yaml +9 -0
- paddlex/configs/pipelines/object_detection.yaml +10 -0
- paddlex/configs/pipelines/open_vocabulary_detection.yaml +12 -0
- paddlex/configs/pipelines/open_vocabulary_segmentation.yaml +13 -0
- paddlex/configs/pipelines/pedestrian_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/rotated_object_detection.yaml +10 -0
- paddlex/configs/pipelines/seal_recognition.yaml +52 -0
- paddlex/configs/pipelines/semantic_segmentation.yaml +10 -0
- paddlex/configs/pipelines/small_object_detection.yaml +10 -0
- paddlex/configs/pipelines/table_recognition.yaml +57 -0
- paddlex/configs/pipelines/table_recognition_v2.yaml +82 -0
- paddlex/configs/pipelines/ts_anomaly_detection.yaml +8 -0
- paddlex/configs/pipelines/ts_classification.yaml +8 -0
- paddlex/configs/pipelines/ts_forecast.yaml +8 -0
- paddlex/configs/pipelines/vehicle_attribute_recognition.yaml +15 -0
- paddlex/configs/pipelines/video_classification.yaml +9 -0
- paddlex/configs/pipelines/video_detection.yaml +10 -0
- paddlex/constants.py +17 -0
- paddlex/engine.py +56 -0
- paddlex/hpip_links.html +31 -0
- paddlex/inference/__init__.py +19 -0
- paddlex/inference/common/__init__.py +13 -0
- paddlex/inference/common/batch_sampler/__init__.py +21 -0
- paddlex/inference/common/batch_sampler/audio_batch_sampler.py +83 -0
- paddlex/inference/common/batch_sampler/base_batch_sampler.py +94 -0
- paddlex/inference/common/batch_sampler/det_3d_batch_sampler.py +144 -0
- paddlex/inference/common/batch_sampler/doc_vlm_batch_sampler.py +87 -0
- paddlex/inference/common/batch_sampler/image_batch_sampler.py +121 -0
- paddlex/inference/common/batch_sampler/ts_batch_sampler.py +109 -0
- paddlex/inference/common/batch_sampler/video_batch_sampler.py +74 -0
- paddlex/inference/common/reader/__init__.py +19 -0
- paddlex/inference/common/reader/audio_reader.py +46 -0
- paddlex/inference/common/reader/det_3d_reader.py +241 -0
- paddlex/inference/common/reader/image_reader.py +73 -0
- paddlex/inference/common/reader/ts_reader.py +46 -0
- paddlex/inference/common/reader/video_reader.py +42 -0
- paddlex/inference/common/result/__init__.py +29 -0
- paddlex/inference/common/result/base_cv_result.py +41 -0
- paddlex/inference/common/result/base_result.py +72 -0
- paddlex/inference/common/result/base_ts_result.py +41 -0
- paddlex/inference/common/result/base_video_result.py +36 -0
- paddlex/inference/common/result/mixin.py +709 -0
- paddlex/inference/models/__init__.py +86 -0
- paddlex/inference/models/anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/anomaly_detection/predictor.py +135 -0
- paddlex/inference/models/anomaly_detection/processors.py +53 -0
- paddlex/inference/models/anomaly_detection/result.py +71 -0
- paddlex/inference/models/base/__init__.py +15 -0
- paddlex/inference/models/base/predictor/__init__.py +15 -0
- paddlex/inference/models/base/predictor/base_predictor.py +414 -0
- paddlex/inference/models/common/__init__.py +26 -0
- paddlex/inference/models/common/static_infer.py +801 -0
- paddlex/inference/models/common/tokenizer/__init__.py +21 -0
- paddlex/inference/models/common/tokenizer/bert_tokenizer.py +655 -0
- paddlex/inference/models/common/tokenizer/clip_tokenizer.py +609 -0
- paddlex/inference/models/common/tokenizer/gpt_tokenizer.py +453 -0
- paddlex/inference/models/common/tokenizer/qwen2_5_tokenizer.py +112 -0
- paddlex/inference/models/common/tokenizer/qwen2_tokenizer.py +438 -0
- paddlex/inference/models/common/tokenizer/qwen_tokenizer.py +288 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils.py +2149 -0
- paddlex/inference/models/common/tokenizer/tokenizer_utils_base.py +3720 -0
- paddlex/inference/models/common/tokenizer/utils.py +66 -0
- paddlex/inference/models/common/tokenizer/vocab.py +647 -0
- paddlex/inference/models/common/ts/__init__.py +15 -0
- paddlex/inference/models/common/ts/funcs.py +540 -0
- paddlex/inference/models/common/ts/processors.py +322 -0
- paddlex/inference/models/common/vision/__init__.py +23 -0
- paddlex/inference/models/common/vision/funcs.py +98 -0
- paddlex/inference/models/common/vision/processors.py +285 -0
- paddlex/inference/models/common/vlm/__init__.py +13 -0
- paddlex/inference/models/common/vlm/activations.py +189 -0
- paddlex/inference/models/common/vlm/bert_padding.py +127 -0
- paddlex/inference/models/common/vlm/conversion_utils.py +99 -0
- paddlex/inference/models/common/vlm/distributed.py +229 -0
- paddlex/inference/models/common/vlm/flash_attn_utils.py +119 -0
- paddlex/inference/models/common/vlm/fusion_ops.py +205 -0
- paddlex/inference/models/common/vlm/generation/__init__.py +34 -0
- paddlex/inference/models/common/vlm/generation/configuration_utils.py +533 -0
- paddlex/inference/models/common/vlm/generation/logits_process.py +730 -0
- paddlex/inference/models/common/vlm/generation/stopping_criteria.py +106 -0
- paddlex/inference/models/common/vlm/generation/utils.py +2162 -0
- paddlex/inference/models/common/vlm/transformers/__init__.py +16 -0
- paddlex/inference/models/common/vlm/transformers/configuration_utils.py +1037 -0
- paddlex/inference/models/common/vlm/transformers/conversion_utils.py +408 -0
- paddlex/inference/models/common/vlm/transformers/model_outputs.py +1612 -0
- paddlex/inference/models/common/vlm/transformers/model_utils.py +2014 -0
- paddlex/inference/models/common/vlm/transformers/utils.py +178 -0
- paddlex/inference/models/common/vlm/utils.py +109 -0
- paddlex/inference/models/doc_vlm/__init__.py +15 -0
- paddlex/inference/models/doc_vlm/modeling/GOT_ocr_2_0.py +830 -0
- paddlex/inference/models/doc_vlm/modeling/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2.py +1606 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_5_vl.py +3006 -0
- paddlex/inference/models/doc_vlm/modeling/qwen2_vl.py +2495 -0
- paddlex/inference/models/doc_vlm/predictor.py +253 -0
- paddlex/inference/models/doc_vlm/processors/GOT_ocr_2_0.py +97 -0
- paddlex/inference/models/doc_vlm/processors/__init__.py +17 -0
- paddlex/inference/models/doc_vlm/processors/common.py +561 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_5_vl.py +548 -0
- paddlex/inference/models/doc_vlm/processors/qwen2_vl.py +543 -0
- paddlex/inference/models/doc_vlm/result.py +21 -0
- paddlex/inference/models/face_feature/__init__.py +15 -0
- paddlex/inference/models/face_feature/predictor.py +66 -0
- paddlex/inference/models/formula_recognition/__init__.py +15 -0
- paddlex/inference/models/formula_recognition/predictor.py +193 -0
- paddlex/inference/models/formula_recognition/processors.py +1015 -0
- paddlex/inference/models/formula_recognition/result.py +411 -0
- paddlex/inference/models/image_classification/__init__.py +15 -0
- paddlex/inference/models/image_classification/predictor.py +172 -0
- paddlex/inference/models/image_classification/processors.py +89 -0
- paddlex/inference/models/image_classification/result.py +93 -0
- paddlex/inference/models/image_feature/__init__.py +15 -0
- paddlex/inference/models/image_feature/predictor.py +146 -0
- paddlex/inference/models/image_feature/processors.py +31 -0
- paddlex/inference/models/image_feature/result.py +32 -0
- paddlex/inference/models/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/models/image_multilabel_classification/predictor.py +95 -0
- paddlex/inference/models/image_multilabel_classification/processors.py +89 -0
- paddlex/inference/models/image_multilabel_classification/result.py +96 -0
- paddlex/inference/models/image_unwarping/__init__.py +15 -0
- paddlex/inference/models/image_unwarping/predictor.py +97 -0
- paddlex/inference/models/image_unwarping/processors.py +92 -0
- paddlex/inference/models/image_unwarping/result.py +47 -0
- paddlex/inference/models/instance_segmentation/__init__.py +15 -0
- paddlex/inference/models/instance_segmentation/predictor.py +202 -0
- paddlex/inference/models/instance_segmentation/processors.py +102 -0
- paddlex/inference/models/instance_segmentation/result.py +162 -0
- paddlex/inference/models/keypoint_detection/__init__.py +15 -0
- paddlex/inference/models/keypoint_detection/predictor.py +190 -0
- paddlex/inference/models/keypoint_detection/processors.py +367 -0
- paddlex/inference/models/keypoint_detection/result.py +197 -0
- paddlex/inference/models/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/models/m_3d_bev_detection/predictor.py +303 -0
- paddlex/inference/models/m_3d_bev_detection/processors.py +990 -0
- paddlex/inference/models/m_3d_bev_detection/result.py +68 -0
- paddlex/inference/models/m_3d_bev_detection/visualizer_3d.py +169 -0
- paddlex/inference/models/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/models/multilingual_speech_recognition/predictor.py +137 -0
- paddlex/inference/models/multilingual_speech_recognition/processors.py +1933 -0
- paddlex/inference/models/multilingual_speech_recognition/result.py +21 -0
- paddlex/inference/models/object_detection/__init__.py +15 -0
- paddlex/inference/models/object_detection/predictor.py +344 -0
- paddlex/inference/models/object_detection/processors.py +885 -0
- paddlex/inference/models/object_detection/result.py +114 -0
- paddlex/inference/models/object_detection/utils.py +70 -0
- paddlex/inference/models/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_detection/predictor.py +172 -0
- paddlex/inference/models/open_vocabulary_detection/processors/__init__.py +16 -0
- paddlex/inference/models/open_vocabulary_detection/processors/common.py +114 -0
- paddlex/inference/models/open_vocabulary_detection/processors/groundingdino_processors.py +496 -0
- paddlex/inference/models/open_vocabulary_detection/processors/yoloworld_processors.py +209 -0
- paddlex/inference/models/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/predictor.py +113 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/processors/sam_processer.py +249 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/__init__.py +15 -0
- paddlex/inference/models/open_vocabulary_segmentation/results/sam_result.py +149 -0
- paddlex/inference/models/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/models/semantic_segmentation/predictor.py +158 -0
- paddlex/inference/models/semantic_segmentation/processors.py +117 -0
- paddlex/inference/models/semantic_segmentation/result.py +73 -0
- paddlex/inference/models/table_structure_recognition/__init__.py +15 -0
- paddlex/inference/models/table_structure_recognition/predictor.py +161 -0
- paddlex/inference/models/table_structure_recognition/processors.py +229 -0
- paddlex/inference/models/table_structure_recognition/result.py +63 -0
- paddlex/inference/models/text_detection/__init__.py +15 -0
- paddlex/inference/models/text_detection/predictor.py +191 -0
- paddlex/inference/models/text_detection/processors.py +538 -0
- paddlex/inference/models/text_detection/result.py +46 -0
- paddlex/inference/models/text_recognition/__init__.py +15 -0
- paddlex/inference/models/text_recognition/predictor.py +98 -0
- paddlex/inference/models/text_recognition/processors.py +245 -0
- paddlex/inference/models/text_recognition/result.py +76 -0
- paddlex/inference/models/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/models/ts_anomaly_detection/predictor.py +141 -0
- paddlex/inference/models/ts_anomaly_detection/processors.py +98 -0
- paddlex/inference/models/ts_anomaly_detection/result.py +83 -0
- paddlex/inference/models/ts_classification/__init__.py +15 -0
- paddlex/inference/models/ts_classification/predictor.py +122 -0
- paddlex/inference/models/ts_classification/processors.py +122 -0
- paddlex/inference/models/ts_classification/result.py +87 -0
- paddlex/inference/models/ts_forecasting/__init__.py +15 -0
- paddlex/inference/models/ts_forecasting/predictor.py +154 -0
- paddlex/inference/models/ts_forecasting/processors.py +158 -0
- paddlex/inference/models/ts_forecasting/result.py +96 -0
- paddlex/inference/models/video_classification/__init__.py +15 -0
- paddlex/inference/models/video_classification/predictor.py +141 -0
- paddlex/inference/models/video_classification/processors.py +409 -0
- paddlex/inference/models/video_classification/result.py +96 -0
- paddlex/inference/models/video_detection/__init__.py +15 -0
- paddlex/inference/models/video_detection/predictor.py +129 -0
- paddlex/inference/models/video_detection/processors.py +463 -0
- paddlex/inference/models/video_detection/result.py +109 -0
- paddlex/inference/pipelines/__init__.py +239 -0
- paddlex/inference/pipelines/_parallel.py +172 -0
- paddlex/inference/pipelines/anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/anomaly_detection/pipeline.py +82 -0
- paddlex/inference/pipelines/attribute_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/attribute_recognition/pipeline.py +120 -0
- paddlex/inference/pipelines/attribute_recognition/result.py +102 -0
- paddlex/inference/pipelines/base.py +156 -0
- paddlex/inference/pipelines/components/__init__.py +29 -0
- paddlex/inference/pipelines/components/chat_server/__init__.py +16 -0
- paddlex/inference/pipelines/components/chat_server/base.py +39 -0
- paddlex/inference/pipelines/components/chat_server/openai_bot_chat.py +236 -0
- paddlex/inference/pipelines/components/common/__init__.py +19 -0
- paddlex/inference/pipelines/components/common/base_operator.py +37 -0
- paddlex/inference/pipelines/components/common/base_result.py +66 -0
- paddlex/inference/pipelines/components/common/convert_points_and_boxes.py +45 -0
- paddlex/inference/pipelines/components/common/crop_image_regions.py +556 -0
- paddlex/inference/pipelines/components/common/seal_det_warp.py +972 -0
- paddlex/inference/pipelines/components/common/sort_boxes.py +85 -0
- paddlex/inference/pipelines/components/common/warp_image.py +50 -0
- paddlex/inference/pipelines/components/faisser.py +357 -0
- paddlex/inference/pipelines/components/prompt_engineering/__init__.py +16 -0
- paddlex/inference/pipelines/components/prompt_engineering/base.py +35 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_ensemble_prompt.py +128 -0
- paddlex/inference/pipelines/components/prompt_engineering/generate_kie_prompt.py +148 -0
- paddlex/inference/pipelines/components/retriever/__init__.py +16 -0
- paddlex/inference/pipelines/components/retriever/base.py +228 -0
- paddlex/inference/pipelines/components/retriever/openai_bot_retriever.py +70 -0
- paddlex/inference/pipelines/components/retriever/qianfan_bot_retriever.py +166 -0
- paddlex/inference/pipelines/components/utils/__init__.py +13 -0
- paddlex/inference/pipelines/components/utils/mixin.py +206 -0
- paddlex/inference/pipelines/doc_preprocessor/__init__.py +15 -0
- paddlex/inference/pipelines/doc_preprocessor/pipeline.py +209 -0
- paddlex/inference/pipelines/doc_preprocessor/result.py +98 -0
- paddlex/inference/pipelines/doc_understanding/__init__.py +15 -0
- paddlex/inference/pipelines/doc_understanding/pipeline.py +71 -0
- paddlex/inference/pipelines/face_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/face_recognition/pipeline.py +63 -0
- paddlex/inference/pipelines/face_recognition/result.py +44 -0
- paddlex/inference/pipelines/formula_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/formula_recognition/pipeline.py +347 -0
- paddlex/inference/pipelines/formula_recognition/result.py +282 -0
- paddlex/inference/pipelines/image_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_classification/pipeline.py +90 -0
- paddlex/inference/pipelines/image_multilabel_classification/__init__.py +15 -0
- paddlex/inference/pipelines/image_multilabel_classification/pipeline.py +97 -0
- paddlex/inference/pipelines/instance_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/instance_segmentation/pipeline.py +91 -0
- paddlex/inference/pipelines/keypoint_detection/__init__.py +15 -0
- paddlex/inference/pipelines/keypoint_detection/pipeline.py +158 -0
- paddlex/inference/pipelines/layout_parsing/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/pipeline.py +568 -0
- paddlex/inference/pipelines/layout_parsing/pipeline_v2.py +1382 -0
- paddlex/inference/pipelines/layout_parsing/result.py +191 -0
- paddlex/inference/pipelines/layout_parsing/result_v2.py +745 -0
- paddlex/inference/pipelines/layout_parsing/setting.py +87 -0
- paddlex/inference/pipelines/layout_parsing/utils.py +951 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/__init__.py +16 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/utils.py +1143 -0
- paddlex/inference/pipelines/layout_parsing/xycut_enhanced/xycuts.py +562 -0
- paddlex/inference/pipelines/m_3d_bev_detection/__init__.py +15 -0
- paddlex/inference/pipelines/m_3d_bev_detection/pipeline.py +74 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/multilingual_speech_recognition/pipeline.py +78 -0
- paddlex/inference/pipelines/object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/object_detection/pipeline.py +115 -0
- paddlex/inference/pipelines/ocr/__init__.py +15 -0
- paddlex/inference/pipelines/ocr/pipeline.py +463 -0
- paddlex/inference/pipelines/ocr/result.py +255 -0
- paddlex/inference/pipelines/open_vocabulary_detection/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_detection/pipeline.py +86 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/open_vocabulary_segmentation/pipeline.py +100 -0
- paddlex/inference/pipelines/pp_chatocr/__init__.py +16 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_base.py +111 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v3.py +781 -0
- paddlex/inference/pipelines/pp_chatocr/pipeline_v4.py +992 -0
- paddlex/inference/pipelines/pp_shitu_v2/__init__.py +15 -0
- paddlex/inference/pipelines/pp_shitu_v2/pipeline.py +156 -0
- paddlex/inference/pipelines/pp_shitu_v2/result.py +126 -0
- paddlex/inference/pipelines/rotated_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/rotated_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/seal_recognition/__init__.py +15 -0
- paddlex/inference/pipelines/seal_recognition/pipeline.py +335 -0
- paddlex/inference/pipelines/seal_recognition/result.py +89 -0
- paddlex/inference/pipelines/semantic_segmentation/__init__.py +15 -0
- paddlex/inference/pipelines/semantic_segmentation/pipeline.py +95 -0
- paddlex/inference/pipelines/small_object_detection/__init__.py +15 -0
- paddlex/inference/pipelines/small_object_detection/pipeline.py +95 -0
- paddlex/inference/pipelines/table_recognition/__init__.py +16 -0
- paddlex/inference/pipelines/table_recognition/pipeline.py +486 -0
- paddlex/inference/pipelines/table_recognition/pipeline_v2.py +1395 -0
- paddlex/inference/pipelines/table_recognition/result.py +218 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing.py +366 -0
- paddlex/inference/pipelines/table_recognition/table_recognition_post_processing_v2.py +488 -0
- paddlex/inference/pipelines/table_recognition/utils.py +44 -0
- paddlex/inference/pipelines/ts_anomaly_detection/__init__.py +15 -0
- paddlex/inference/pipelines/ts_anomaly_detection/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_classification/__init__.py +15 -0
- paddlex/inference/pipelines/ts_classification/pipeline.py +72 -0
- paddlex/inference/pipelines/ts_forecasting/__init__.py +15 -0
- paddlex/inference/pipelines/ts_forecasting/pipeline.py +72 -0
- paddlex/inference/pipelines/video_classification/__init__.py +15 -0
- paddlex/inference/pipelines/video_classification/pipeline.py +79 -0
- paddlex/inference/pipelines/video_detection/__init__.py +15 -0
- paddlex/inference/pipelines/video_detection/pipeline.py +86 -0
- paddlex/inference/serving/__init__.py +17 -0
- paddlex/inference/serving/basic_serving/__init__.py +18 -0
- paddlex/inference/serving/basic_serving/_app.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/__init__.py +44 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/__init__.py +13 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/common.py +104 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/image_recognition.py +36 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/_common/ocr.py +95 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/anomaly_detection.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_preprocessor.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/doc_understanding.py +153 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/face_recognition.py +226 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/formula_recognition.py +100 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/human_keypoint_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_classification.py +69 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/image_multilabel_classification.py +73 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/instance_segmentation.py +87 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/layout_parsing.py +117 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/m_3d_bev_detection.py +79 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/multilingual_speech_recognition.py +92 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/object_detection.py +77 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ocr.py +102 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/open_vocabulary_segmentation.py +91 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pedestrian_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv3_doc.py +193 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_chatocrv4_doc.py +223 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_shituv2.py +221 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/pp_structurev3.py +143 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/rotated_object_detection.py +81 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/seal_recognition.py +106 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/semantic_segmentation.py +67 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/small_object_detection.py +72 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition.py +108 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/table_recognition_v2.py +113 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_anomaly_detection.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_classification.py +64 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/ts_forecast.py +65 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/vehicle_attribute_recognition.py +84 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_classification.py +76 -0
- paddlex/inference/serving/basic_serving/_pipeline_apps/video_detection.py +92 -0
- paddlex/inference/serving/basic_serving/_server.py +40 -0
- paddlex/inference/serving/infra/__init__.py +13 -0
- paddlex/inference/serving/infra/config.py +36 -0
- paddlex/inference/serving/infra/models.py +79 -0
- paddlex/inference/serving/infra/storage.py +180 -0
- paddlex/inference/serving/infra/utils.py +285 -0
- paddlex/inference/serving/schemas/__init__.py +13 -0
- paddlex/inference/serving/schemas/anomaly_detection.py +39 -0
- paddlex/inference/serving/schemas/doc_preprocessor.py +54 -0
- paddlex/inference/serving/schemas/doc_understanding.py +78 -0
- paddlex/inference/serving/schemas/face_recognition.py +124 -0
- paddlex/inference/serving/schemas/formula_recognition.py +56 -0
- paddlex/inference/serving/schemas/human_keypoint_detection.py +55 -0
- paddlex/inference/serving/schemas/image_classification.py +45 -0
- paddlex/inference/serving/schemas/image_multilabel_classification.py +47 -0
- paddlex/inference/serving/schemas/instance_segmentation.py +53 -0
- paddlex/inference/serving/schemas/layout_parsing.py +71 -0
- paddlex/inference/serving/schemas/m_3d_bev_detection.py +48 -0
- paddlex/inference/serving/schemas/multilingual_speech_recognition.py +57 -0
- paddlex/inference/serving/schemas/object_detection.py +52 -0
- paddlex/inference/serving/schemas/ocr.py +60 -0
- paddlex/inference/serving/schemas/open_vocabulary_detection.py +52 -0
- paddlex/inference/serving/schemas/open_vocabulary_segmentation.py +52 -0
- paddlex/inference/serving/schemas/pedestrian_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/pp_chatocrv3_doc.py +133 -0
- paddlex/inference/serving/schemas/pp_chatocrv4_doc.py +150 -0
- paddlex/inference/serving/schemas/pp_shituv2.py +124 -0
- paddlex/inference/serving/schemas/pp_structurev3.py +88 -0
- paddlex/inference/serving/schemas/rotated_object_detection.py +52 -0
- paddlex/inference/serving/schemas/seal_recognition.py +62 -0
- paddlex/inference/serving/schemas/semantic_segmentation.py +45 -0
- paddlex/inference/serving/schemas/shared/__init__.py +13 -0
- paddlex/inference/serving/schemas/shared/classification.py +23 -0
- paddlex/inference/serving/schemas/shared/image_segmentation.py +28 -0
- paddlex/inference/serving/schemas/shared/object_detection.py +24 -0
- paddlex/inference/serving/schemas/shared/ocr.py +25 -0
- paddlex/inference/serving/schemas/small_object_detection.py +52 -0
- paddlex/inference/serving/schemas/table_recognition.py +64 -0
- paddlex/inference/serving/schemas/table_recognition_v2.py +69 -0
- paddlex/inference/serving/schemas/ts_anomaly_detection.py +37 -0
- paddlex/inference/serving/schemas/ts_classification.py +38 -0
- paddlex/inference/serving/schemas/ts_forecast.py +37 -0
- paddlex/inference/serving/schemas/vehicle_attribute_recognition.py +61 -0
- paddlex/inference/serving/schemas/video_classification.py +44 -0
- paddlex/inference/serving/schemas/video_detection.py +56 -0
- paddlex/inference/utils/__init__.py +13 -0
- paddlex/inference/utils/benchmark.py +379 -0
- paddlex/inference/utils/color_map.py +123 -0
- paddlex/inference/utils/get_pipeline_path.py +27 -0
- paddlex/inference/utils/hpi.py +254 -0
- paddlex/inference/utils/hpi_model_info_collection.json +2331 -0
- paddlex/inference/utils/io/__init__.py +36 -0
- paddlex/inference/utils/io/readers.py +504 -0
- paddlex/inference/utils/io/style.py +381 -0
- paddlex/inference/utils/io/tablepyxl.py +157 -0
- paddlex/inference/utils/io/writers.py +458 -0
- paddlex/inference/utils/model_paths.py +48 -0
- paddlex/inference/utils/new_ir_blocklist.py +27 -0
- paddlex/inference/utils/official_models.py +367 -0
- paddlex/inference/utils/pp_option.py +339 -0
- paddlex/inference/utils/trt_blocklist.py +43 -0
- paddlex/inference/utils/trt_config.py +420 -0
- paddlex/model.py +131 -0
- paddlex/modules/__init__.py +115 -0
- paddlex/modules/anomaly_detection/__init__.py +18 -0
- paddlex/modules/anomaly_detection/dataset_checker/__init__.py +94 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +233 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/anomaly_detection/dataset_checker/dataset_src/utils/visualizer.py +76 -0
- paddlex/modules/anomaly_detection/evaluator.py +58 -0
- paddlex/modules/anomaly_detection/exportor.py +22 -0
- paddlex/modules/anomaly_detection/model_list.py +16 -0
- paddlex/modules/anomaly_detection/trainer.py +70 -0
- paddlex/modules/base/__init__.py +18 -0
- paddlex/modules/base/build_model.py +33 -0
- paddlex/modules/base/dataset_checker/__init__.py +16 -0
- paddlex/modules/base/dataset_checker/dataset_checker.py +169 -0
- paddlex/modules/base/dataset_checker/utils.py +108 -0
- paddlex/modules/base/evaluator.py +170 -0
- paddlex/modules/base/exportor.py +145 -0
- paddlex/modules/base/trainer.py +144 -0
- paddlex/modules/base/utils/__init__.py +13 -0
- paddlex/modules/base/utils/cinn_setting.py +89 -0
- paddlex/modules/base/utils/coco_eval.py +94 -0
- paddlex/modules/base/utils/topk_eval.py +118 -0
- paddlex/modules/doc_vlm/__init__.py +18 -0
- paddlex/modules/doc_vlm/dataset_checker.py +29 -0
- paddlex/modules/doc_vlm/evaluator.py +29 -0
- paddlex/modules/doc_vlm/exportor.py +29 -0
- paddlex/modules/doc_vlm/model_list.py +16 -0
- paddlex/modules/doc_vlm/trainer.py +41 -0
- paddlex/modules/face_recognition/__init__.py +18 -0
- paddlex/modules/face_recognition/dataset_checker/__init__.py +71 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/__init__.py +16 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/check_dataset.py +172 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/face_recognition/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/face_recognition/evaluator.py +52 -0
- paddlex/modules/face_recognition/exportor.py +22 -0
- paddlex/modules/face_recognition/model_list.py +15 -0
- paddlex/modules/face_recognition/trainer.py +75 -0
- paddlex/modules/formula_recognition/__init__.py +18 -0
- paddlex/modules/formula_recognition/dataset_checker/__init__.py +113 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/analyse_dataset.py +158 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/check_dataset.py +76 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/formula_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/formula_recognition/evaluator.py +80 -0
- paddlex/modules/formula_recognition/exportor.py +22 -0
- paddlex/modules/formula_recognition/model_list.py +23 -0
- paddlex/modules/formula_recognition/trainer.py +123 -0
- paddlex/modules/general_recognition/__init__.py +18 -0
- paddlex/modules/general_recognition/dataset_checker/__init__.py +107 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/analyse_dataset.py +96 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/check_dataset.py +99 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/convert_dataset.py +100 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/general_recognition/dataset_checker/dataset_src/utils/visualizer.py +147 -0
- paddlex/modules/general_recognition/evaluator.py +31 -0
- paddlex/modules/general_recognition/exportor.py +22 -0
- paddlex/modules/general_recognition/model_list.py +19 -0
- paddlex/modules/general_recognition/trainer.py +52 -0
- paddlex/modules/image_classification/__init__.py +18 -0
- paddlex/modules/image_classification/dataset_checker/__init__.py +104 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/analyse_dataset.py +92 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/convert_dataset.py +51 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/image_classification/dataset_checker/dataset_src/utils/visualizer.py +153 -0
- paddlex/modules/image_classification/evaluator.py +43 -0
- paddlex/modules/image_classification/exportor.py +22 -0
- paddlex/modules/image_classification/model_list.py +99 -0
- paddlex/modules/image_classification/trainer.py +82 -0
- paddlex/modules/image_unwarping/__init__.py +13 -0
- paddlex/modules/image_unwarping/model_list.py +17 -0
- paddlex/modules/instance_segmentation/__init__.py +18 -0
- paddlex/modules/instance_segmentation/dataset_checker/__init__.py +107 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/check_dataset.py +95 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/convert_dataset.py +241 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/split_dataset.py +122 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/instance_segmentation/dataset_checker/dataset_src/utils/visualizer.py +223 -0
- paddlex/modules/instance_segmentation/evaluator.py +32 -0
- paddlex/modules/instance_segmentation/exportor.py +22 -0
- paddlex/modules/instance_segmentation/model_list.py +33 -0
- paddlex/modules/instance_segmentation/trainer.py +31 -0
- paddlex/modules/keypoint_detection/__init__.py +18 -0
- paddlex/modules/keypoint_detection/dataset_checker/__init__.py +56 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/__init__.py +15 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/keypoint_detection/dataset_checker/dataset_src/utils/visualizer.py +124 -0
- paddlex/modules/keypoint_detection/evaluator.py +41 -0
- paddlex/modules/keypoint_detection/exportor.py +22 -0
- paddlex/modules/keypoint_detection/model_list.py +16 -0
- paddlex/modules/keypoint_detection/trainer.py +39 -0
- paddlex/modules/m_3d_bev_detection/__init__.py +18 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/__init__.py +95 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/analyse_dataset.py +106 -0
- paddlex/modules/m_3d_bev_detection/dataset_checker/dataset_src/check_dataset.py +101 -0
- paddlex/modules/m_3d_bev_detection/evaluator.py +46 -0
- paddlex/modules/m_3d_bev_detection/exportor.py +22 -0
- paddlex/modules/m_3d_bev_detection/model_list.py +18 -0
- paddlex/modules/m_3d_bev_detection/trainer.py +68 -0
- paddlex/modules/multilabel_classification/__init__.py +18 -0
- paddlex/modules/multilabel_classification/dataset_checker/__init__.py +106 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/analyse_dataset.py +94 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/convert_dataset.py +120 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/split_dataset.py +81 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/visualizer.py +149 -0
- paddlex/modules/multilabel_classification/evaluator.py +43 -0
- paddlex/modules/multilabel_classification/exportor.py +22 -0
- paddlex/modules/multilabel_classification/model_list.py +24 -0
- paddlex/modules/multilabel_classification/trainer.py +85 -0
- paddlex/modules/multilingual_speech_recognition/__init__.py +18 -0
- paddlex/modules/multilingual_speech_recognition/dataset_checker.py +27 -0
- paddlex/modules/multilingual_speech_recognition/evaluator.py +27 -0
- paddlex/modules/multilingual_speech_recognition/exportor.py +27 -0
- paddlex/modules/multilingual_speech_recognition/model_list.py +22 -0
- paddlex/modules/multilingual_speech_recognition/trainer.py +42 -0
- paddlex/modules/object_detection/__init__.py +18 -0
- paddlex/modules/object_detection/dataset_checker/__init__.py +106 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/analyse_dataset.py +82 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/check_dataset.py +91 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/convert_dataset.py +438 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/split_dataset.py +123 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/object_detection/dataset_checker/dataset_src/utils/visualizer.py +193 -0
- paddlex/modules/object_detection/evaluator.py +57 -0
- paddlex/modules/object_detection/exportor.py +22 -0
- paddlex/modules/object_detection/model_list.py +86 -0
- paddlex/modules/object_detection/trainer.py +98 -0
- paddlex/modules/open_vocabulary_detection/__init__.py +18 -0
- paddlex/modules/open_vocabulary_detection/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_detection/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_detection/exportor.py +29 -0
- paddlex/modules/open_vocabulary_detection/model_list.py +16 -0
- paddlex/modules/open_vocabulary_detection/trainer.py +44 -0
- paddlex/modules/open_vocabulary_segmentation/__init__.py +18 -0
- paddlex/modules/open_vocabulary_segmentation/dataset_checker.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/evaluator.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/exportor.py +29 -0
- paddlex/modules/open_vocabulary_segmentation/model_list.py +19 -0
- paddlex/modules/open_vocabulary_segmentation/trainer.py +44 -0
- paddlex/modules/semantic_segmentation/__init__.py +18 -0
- paddlex/modules/semantic_segmentation/dataset_checker/__init__.py +109 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/analyse_dataset.py +76 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/check_dataset.py +80 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/convert_dataset.py +165 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/split_dataset.py +87 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/__init__.py +13 -0
- paddlex/modules/semantic_segmentation/dataset_checker/dataset_src/utils/visualizer.py +75 -0
- paddlex/modules/semantic_segmentation/evaluator.py +58 -0
- paddlex/modules/semantic_segmentation/exportor.py +31 -0
- paddlex/modules/semantic_segmentation/model_list.py +37 -0
- paddlex/modules/semantic_segmentation/trainer.py +72 -0
- paddlex/modules/table_recognition/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/__init__.py +98 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/analyse_dataset.py +59 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/check_dataset.py +87 -0
- paddlex/modules/table_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/table_recognition/evaluator.py +43 -0
- paddlex/modules/table_recognition/exportor.py +22 -0
- paddlex/modules/table_recognition/model_list.py +21 -0
- paddlex/modules/table_recognition/trainer.py +67 -0
- paddlex/modules/text_detection/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/__init__.py +107 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/analyse_dataset.py +220 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/check_dataset.py +106 -0
- paddlex/modules/text_detection/dataset_checker/dataset_src/split_dataset.py +140 -0
- paddlex/modules/text_detection/evaluator.py +41 -0
- paddlex/modules/text_detection/exportor.py +22 -0
- paddlex/modules/text_detection/model_list.py +26 -0
- paddlex/modules/text_detection/trainer.py +65 -0
- paddlex/modules/text_recognition/__init__.py +18 -0
- paddlex/modules/text_recognition/dataset_checker/__init__.py +125 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/analyse_dataset.py +162 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/check_dataset.py +104 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/convert_dataset.py +95 -0
- paddlex/modules/text_recognition/dataset_checker/dataset_src/split_dataset.py +80 -0
- paddlex/modules/text_recognition/evaluator.py +64 -0
- paddlex/modules/text_recognition/exportor.py +22 -0
- paddlex/modules/text_recognition/model_list.py +36 -0
- paddlex/modules/text_recognition/trainer.py +105 -0
- paddlex/modules/ts_anomaly_detection/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_anomaly_detection/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_anomaly_detection/evaluator.py +67 -0
- paddlex/modules/ts_anomaly_detection/exportor.py +44 -0
- paddlex/modules/ts_anomaly_detection/model_list.py +22 -0
- paddlex/modules/ts_anomaly_detection/trainer.py +113 -0
- paddlex/modules/ts_classification/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/analyse_dataset.py +77 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/convert_dataset.py +74 -0
- paddlex/modules/ts_classification/dataset_checker/dataset_src/split_dataset.py +88 -0
- paddlex/modules/ts_classification/evaluator.py +66 -0
- paddlex/modules/ts_classification/exportor.py +44 -0
- paddlex/modules/ts_classification/model_list.py +18 -0
- paddlex/modules/ts_classification/trainer.py +108 -0
- paddlex/modules/ts_forecast/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/__init__.py +111 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/__init__.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/analyse_dataset.py +19 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/check_dataset.py +64 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/convert_dataset.py +73 -0
- paddlex/modules/ts_forecast/dataset_checker/dataset_src/split_dataset.py +63 -0
- paddlex/modules/ts_forecast/evaluator.py +66 -0
- paddlex/modules/ts_forecast/exportor.py +44 -0
- paddlex/modules/ts_forecast/model_list.py +24 -0
- paddlex/modules/ts_forecast/trainer.py +108 -0
- paddlex/modules/video_classification/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/__init__.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/__init__.py +18 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/analyse_dataset.py +93 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/check_dataset.py +120 -0
- paddlex/modules/video_classification/dataset_checker/dataset_src/split_dataset.py +82 -0
- paddlex/modules/video_classification/evaluator.py +44 -0
- paddlex/modules/video_classification/exportor.py +22 -0
- paddlex/modules/video_classification/model_list.py +19 -0
- paddlex/modules/video_classification/trainer.py +88 -0
- paddlex/modules/video_detection/__init__.py +18 -0
- paddlex/modules/video_detection/dataset_checker/__init__.py +86 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/__init__.py +17 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/analyse_dataset.py +100 -0
- paddlex/modules/video_detection/dataset_checker/dataset_src/check_dataset.py +132 -0
- paddlex/modules/video_detection/evaluator.py +42 -0
- paddlex/modules/video_detection/exportor.py +22 -0
- paddlex/modules/video_detection/model_list.py +15 -0
- paddlex/modules/video_detection/trainer.py +82 -0
- paddlex/ops/__init__.py +152 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.cpp +266 -0
- paddlex/ops/iou3d_nms/iou3d_cpu.h +28 -0
- paddlex/ops/iou3d_nms/iou3d_nms.cpp +206 -0
- paddlex/ops/iou3d_nms/iou3d_nms.h +35 -0
- paddlex/ops/iou3d_nms/iou3d_nms_api.cpp +114 -0
- paddlex/ops/iou3d_nms/iou3d_nms_kernel.cu +484 -0
- paddlex/ops/setup.py +37 -0
- paddlex/ops/voxel/voxelize_op.cc +194 -0
- paddlex/ops/voxel/voxelize_op.cu +346 -0
- paddlex/paddlex_cli.py +476 -0
- paddlex/repo_apis/Paddle3D_api/__init__.py +17 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/__init__.py +18 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/config.py +118 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py +238 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/register.py +55 -0
- paddlex/repo_apis/Paddle3D_api/bev_fusion/runner.py +104 -0
- paddlex/repo_apis/Paddle3D_api/pp3d_config.py +145 -0
- paddlex/repo_apis/PaddleClas_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleClas_api/cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleClas_api/cls/config.py +595 -0
- paddlex/repo_apis/PaddleClas_api/cls/model.py +355 -0
- paddlex/repo_apis/PaddleClas_api/cls/register.py +907 -0
- paddlex/repo_apis/PaddleClas_api/cls/runner.py +218 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/__init__.py +18 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/config.py +141 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/model.py +20 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/register.py +68 -0
- paddlex/repo_apis/PaddleClas_api/shitu_rec/runner.py +50 -0
- paddlex/repo_apis/PaddleDetection_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleDetection_api/config_helper.py +280 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/__init__.py +18 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/config.py +457 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py +403 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/register.py +262 -0
- paddlex/repo_apis/PaddleDetection_api/instance_seg/runner.py +225 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/config.py +540 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/model.py +429 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py +245 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/register.py +1135 -0
- paddlex/repo_apis/PaddleDetection_api/object_det/runner.py +225 -0
- paddlex/repo_apis/PaddleNLP_api/__init__.py +13 -0
- paddlex/repo_apis/PaddleOCR_api/__init__.py +22 -0
- paddlex/repo_apis/PaddleOCR_api/config_utils.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/config.py +571 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/register.py +99 -0
- paddlex/repo_apis/PaddleOCR_api/formula_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/config.py +64 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/model.py +126 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/register.py +70 -0
- paddlex/repo_apis/PaddleOCR_api/table_rec/runner.py +51 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/config.py +62 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/model.py +72 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/register.py +107 -0
- paddlex/repo_apis/PaddleOCR_api/text_det/runner.py +53 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/__init__.py +16 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/config.py +564 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/model.py +398 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/register.py +216 -0
- paddlex/repo_apis/PaddleOCR_api/text_rec/runner.py +239 -0
- paddlex/repo_apis/PaddleSeg_api/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/base_seg_config.py +134 -0
- paddlex/repo_apis/PaddleSeg_api/seg/__init__.py +16 -0
- paddlex/repo_apis/PaddleSeg_api/seg/config.py +183 -0
- paddlex/repo_apis/PaddleSeg_api/seg/model.py +491 -0
- paddlex/repo_apis/PaddleSeg_api/seg/register.py +272 -0
- paddlex/repo_apis/PaddleSeg_api/seg/runner.py +261 -0
- paddlex/repo_apis/PaddleTS_api/__init__.py +20 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/config.py +88 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/register.py +146 -0
- paddlex/repo_apis/PaddleTS_api/ts_ad/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/__init__.py +13 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/config.py +244 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/model.py +276 -0
- paddlex/repo_apis/PaddleTS_api/ts_base/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/config.py +72 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/register.py +59 -0
- paddlex/repo_apis/PaddleTS_api/ts_cls/runner.py +158 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/__init__.py +16 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/config.py +136 -0
- paddlex/repo_apis/PaddleTS_api/ts_fc/register.py +186 -0
- paddlex/repo_apis/PaddleVideo_api/__init__.py +17 -0
- paddlex/repo_apis/PaddleVideo_api/config_utils.py +51 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/config.py +548 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/model.py +346 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/register.py +70 -0
- paddlex/repo_apis/PaddleVideo_api/video_cls/runner.py +204 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/__init__.py +19 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/config.py +549 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/model.py +298 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/register.py +44 -0
- paddlex/repo_apis/PaddleVideo_api/video_det/runner.py +199 -0
- paddlex/repo_apis/__init__.py +13 -0
- paddlex/repo_apis/base/__init__.py +22 -0
- paddlex/repo_apis/base/config.py +237 -0
- paddlex/repo_apis/base/model.py +563 -0
- paddlex/repo_apis/base/register.py +135 -0
- paddlex/repo_apis/base/runner.py +390 -0
- paddlex/repo_apis/base/utils/__init__.py +13 -0
- paddlex/repo_apis/base/utils/arg.py +64 -0
- paddlex/repo_apis/base/utils/subprocess.py +107 -0
- paddlex/repo_manager/__init__.py +17 -0
- paddlex/repo_manager/core.py +253 -0
- paddlex/repo_manager/meta.py +180 -0
- paddlex/repo_manager/repo.py +425 -0
- paddlex/repo_manager/utils.py +148 -0
- paddlex/utils/__init__.py +1 -12
- paddlex/utils/cache.py +146 -0
- paddlex/utils/config.py +216 -0
- paddlex/utils/custom_device_list.py +311 -0
- paddlex/utils/deps.py +249 -0
- paddlex/utils/device.py +195 -0
- paddlex/utils/download.py +168 -182
- paddlex/utils/env.py +31 -48
- paddlex/utils/errors/__init__.py +17 -0
- paddlex/utils/errors/dataset_checker.py +78 -0
- paddlex/utils/errors/others.py +138 -0
- paddlex/utils/file_interface.py +211 -0
- paddlex/utils/flags.py +70 -0
- paddlex/utils/fonts/__init__.py +97 -0
- paddlex/utils/func_register.py +41 -0
- paddlex/utils/install.py +87 -0
- paddlex/utils/interactive_get_pipeline.py +55 -0
- paddlex/utils/lazy_loader.py +68 -0
- paddlex/utils/logging.py +140 -33
- paddlex/utils/misc.py +201 -0
- paddlex/utils/pipeline_arguments.py +719 -0
- paddlex/utils/result_saver.py +58 -0
- paddlex/utils/subclass_register.py +99 -0
- paddlex/version.py +55 -0
- paddlex-3.0.0.dist-info/METADATA +1168 -0
- paddlex-3.0.0.dist-info/RECORD +1093 -0
- paddlex-3.0.0.dist-info/WHEEL +5 -0
- paddlex-3.0.0.dist-info/entry_points.txt +2 -0
- paddlex-3.0.0.dist-info/licenses/LICENSE +169 -0
- paddlex-3.0.0.dist-info/top_level.txt +1 -0
- PaddleClas/__init__.py +0 -16
- PaddleClas/paddleclas.py +0 -375
- PaddleClas/ppcls/__init__.py +0 -20
- PaddleClas/ppcls/data/__init__.py +0 -15
- PaddleClas/ppcls/data/imaug/__init__.py +0 -94
- PaddleClas/ppcls/data/imaug/autoaugment.py +0 -264
- PaddleClas/ppcls/data/imaug/batch_operators.py +0 -117
- PaddleClas/ppcls/data/imaug/cutout.py +0 -41
- PaddleClas/ppcls/data/imaug/fmix.py +0 -217
- PaddleClas/ppcls/data/imaug/grid.py +0 -89
- PaddleClas/ppcls/data/imaug/hide_and_seek.py +0 -44
- PaddleClas/ppcls/data/imaug/operators.py +0 -244
- PaddleClas/ppcls/data/imaug/randaugment.py +0 -106
- PaddleClas/ppcls/data/imaug/random_erasing.py +0 -55
- PaddleClas/ppcls/data/reader.py +0 -318
- PaddleClas/ppcls/modeling/__init__.py +0 -20
- PaddleClas/ppcls/modeling/architectures/__init__.py +0 -51
- PaddleClas/ppcls/modeling/architectures/alexnet.py +0 -132
- PaddleClas/ppcls/modeling/architectures/darknet.py +0 -161
- PaddleClas/ppcls/modeling/architectures/densenet.py +0 -308
- PaddleClas/ppcls/modeling/architectures/distillation_models.py +0 -65
- PaddleClas/ppcls/modeling/architectures/distilled_vision_transformer.py +0 -196
- PaddleClas/ppcls/modeling/architectures/dpn.py +0 -425
- PaddleClas/ppcls/modeling/architectures/efficientnet.py +0 -901
- PaddleClas/ppcls/modeling/architectures/ghostnet.py +0 -331
- PaddleClas/ppcls/modeling/architectures/googlenet.py +0 -207
- PaddleClas/ppcls/modeling/architectures/hrnet.py +0 -742
- PaddleClas/ppcls/modeling/architectures/inception_v3.py +0 -481
- PaddleClas/ppcls/modeling/architectures/inception_v4.py +0 -455
- PaddleClas/ppcls/modeling/architectures/mixnet.py +0 -782
- PaddleClas/ppcls/modeling/architectures/mobilenet_v1.py +0 -266
- PaddleClas/ppcls/modeling/architectures/mobilenet_v2.py +0 -248
- PaddleClas/ppcls/modeling/architectures/mobilenet_v3.py +0 -359
- PaddleClas/ppcls/modeling/architectures/regnet.py +0 -383
- PaddleClas/ppcls/modeling/architectures/repvgg.py +0 -339
- PaddleClas/ppcls/modeling/architectures/res2net.py +0 -272
- PaddleClas/ppcls/modeling/architectures/res2net_vd.py +0 -295
- PaddleClas/ppcls/modeling/architectures/resnest.py +0 -705
- PaddleClas/ppcls/modeling/architectures/resnet.py +0 -316
- PaddleClas/ppcls/modeling/architectures/resnet_vc.py +0 -309
- PaddleClas/ppcls/modeling/architectures/resnet_vd.py +0 -354
- PaddleClas/ppcls/modeling/architectures/resnext.py +0 -253
- PaddleClas/ppcls/modeling/architectures/resnext101_wsl.py +0 -447
- PaddleClas/ppcls/modeling/architectures/resnext_vd.py +0 -266
- PaddleClas/ppcls/modeling/architectures/rexnet.py +0 -240
- PaddleClas/ppcls/modeling/architectures/se_resnet_vd.py +0 -378
- PaddleClas/ppcls/modeling/architectures/se_resnext.py +0 -290
- PaddleClas/ppcls/modeling/architectures/se_resnext_vd.py +0 -285
- PaddleClas/ppcls/modeling/architectures/shufflenet_v2.py +0 -320
- PaddleClas/ppcls/modeling/architectures/squeezenet.py +0 -154
- PaddleClas/ppcls/modeling/architectures/vgg.py +0 -152
- PaddleClas/ppcls/modeling/architectures/vision_transformer.py +0 -402
- PaddleClas/ppcls/modeling/architectures/xception.py +0 -345
- PaddleClas/ppcls/modeling/architectures/xception_deeplab.py +0 -386
- PaddleClas/ppcls/modeling/loss.py +0 -154
- PaddleClas/ppcls/modeling/utils.py +0 -53
- PaddleClas/ppcls/optimizer/__init__.py +0 -19
- PaddleClas/ppcls/optimizer/learning_rate.py +0 -159
- PaddleClas/ppcls/optimizer/optimizer.py +0 -165
- PaddleClas/ppcls/utils/__init__.py +0 -27
- PaddleClas/ppcls/utils/check.py +0 -151
- PaddleClas/ppcls/utils/config.py +0 -201
- PaddleClas/ppcls/utils/logger.py +0 -120
- PaddleClas/ppcls/utils/metrics.py +0 -107
- PaddleClas/ppcls/utils/misc.py +0 -62
- PaddleClas/ppcls/utils/model_zoo.py +0 -213
- PaddleClas/ppcls/utils/save_load.py +0 -163
- PaddleClas/setup.py +0 -55
- PaddleClas/tools/__init__.py +0 -15
- PaddleClas/tools/download.py +0 -50
- PaddleClas/tools/ema.py +0 -58
- PaddleClas/tools/eval.py +0 -112
- PaddleClas/tools/export_model.py +0 -85
- PaddleClas/tools/export_serving_model.py +0 -76
- PaddleClas/tools/infer/__init__.py +0 -16
- PaddleClas/tools/infer/infer.py +0 -94
- PaddleClas/tools/infer/predict.py +0 -117
- PaddleClas/tools/infer/utils.py +0 -233
- PaddleClas/tools/program.py +0 -444
- PaddleClas/tools/test_hubserving.py +0 -113
- PaddleClas/tools/train.py +0 -141
- paddlex/cls.py +0 -76
- paddlex/command.py +0 -215
- paddlex/cv/__init__.py +0 -17
- paddlex/cv/datasets/__init__.py +0 -18
- paddlex/cv/datasets/coco.py +0 -169
- paddlex/cv/datasets/imagenet.py +0 -88
- paddlex/cv/datasets/seg_dataset.py +0 -91
- paddlex/cv/datasets/voc.py +0 -301
- paddlex/cv/models/__init__.py +0 -18
- paddlex/cv/models/base.py +0 -623
- paddlex/cv/models/classifier.py +0 -814
- paddlex/cv/models/detector.py +0 -1747
- paddlex/cv/models/load_model.py +0 -126
- paddlex/cv/models/segmenter.py +0 -673
- paddlex/cv/models/slim/__init__.py +0 -13
- paddlex/cv/models/slim/prune.py +0 -55
- paddlex/cv/models/utils/__init__.py +0 -13
- paddlex/cv/models/utils/det_metrics/__init__.py +0 -15
- paddlex/cv/models/utils/det_metrics/coco_utils.py +0 -217
- paddlex/cv/models/utils/det_metrics/metrics.py +0 -220
- paddlex/cv/models/utils/ema.py +0 -48
- paddlex/cv/models/utils/seg_metrics.py +0 -62
- paddlex/cv/models/utils/visualize.py +0 -394
- paddlex/cv/transforms/__init__.py +0 -46
- paddlex/cv/transforms/batch_operators.py +0 -286
- paddlex/cv/transforms/box_utils.py +0 -41
- paddlex/cv/transforms/functions.py +0 -193
- paddlex/cv/transforms/operators.py +0 -1402
- paddlex/det.py +0 -43
- paddlex/paddleseg/__init__.py +0 -17
- paddlex/paddleseg/core/__init__.py +0 -20
- paddlex/paddleseg/core/infer.py +0 -289
- paddlex/paddleseg/core/predict.py +0 -145
- paddlex/paddleseg/core/train.py +0 -258
- paddlex/paddleseg/core/val.py +0 -172
- paddlex/paddleseg/cvlibs/__init__.py +0 -17
- paddlex/paddleseg/cvlibs/callbacks.py +0 -279
- paddlex/paddleseg/cvlibs/config.py +0 -359
- paddlex/paddleseg/cvlibs/manager.py +0 -142
- paddlex/paddleseg/cvlibs/param_init.py +0 -91
- paddlex/paddleseg/datasets/__init__.py +0 -21
- paddlex/paddleseg/datasets/ade.py +0 -112
- paddlex/paddleseg/datasets/cityscapes.py +0 -86
- paddlex/paddleseg/datasets/cocostuff.py +0 -79
- paddlex/paddleseg/datasets/dataset.py +0 -164
- paddlex/paddleseg/datasets/mini_deep_globe_road_extraction.py +0 -95
- paddlex/paddleseg/datasets/optic_disc_seg.py +0 -97
- paddlex/paddleseg/datasets/pascal_context.py +0 -80
- paddlex/paddleseg/datasets/voc.py +0 -113
- paddlex/paddleseg/models/__init__.py +0 -39
- paddlex/paddleseg/models/ann.py +0 -436
- paddlex/paddleseg/models/attention_unet.py +0 -189
- paddlex/paddleseg/models/backbones/__init__.py +0 -18
- paddlex/paddleseg/models/backbones/hrnet.py +0 -815
- paddlex/paddleseg/models/backbones/mobilenetv3.py +0 -365
- paddlex/paddleseg/models/backbones/resnet_vd.py +0 -364
- paddlex/paddleseg/models/backbones/xception_deeplab.py +0 -415
- paddlex/paddleseg/models/bisenet.py +0 -311
- paddlex/paddleseg/models/danet.py +0 -220
- paddlex/paddleseg/models/decoupled_segnet.py +0 -233
- paddlex/paddleseg/models/deeplab.py +0 -258
- paddlex/paddleseg/models/dnlnet.py +0 -231
- paddlex/paddleseg/models/emanet.py +0 -219
- paddlex/paddleseg/models/fast_scnn.py +0 -318
- paddlex/paddleseg/models/fcn.py +0 -135
- paddlex/paddleseg/models/gcnet.py +0 -223
- paddlex/paddleseg/models/gscnn.py +0 -357
- paddlex/paddleseg/models/hardnet.py +0 -309
- paddlex/paddleseg/models/isanet.py +0 -202
- paddlex/paddleseg/models/layers/__init__.py +0 -19
- paddlex/paddleseg/models/layers/activation.py +0 -73
- paddlex/paddleseg/models/layers/attention.py +0 -146
- paddlex/paddleseg/models/layers/layer_libs.py +0 -168
- paddlex/paddleseg/models/layers/nonlocal2d.py +0 -155
- paddlex/paddleseg/models/layers/pyramid_pool.py +0 -182
- paddlex/paddleseg/models/losses/__init__.py +0 -27
- paddlex/paddleseg/models/losses/binary_cross_entropy_loss.py +0 -174
- paddlex/paddleseg/models/losses/bootstrapped_cross_entropy.py +0 -73
- paddlex/paddleseg/models/losses/cross_entropy_loss.py +0 -94
- paddlex/paddleseg/models/losses/decoupledsegnet_relax_boundary_loss.py +0 -129
- paddlex/paddleseg/models/losses/dice_loss.py +0 -61
- paddlex/paddleseg/models/losses/edge_attention_loss.py +0 -78
- paddlex/paddleseg/models/losses/gscnn_dual_task_loss.py +0 -141
- paddlex/paddleseg/models/losses/l1_loss.py +0 -76
- paddlex/paddleseg/models/losses/lovasz_loss.py +0 -222
- paddlex/paddleseg/models/losses/mean_square_error_loss.py +0 -65
- paddlex/paddleseg/models/losses/mixed_loss.py +0 -58
- paddlex/paddleseg/models/losses/ohem_cross_entropy_loss.py +0 -99
- paddlex/paddleseg/models/losses/ohem_edge_attention_loss.py +0 -114
- paddlex/paddleseg/models/ocrnet.py +0 -248
- paddlex/paddleseg/models/pspnet.py +0 -147
- paddlex/paddleseg/models/sfnet.py +0 -236
- paddlex/paddleseg/models/shufflenet_slim.py +0 -268
- paddlex/paddleseg/models/u2net.py +0 -574
- paddlex/paddleseg/models/unet.py +0 -155
- paddlex/paddleseg/models/unet_3plus.py +0 -316
- paddlex/paddleseg/models/unet_plusplus.py +0 -237
- paddlex/paddleseg/transforms/__init__.py +0 -16
- paddlex/paddleseg/transforms/functional.py +0 -161
- paddlex/paddleseg/transforms/transforms.py +0 -937
- paddlex/paddleseg/utils/__init__.py +0 -22
- paddlex/paddleseg/utils/config_check.py +0 -60
- paddlex/paddleseg/utils/download.py +0 -163
- paddlex/paddleseg/utils/env/__init__.py +0 -16
- paddlex/paddleseg/utils/env/seg_env.py +0 -56
- paddlex/paddleseg/utils/env/sys_env.py +0 -122
- paddlex/paddleseg/utils/logger.py +0 -48
- paddlex/paddleseg/utils/metrics.py +0 -146
- paddlex/paddleseg/utils/progbar.py +0 -212
- paddlex/paddleseg/utils/timer.py +0 -53
- paddlex/paddleseg/utils/utils.py +0 -120
- paddlex/paddleseg/utils/visualize.py +0 -90
- paddlex/ppcls/__init__.py +0 -20
- paddlex/ppcls/data/__init__.py +0 -15
- paddlex/ppcls/data/imaug/__init__.py +0 -94
- paddlex/ppcls/data/imaug/autoaugment.py +0 -264
- paddlex/ppcls/data/imaug/batch_operators.py +0 -117
- paddlex/ppcls/data/imaug/cutout.py +0 -41
- paddlex/ppcls/data/imaug/fmix.py +0 -217
- paddlex/ppcls/data/imaug/grid.py +0 -89
- paddlex/ppcls/data/imaug/hide_and_seek.py +0 -44
- paddlex/ppcls/data/imaug/operators.py +0 -256
- paddlex/ppcls/data/imaug/randaugment.py +0 -106
- paddlex/ppcls/data/imaug/random_erasing.py +0 -55
- paddlex/ppcls/data/reader.py +0 -318
- paddlex/ppcls/modeling/__init__.py +0 -20
- paddlex/ppcls/modeling/architectures/__init__.py +0 -51
- paddlex/ppcls/modeling/architectures/alexnet.py +0 -132
- paddlex/ppcls/modeling/architectures/darknet.py +0 -161
- paddlex/ppcls/modeling/architectures/densenet.py +0 -308
- paddlex/ppcls/modeling/architectures/distillation_models.py +0 -65
- paddlex/ppcls/modeling/architectures/distilled_vision_transformer.py +0 -196
- paddlex/ppcls/modeling/architectures/dpn.py +0 -425
- paddlex/ppcls/modeling/architectures/efficientnet.py +0 -901
- paddlex/ppcls/modeling/architectures/ghostnet.py +0 -331
- paddlex/ppcls/modeling/architectures/googlenet.py +0 -207
- paddlex/ppcls/modeling/architectures/hrnet.py +0 -742
- paddlex/ppcls/modeling/architectures/inception_v3.py +0 -541
- paddlex/ppcls/modeling/architectures/inception_v4.py +0 -455
- paddlex/ppcls/modeling/architectures/mixnet.py +0 -782
- paddlex/ppcls/modeling/architectures/mobilenet_v1.py +0 -266
- paddlex/ppcls/modeling/architectures/mobilenet_v2.py +0 -248
- paddlex/ppcls/modeling/architectures/mobilenet_v3.py +0 -359
- paddlex/ppcls/modeling/architectures/regnet.py +0 -383
- paddlex/ppcls/modeling/architectures/repvgg.py +0 -339
- paddlex/ppcls/modeling/architectures/res2net.py +0 -272
- paddlex/ppcls/modeling/architectures/res2net_vd.py +0 -295
- paddlex/ppcls/modeling/architectures/resnest.py +0 -705
- paddlex/ppcls/modeling/architectures/resnet.py +0 -317
- paddlex/ppcls/modeling/architectures/resnet_vc.py +0 -309
- paddlex/ppcls/modeling/architectures/resnet_vd.py +0 -354
- paddlex/ppcls/modeling/architectures/resnext.py +0 -259
- paddlex/ppcls/modeling/architectures/resnext101_wsl.py +0 -447
- paddlex/ppcls/modeling/architectures/resnext_vd.py +0 -266
- paddlex/ppcls/modeling/architectures/rexnet.py +0 -240
- paddlex/ppcls/modeling/architectures/se_resnet_vd.py +0 -378
- paddlex/ppcls/modeling/architectures/se_resnext.py +0 -290
- paddlex/ppcls/modeling/architectures/se_resnext_vd.py +0 -285
- paddlex/ppcls/modeling/architectures/shufflenet_v2.py +0 -320
- paddlex/ppcls/modeling/architectures/squeezenet.py +0 -154
- paddlex/ppcls/modeling/architectures/vgg.py +0 -152
- paddlex/ppcls/modeling/architectures/vision_transformer.py +0 -402
- paddlex/ppcls/modeling/architectures/xception.py +0 -345
- paddlex/ppcls/modeling/architectures/xception_deeplab.py +0 -386
- paddlex/ppcls/modeling/loss.py +0 -158
- paddlex/ppcls/modeling/utils.py +0 -53
- paddlex/ppcls/optimizer/__init__.py +0 -19
- paddlex/ppcls/optimizer/learning_rate.py +0 -159
- paddlex/ppcls/optimizer/optimizer.py +0 -165
- paddlex/ppcls/utils/__init__.py +0 -27
- paddlex/ppcls/utils/check.py +0 -151
- paddlex/ppcls/utils/config.py +0 -201
- paddlex/ppcls/utils/logger.py +0 -120
- paddlex/ppcls/utils/metrics.py +0 -112
- paddlex/ppcls/utils/misc.py +0 -62
- paddlex/ppcls/utils/model_zoo.py +0 -213
- paddlex/ppcls/utils/save_load.py +0 -163
- paddlex/ppdet/__init__.py +0 -16
- paddlex/ppdet/core/__init__.py +0 -15
- paddlex/ppdet/core/config/__init__.py +0 -13
- paddlex/ppdet/core/config/schema.py +0 -248
- paddlex/ppdet/core/config/yaml_helpers.py +0 -118
- paddlex/ppdet/core/workspace.py +0 -279
- paddlex/ppdet/data/__init__.py +0 -21
- paddlex/ppdet/data/reader.py +0 -304
- paddlex/ppdet/data/shm_utils.py +0 -67
- paddlex/ppdet/data/source/__init__.py +0 -27
- paddlex/ppdet/data/source/category.py +0 -823
- paddlex/ppdet/data/source/coco.py +0 -243
- paddlex/ppdet/data/source/dataset.py +0 -192
- paddlex/ppdet/data/source/keypoint_coco.py +0 -656
- paddlex/ppdet/data/source/mot.py +0 -360
- paddlex/ppdet/data/source/voc.py +0 -204
- paddlex/ppdet/data/source/widerface.py +0 -180
- paddlex/ppdet/data/transform/__init__.py +0 -28
- paddlex/ppdet/data/transform/autoaugment_utils.py +0 -1593
- paddlex/ppdet/data/transform/batch_operators.py +0 -758
- paddlex/ppdet/data/transform/gridmask_utils.py +0 -83
- paddlex/ppdet/data/transform/keypoint_operators.py +0 -665
- paddlex/ppdet/data/transform/mot_operators.py +0 -636
- paddlex/ppdet/data/transform/op_helper.py +0 -468
- paddlex/ppdet/data/transform/operators.py +0 -2103
- paddlex/ppdet/engine/__init__.py +0 -29
- paddlex/ppdet/engine/callbacks.py +0 -262
- paddlex/ppdet/engine/env.py +0 -47
- paddlex/ppdet/engine/export_utils.py +0 -118
- paddlex/ppdet/engine/tracker.py +0 -425
- paddlex/ppdet/engine/trainer.py +0 -535
- paddlex/ppdet/metrics/__init__.py +0 -23
- paddlex/ppdet/metrics/coco_utils.py +0 -184
- paddlex/ppdet/metrics/json_results.py +0 -151
- paddlex/ppdet/metrics/keypoint_metrics.py +0 -202
- paddlex/ppdet/metrics/map_utils.py +0 -396
- paddlex/ppdet/metrics/metrics.py +0 -300
- paddlex/ppdet/metrics/mot_eval_utils.py +0 -192
- paddlex/ppdet/metrics/mot_metrics.py +0 -184
- paddlex/ppdet/metrics/widerface_utils.py +0 -393
- paddlex/ppdet/model_zoo/__init__.py +0 -18
- paddlex/ppdet/model_zoo/model_zoo.py +0 -86
- paddlex/ppdet/model_zoo/tests/__init__.py +0 -13
- paddlex/ppdet/model_zoo/tests/test_get_model.py +0 -48
- paddlex/ppdet/model_zoo/tests/test_list_model.py +0 -68
- paddlex/ppdet/modeling/__init__.py +0 -41
- paddlex/ppdet/modeling/architectures/__init__.py +0 -40
- paddlex/ppdet/modeling/architectures/cascade_rcnn.py +0 -144
- paddlex/ppdet/modeling/architectures/centernet.py +0 -103
- paddlex/ppdet/modeling/architectures/deepsort.py +0 -111
- paddlex/ppdet/modeling/architectures/fairmot.py +0 -107
- paddlex/ppdet/modeling/architectures/faster_rcnn.py +0 -106
- paddlex/ppdet/modeling/architectures/fcos.py +0 -105
- paddlex/ppdet/modeling/architectures/jde.py +0 -125
- paddlex/ppdet/modeling/architectures/keypoint_hrhrnet.py +0 -286
- paddlex/ppdet/modeling/architectures/keypoint_hrnet.py +0 -203
- paddlex/ppdet/modeling/architectures/mask_rcnn.py +0 -135
- paddlex/ppdet/modeling/architectures/meta_arch.py +0 -45
- paddlex/ppdet/modeling/architectures/s2anet.py +0 -103
- paddlex/ppdet/modeling/architectures/solov2.py +0 -110
- paddlex/ppdet/modeling/architectures/ssd.py +0 -84
- paddlex/ppdet/modeling/architectures/ttfnet.py +0 -98
- paddlex/ppdet/modeling/architectures/yolo.py +0 -104
- paddlex/ppdet/modeling/backbones/__init__.py +0 -37
- paddlex/ppdet/modeling/backbones/blazenet.py +0 -322
- paddlex/ppdet/modeling/backbones/darknet.py +0 -341
- paddlex/ppdet/modeling/backbones/dla.py +0 -244
- paddlex/ppdet/modeling/backbones/ghostnet.py +0 -476
- paddlex/ppdet/modeling/backbones/hrnet.py +0 -724
- paddlex/ppdet/modeling/backbones/mobilenet_v1.py +0 -410
- paddlex/ppdet/modeling/backbones/mobilenet_v3.py +0 -497
- paddlex/ppdet/modeling/backbones/name_adapter.py +0 -69
- paddlex/ppdet/modeling/backbones/res2net.py +0 -358
- paddlex/ppdet/modeling/backbones/resnet.py +0 -606
- paddlex/ppdet/modeling/backbones/senet.py +0 -140
- paddlex/ppdet/modeling/backbones/vgg.py +0 -216
- paddlex/ppdet/modeling/bbox_utils.py +0 -464
- paddlex/ppdet/modeling/heads/__init__.py +0 -41
- paddlex/ppdet/modeling/heads/bbox_head.py +0 -379
- paddlex/ppdet/modeling/heads/cascade_head.py +0 -285
- paddlex/ppdet/modeling/heads/centernet_head.py +0 -194
- paddlex/ppdet/modeling/heads/face_head.py +0 -113
- paddlex/ppdet/modeling/heads/fcos_head.py +0 -270
- paddlex/ppdet/modeling/heads/keypoint_hrhrnet_head.py +0 -108
- paddlex/ppdet/modeling/heads/mask_head.py +0 -253
- paddlex/ppdet/modeling/heads/roi_extractor.py +0 -111
- paddlex/ppdet/modeling/heads/s2anet_head.py +0 -845
- paddlex/ppdet/modeling/heads/solov2_head.py +0 -537
- paddlex/ppdet/modeling/heads/ssd_head.py +0 -175
- paddlex/ppdet/modeling/heads/ttf_head.py +0 -314
- paddlex/ppdet/modeling/heads/yolo_head.py +0 -124
- paddlex/ppdet/modeling/keypoint_utils.py +0 -302
- paddlex/ppdet/modeling/layers.py +0 -1142
- paddlex/ppdet/modeling/losses/__init__.py +0 -35
- paddlex/ppdet/modeling/losses/ctfocal_loss.py +0 -67
- paddlex/ppdet/modeling/losses/fairmot_loss.py +0 -41
- paddlex/ppdet/modeling/losses/fcos_loss.py +0 -225
- paddlex/ppdet/modeling/losses/iou_aware_loss.py +0 -48
- paddlex/ppdet/modeling/losses/iou_loss.py +0 -210
- paddlex/ppdet/modeling/losses/jde_loss.py +0 -182
- paddlex/ppdet/modeling/losses/keypoint_loss.py +0 -228
- paddlex/ppdet/modeling/losses/solov2_loss.py +0 -101
- paddlex/ppdet/modeling/losses/ssd_loss.py +0 -163
- paddlex/ppdet/modeling/losses/yolo_loss.py +0 -212
- paddlex/ppdet/modeling/mot/__init__.py +0 -25
- paddlex/ppdet/modeling/mot/matching/__init__.py +0 -19
- paddlex/ppdet/modeling/mot/matching/deepsort_matching.py +0 -382
- paddlex/ppdet/modeling/mot/matching/jde_matching.py +0 -145
- paddlex/ppdet/modeling/mot/motion/__init__.py +0 -17
- paddlex/ppdet/modeling/mot/motion/kalman_filter.py +0 -270
- paddlex/ppdet/modeling/mot/tracker/__init__.py +0 -23
- paddlex/ppdet/modeling/mot/tracker/base_jde_tracker.py +0 -267
- paddlex/ppdet/modeling/mot/tracker/base_sde_tracker.py +0 -145
- paddlex/ppdet/modeling/mot/tracker/deepsort_tracker.py +0 -165
- paddlex/ppdet/modeling/mot/tracker/jde_tracker.py +0 -262
- paddlex/ppdet/modeling/mot/utils.py +0 -181
- paddlex/ppdet/modeling/mot/visualization.py +0 -130
- paddlex/ppdet/modeling/necks/__init__.py +0 -25
- paddlex/ppdet/modeling/necks/centernet_fpn.py +0 -185
- paddlex/ppdet/modeling/necks/fpn.py +0 -233
- paddlex/ppdet/modeling/necks/hrfpn.py +0 -131
- paddlex/ppdet/modeling/necks/ttf_fpn.py +0 -243
- paddlex/ppdet/modeling/necks/yolo_fpn.py +0 -1034
- paddlex/ppdet/modeling/ops.py +0 -1599
- paddlex/ppdet/modeling/post_process.py +0 -449
- paddlex/ppdet/modeling/proposal_generator/__init__.py +0 -2
- paddlex/ppdet/modeling/proposal_generator/anchor_generator.py +0 -135
- paddlex/ppdet/modeling/proposal_generator/proposal_generator.py +0 -81
- paddlex/ppdet/modeling/proposal_generator/rpn_head.py +0 -269
- paddlex/ppdet/modeling/proposal_generator/target.py +0 -671
- paddlex/ppdet/modeling/proposal_generator/target_layer.py +0 -476
- paddlex/ppdet/modeling/reid/__init__.py +0 -23
- paddlex/ppdet/modeling/reid/fairmot_embedding_head.py +0 -117
- paddlex/ppdet/modeling/reid/jde_embedding_head.py +0 -189
- paddlex/ppdet/modeling/reid/pyramidal_embedding.py +0 -151
- paddlex/ppdet/modeling/reid/resnet.py +0 -320
- paddlex/ppdet/modeling/shape_spec.py +0 -33
- paddlex/ppdet/modeling/tests/__init__.py +0 -13
- paddlex/ppdet/modeling/tests/test_architectures.py +0 -59
- paddlex/ppdet/modeling/tests/test_base.py +0 -75
- paddlex/ppdet/modeling/tests/test_ops.py +0 -839
- paddlex/ppdet/modeling/tests/test_yolov3_loss.py +0 -420
- paddlex/ppdet/optimizer.py +0 -285
- paddlex/ppdet/slim/__init__.py +0 -62
- paddlex/ppdet/slim/distill.py +0 -111
- paddlex/ppdet/slim/prune.py +0 -85
- paddlex/ppdet/slim/quant.py +0 -52
- paddlex/ppdet/utils/__init__.py +0 -13
- paddlex/ppdet/utils/check.py +0 -93
- paddlex/ppdet/utils/checkpoint.py +0 -216
- paddlex/ppdet/utils/cli.py +0 -151
- paddlex/ppdet/utils/colormap.py +0 -56
- paddlex/ppdet/utils/download.py +0 -477
- paddlex/ppdet/utils/logger.py +0 -71
- paddlex/ppdet/utils/stats.py +0 -95
- paddlex/ppdet/utils/visualizer.py +0 -292
- paddlex/ppdet/utils/voc_utils.py +0 -87
- paddlex/seg.py +0 -38
- paddlex/tools/__init__.py +0 -16
- paddlex/tools/convert.py +0 -52
- paddlex/tools/dataset_conversion/__init__.py +0 -24
- paddlex/tools/dataset_conversion/x2coco.py +0 -379
- paddlex/tools/dataset_conversion/x2imagenet.py +0 -82
- paddlex/tools/dataset_conversion/x2seg.py +0 -343
- paddlex/tools/dataset_conversion/x2voc.py +0 -230
- paddlex/tools/dataset_split/__init__.py +0 -23
- paddlex/tools/dataset_split/coco_split.py +0 -69
- paddlex/tools/dataset_split/imagenet_split.py +0 -75
- paddlex/tools/dataset_split/seg_split.py +0 -96
- paddlex/tools/dataset_split/utils.py +0 -75
- paddlex/tools/dataset_split/voc_split.py +0 -91
- paddlex/tools/split.py +0 -41
- paddlex/utils/checkpoint.py +0 -439
- paddlex/utils/shm.py +0 -67
- paddlex/utils/stats.py +0 -68
- paddlex/utils/utils.py +0 -140
- paddlex-2.0.0rc4.dist-info/LICENSE +0 -201
- paddlex-2.0.0rc4.dist-info/METADATA +0 -29
- paddlex-2.0.0rc4.dist-info/RECORD +0 -445
- paddlex-2.0.0rc4.dist-info/WHEEL +0 -5
- paddlex-2.0.0rc4.dist-info/entry_points.txt +0 -3
- paddlex-2.0.0rc4.dist-info/top_level.txt +0 -2
paddlex/ppdet/utils/cli.py
DELETED
@@ -1,151 +0,0 @@
|
|
1
|
-
# Copyright (c) 2019 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
|
-
from argparse import ArgumentParser, RawDescriptionHelpFormatter
|
16
|
-
|
17
|
-
import yaml
|
18
|
-
import re
|
19
|
-
from paddlex.ppdet.core.workspace import get_registered_modules, dump_value
|
20
|
-
|
21
|
-
__all__ = ['ColorTTY', 'ArgsParser']
|
22
|
-
|
23
|
-
|
24
|
-
class ColorTTY(object):
|
25
|
-
def __init__(self):
|
26
|
-
super(ColorTTY, self).__init__()
|
27
|
-
self.colors = ['red', 'green', 'yellow', 'blue', 'magenta', 'cyan']
|
28
|
-
|
29
|
-
def __getattr__(self, attr):
|
30
|
-
if attr in self.colors:
|
31
|
-
color = self.colors.index(attr) + 31
|
32
|
-
|
33
|
-
def color_message(message):
|
34
|
-
return "[{}m{}[0m".format(color, message)
|
35
|
-
|
36
|
-
setattr(self, attr, color_message)
|
37
|
-
return color_message
|
38
|
-
|
39
|
-
def bold(self, message):
|
40
|
-
return self.with_code('01', message)
|
41
|
-
|
42
|
-
def with_code(self, code, message):
|
43
|
-
return "[{}m{}[0m".format(code, message)
|
44
|
-
|
45
|
-
|
46
|
-
class ArgsParser(ArgumentParser):
|
47
|
-
def __init__(self):
|
48
|
-
super(ArgsParser, self).__init__(
|
49
|
-
formatter_class=RawDescriptionHelpFormatter)
|
50
|
-
self.add_argument("-c", "--config", help="configuration file to use")
|
51
|
-
self.add_argument(
|
52
|
-
"-o", "--opt", nargs='*', help="set configuration options")
|
53
|
-
|
54
|
-
def parse_args(self, argv=None):
|
55
|
-
args = super(ArgsParser, self).parse_args(argv)
|
56
|
-
assert args.config is not None, \
|
57
|
-
"Please specify --config=configure_file_path."
|
58
|
-
args.opt = self._parse_opt(args.opt)
|
59
|
-
return args
|
60
|
-
|
61
|
-
def _parse_opt(self, opts):
|
62
|
-
config = {}
|
63
|
-
if not opts:
|
64
|
-
return config
|
65
|
-
for s in opts:
|
66
|
-
s = s.strip()
|
67
|
-
k, v = s.split('=', 1)
|
68
|
-
if '.' not in k:
|
69
|
-
config[k] = yaml.load(v, Loader=yaml.Loader)
|
70
|
-
else:
|
71
|
-
keys = k.split('.')
|
72
|
-
if keys[0] not in config:
|
73
|
-
config[keys[0]] = {}
|
74
|
-
cur = config[keys[0]]
|
75
|
-
for idx, key in enumerate(keys[1:]):
|
76
|
-
if idx == len(keys) - 2:
|
77
|
-
cur[key] = yaml.load(v, Loader=yaml.Loader)
|
78
|
-
else:
|
79
|
-
cur[key] = {}
|
80
|
-
cur = cur[key]
|
81
|
-
return config
|
82
|
-
|
83
|
-
|
84
|
-
def print_total_cfg(config):
|
85
|
-
modules = get_registered_modules()
|
86
|
-
color_tty = ColorTTY()
|
87
|
-
green = '___{}___'.format(color_tty.colors.index('green') + 31)
|
88
|
-
|
89
|
-
styled = {}
|
90
|
-
for key in config.keys():
|
91
|
-
if not config[key]: # empty schema
|
92
|
-
continue
|
93
|
-
|
94
|
-
if key not in modules and not hasattr(config[key], '__dict__'):
|
95
|
-
styled[key] = config[key]
|
96
|
-
continue
|
97
|
-
elif key in modules:
|
98
|
-
module = modules[key]
|
99
|
-
else:
|
100
|
-
type_name = type(config[key]).__name__
|
101
|
-
if type_name in modules:
|
102
|
-
module = modules[type_name].copy()
|
103
|
-
module.update({
|
104
|
-
k: v
|
105
|
-
for k, v in config[key].__dict__.items()
|
106
|
-
if k in module.schema
|
107
|
-
})
|
108
|
-
key += " ({})".format(type_name)
|
109
|
-
default = module.find_default_keys()
|
110
|
-
missing = module.find_missing_keys()
|
111
|
-
mismatch = module.find_mismatch_keys()
|
112
|
-
extra = module.find_extra_keys()
|
113
|
-
dep_missing = []
|
114
|
-
for dep in module.inject:
|
115
|
-
if isinstance(module[dep], str) and module[dep] != '<value>':
|
116
|
-
if module[dep] not in modules: # not a valid module
|
117
|
-
dep_missing.append(dep)
|
118
|
-
else:
|
119
|
-
dep_mod = modules[module[dep]]
|
120
|
-
# empty dict but mandatory
|
121
|
-
if not dep_mod and dep_mod.mandatory():
|
122
|
-
dep_missing.append(dep)
|
123
|
-
override = list(
|
124
|
-
set(module.keys()) - set(default) - set(extra) - set(dep_missing))
|
125
|
-
replacement = {}
|
126
|
-
for name in set(override + default + extra + mismatch + missing):
|
127
|
-
new_name = name
|
128
|
-
if name in missing:
|
129
|
-
value = "<missing>"
|
130
|
-
else:
|
131
|
-
value = module[name]
|
132
|
-
|
133
|
-
if name in extra:
|
134
|
-
value = dump_value(value) + " <extraneous>"
|
135
|
-
elif name in mismatch:
|
136
|
-
value = dump_value(value) + " <type mismatch>"
|
137
|
-
elif name in dep_missing:
|
138
|
-
value = dump_value(value) + " <module config missing>"
|
139
|
-
elif name in override and value != '<missing>':
|
140
|
-
mark = green
|
141
|
-
new_name = mark + name
|
142
|
-
replacement[new_name] = value
|
143
|
-
styled[key] = replacement
|
144
|
-
buffer = yaml.dump(styled, default_flow_style=False, default_style='')
|
145
|
-
buffer = (re.sub(r"<missing>", r"[31m<missing>[0m", buffer))
|
146
|
-
buffer = (re.sub(r"<extraneous>", r"[33m<extraneous>[0m", buffer))
|
147
|
-
buffer = (re.sub(r"<type mismatch>", r"[31m<type mismatch>[0m", buffer))
|
148
|
-
buffer = (re.sub(r"<module config missing>",
|
149
|
-
r"[31m<module config missing>[0m", buffer))
|
150
|
-
buffer = re.sub(r"___(\d+)___(.*?):", r"[\1m\2[0m:", buffer)
|
151
|
-
print(buffer)
|
paddlex/ppdet/utils/colormap.py
DELETED
@@ -1,56 +0,0 @@
|
|
1
|
-
# Copyright (c) 2019 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
|
-
from __future__ import absolute_import
|
16
|
-
from __future__ import division
|
17
|
-
from __future__ import print_function
|
18
|
-
from __future__ import unicode_literals
|
19
|
-
|
20
|
-
import numpy as np
|
21
|
-
|
22
|
-
|
23
|
-
def colormap(rgb=False):
|
24
|
-
"""
|
25
|
-
Get colormap
|
26
|
-
"""
|
27
|
-
color_list = np.array([
|
28
|
-
0.000, 0.447, 0.741, 0.850, 0.325, 0.098, 0.929, 0.694, 0.125, 0.494,
|
29
|
-
0.184, 0.556, 0.466, 0.674, 0.188, 0.301, 0.745, 0.933, 0.635, 0.078,
|
30
|
-
0.184, 0.300, 0.300, 0.300, 0.600, 0.600, 0.600, 1.000, 0.000, 0.000,
|
31
|
-
1.000, 0.500, 0.000, 0.749, 0.749, 0.000, 0.000, 1.000, 0.000, 0.000,
|
32
|
-
0.000, 1.000, 0.667, 0.000, 1.000, 0.333, 0.333, 0.000, 0.333, 0.667,
|
33
|
-
0.000, 0.333, 1.000, 0.000, 0.667, 0.333, 0.000, 0.667, 0.667, 0.000,
|
34
|
-
0.667, 1.000, 0.000, 1.000, 0.333, 0.000, 1.000, 0.667, 0.000, 1.000,
|
35
|
-
1.000, 0.000, 0.000, 0.333, 0.500, 0.000, 0.667, 0.500, 0.000, 1.000,
|
36
|
-
0.500, 0.333, 0.000, 0.500, 0.333, 0.333, 0.500, 0.333, 0.667, 0.500,
|
37
|
-
0.333, 1.000, 0.500, 0.667, 0.000, 0.500, 0.667, 0.333, 0.500, 0.667,
|
38
|
-
0.667, 0.500, 0.667, 1.000, 0.500, 1.000, 0.000, 0.500, 1.000, 0.333,
|
39
|
-
0.500, 1.000, 0.667, 0.500, 1.000, 1.000, 0.500, 0.000, 0.333, 1.000,
|
40
|
-
0.000, 0.667, 1.000, 0.000, 1.000, 1.000, 0.333, 0.000, 1.000, 0.333,
|
41
|
-
0.333, 1.000, 0.333, 0.667, 1.000, 0.333, 1.000, 1.000, 0.667, 0.000,
|
42
|
-
1.000, 0.667, 0.333, 1.000, 0.667, 0.667, 1.000, 0.667, 1.000, 1.000,
|
43
|
-
1.000, 0.000, 1.000, 1.000, 0.333, 1.000, 1.000, 0.667, 1.000, 0.167,
|
44
|
-
0.000, 0.000, 0.333, 0.000, 0.000, 0.500, 0.000, 0.000, 0.667, 0.000,
|
45
|
-
0.000, 0.833, 0.000, 0.000, 1.000, 0.000, 0.000, 0.000, 0.167, 0.000,
|
46
|
-
0.000, 0.333, 0.000, 0.000, 0.500, 0.000, 0.000, 0.667, 0.000, 0.000,
|
47
|
-
0.833, 0.000, 0.000, 1.000, 0.000, 0.000, 0.000, 0.167, 0.000, 0.000,
|
48
|
-
0.333, 0.000, 0.000, 0.500, 0.000, 0.000, 0.667, 0.000, 0.000, 0.833,
|
49
|
-
0.000, 0.000, 1.000, 0.000, 0.000, 0.000, 0.143, 0.143, 0.143, 0.286,
|
50
|
-
0.286, 0.286, 0.429, 0.429, 0.429, 0.571, 0.571, 0.571, 0.714, 0.714,
|
51
|
-
0.714, 0.857, 0.857, 0.857, 1.000, 1.000, 1.000
|
52
|
-
]).astype(np.float32)
|
53
|
-
color_list = color_list.reshape((-1, 3)) * 255
|
54
|
-
if not rgb:
|
55
|
-
color_list = color_list[:, ::-1]
|
56
|
-
return color_list
|
paddlex/ppdet/utils/download.py
DELETED
@@ -1,477 +0,0 @@
|
|
1
|
-
# Copyright (c) 2019 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
|
-
from __future__ import absolute_import
|
16
|
-
from __future__ import division
|
17
|
-
from __future__ import print_function
|
18
|
-
|
19
|
-
import os
|
20
|
-
import os.path as osp
|
21
|
-
import sys
|
22
|
-
import yaml
|
23
|
-
import shutil
|
24
|
-
import requests
|
25
|
-
import tqdm
|
26
|
-
import hashlib
|
27
|
-
import base64
|
28
|
-
import binascii
|
29
|
-
import tarfile
|
30
|
-
import zipfile
|
31
|
-
|
32
|
-
from .voc_utils import create_list
|
33
|
-
from paddlex.ppdet.core.workspace import BASE_KEY
|
34
|
-
|
35
|
-
from .logger import setup_logger
|
36
|
-
logger = setup_logger(__name__)
|
37
|
-
|
38
|
-
__all__ = [
|
39
|
-
'get_weights_path', 'get_dataset_path', 'get_config_path',
|
40
|
-
'download_dataset', 'create_voc_list'
|
41
|
-
]
|
42
|
-
|
43
|
-
WEIGHTS_HOME = osp.expanduser("~/.cache/paddle/weights")
|
44
|
-
DATASET_HOME = osp.expanduser("~/.cache/paddle/dataset")
|
45
|
-
CONFIGS_HOME = osp.expanduser("~/.cache/paddle/configs")
|
46
|
-
|
47
|
-
# dict of {dataset_name: (download_info, sub_dirs)}
|
48
|
-
# download info: [(url, md5sum)]
|
49
|
-
DATASETS = {
|
50
|
-
'coco': ([
|
51
|
-
(
|
52
|
-
'http://images.cocodataset.org/zips/train2017.zip',
|
53
|
-
'cced6f7f71b7629ddf16f17bbcfab6b2', ),
|
54
|
-
(
|
55
|
-
'http://images.cocodataset.org/zips/val2017.zip',
|
56
|
-
'442b8da7639aecaf257c1dceb8ba8c80', ),
|
57
|
-
(
|
58
|
-
'http://images.cocodataset.org/annotations/annotations_trainval2017.zip',
|
59
|
-
'f4bbac642086de4f52a3fdda2de5fa2c', ),
|
60
|
-
], ["annotations", "train2017", "val2017"]),
|
61
|
-
'voc': ([
|
62
|
-
(
|
63
|
-
'http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar',
|
64
|
-
'6cd6e144f989b92b3379bac3b3de84fd', ),
|
65
|
-
(
|
66
|
-
'http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar',
|
67
|
-
'c52e279531787c972589f7e41ab4ae64', ),
|
68
|
-
(
|
69
|
-
'http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar',
|
70
|
-
'b6e924de25625d8de591ea690078ad9f', ),
|
71
|
-
], ["VOCdevkit/VOC2012", "VOCdevkit/VOC2007"]),
|
72
|
-
'wider_face': ([
|
73
|
-
(
|
74
|
-
'https://dataset.bj.bcebos.com/wider_face/WIDER_train.zip',
|
75
|
-
'3fedf70df600953d25982bcd13d91ba2', ),
|
76
|
-
(
|
77
|
-
'https://dataset.bj.bcebos.com/wider_face/WIDER_val.zip',
|
78
|
-
'dfa7d7e790efa35df3788964cf0bbaea', ),
|
79
|
-
(
|
80
|
-
'https://dataset.bj.bcebos.com/wider_face/wider_face_split.zip',
|
81
|
-
'a4a898d6193db4b9ef3260a68bad0dc7', ),
|
82
|
-
], ["WIDER_train", "WIDER_val", "wider_face_split"]),
|
83
|
-
'fruit': ([(
|
84
|
-
'https://dataset.bj.bcebos.com/PaddleDetection_demo/fruit.tar',
|
85
|
-
'baa8806617a54ccf3685fa7153388ae6', ), ],
|
86
|
-
['Annotations', 'JPEGImages']),
|
87
|
-
'roadsign_voc': ([(
|
88
|
-
'https://paddlemodels.bj.bcebos.com/object_detection/roadsign_voc.tar',
|
89
|
-
'8d629c0f880dd8b48de9aeff44bf1f3e', ), ], ['annotations', 'images']),
|
90
|
-
'roadsign_coco': ([(
|
91
|
-
'https://paddlemodels.bj.bcebos.com/object_detection/roadsign_coco.tar',
|
92
|
-
'49ce5a9b5ad0d6266163cd01de4b018e', ), ], ['annotations', 'images']),
|
93
|
-
'mot': (),
|
94
|
-
'objects365': ()
|
95
|
-
}
|
96
|
-
|
97
|
-
DOWNLOAD_RETRY_LIMIT = 3
|
98
|
-
|
99
|
-
PPDET_WEIGHTS_DOWNLOAD_URL_PREFIX = 'https://paddledet.bj.bcebos.com/'
|
100
|
-
|
101
|
-
|
102
|
-
def parse_url(url):
|
103
|
-
url = url.replace("ppdet://", PPDET_WEIGHTS_DOWNLOAD_URL_PREFIX)
|
104
|
-
return url
|
105
|
-
|
106
|
-
|
107
|
-
def get_weights_path(url):
|
108
|
-
"""Get weights path from WEIGHTS_HOME, if not exists,
|
109
|
-
download it from url.
|
110
|
-
"""
|
111
|
-
url = parse_url(url)
|
112
|
-
path, _ = get_path(url, WEIGHTS_HOME)
|
113
|
-
return path
|
114
|
-
|
115
|
-
|
116
|
-
def get_config_path(url):
|
117
|
-
"""Get weights path from CONFIGS_HOME, if not exists,
|
118
|
-
download it from url.
|
119
|
-
"""
|
120
|
-
url = parse_url(url)
|
121
|
-
path, _ = get_path(url, CONFIGS_HOME)
|
122
|
-
_download_config(path, url, CONFIGS_HOME)
|
123
|
-
|
124
|
-
return path
|
125
|
-
|
126
|
-
|
127
|
-
def _download_config(cfg_path, cfg_url, cur_dir):
|
128
|
-
with open(cfg_path) as f:
|
129
|
-
cfg = yaml.load(f, Loader=yaml.Loader)
|
130
|
-
|
131
|
-
# download dependence base ymls
|
132
|
-
if BASE_KEY in cfg:
|
133
|
-
base_ymls = list(cfg[BASE_KEY])
|
134
|
-
for base_yml in base_ymls:
|
135
|
-
if base_yml.startswith("~"):
|
136
|
-
base_yml = os.path.expanduser(base_yml)
|
137
|
-
relpath = osp.relpath(base_yml, cfg_path)
|
138
|
-
if not base_yml.startswith('/'):
|
139
|
-
relpath = base_yml
|
140
|
-
base_yml = os.path.join(os.path.dirname(cfg_path), base_yml)
|
141
|
-
|
142
|
-
if osp.isfile(base_yml):
|
143
|
-
logger.debug("Found _BASE_ config: {}".format(base_yml))
|
144
|
-
continue
|
145
|
-
|
146
|
-
# download to CONFIGS_HOME firstly
|
147
|
-
base_yml_url = osp.join(osp.split(cfg_url)[0], relpath)
|
148
|
-
path, _ = get_path(base_yml_url, CONFIGS_HOME)
|
149
|
-
|
150
|
-
# move from CONFIGS_HOME to dst_path to restore config directory structure
|
151
|
-
dst_path = osp.join(cur_dir, relpath)
|
152
|
-
dst_dir = osp.split(dst_path)[0]
|
153
|
-
if not osp.isdir(dst_dir):
|
154
|
-
os.makedirs(dst_dir)
|
155
|
-
shutil.move(path, dst_path)
|
156
|
-
|
157
|
-
# perfrom download base yml recursively
|
158
|
-
_download_config(dst_path, base_yml_url, osp.split(dst_path)[0])
|
159
|
-
|
160
|
-
|
161
|
-
def get_dataset_path(path, annotation, image_dir):
|
162
|
-
"""
|
163
|
-
If path exists, return path.
|
164
|
-
Otherwise, get dataset path from DATASET_HOME, if not exists,
|
165
|
-
download it.
|
166
|
-
"""
|
167
|
-
if _dataset_exists(path, annotation, image_dir):
|
168
|
-
return path
|
169
|
-
|
170
|
-
logger.info(
|
171
|
-
"Dataset {} is not valid for reason above, try searching {} or "
|
172
|
-
"downloading dataset...".format(osp.realpath(path), DATASET_HOME))
|
173
|
-
|
174
|
-
data_name = os.path.split(path.strip().lower())[-1]
|
175
|
-
for name, dataset in DATASETS.items():
|
176
|
-
if data_name == name:
|
177
|
-
logger.debug("Parse dataset_dir {} as dataset "
|
178
|
-
"{}".format(path, name))
|
179
|
-
if name == 'objects365':
|
180
|
-
raise NotImplementedError(
|
181
|
-
"Dataset {} is not valid for download automatically. "
|
182
|
-
"Please apply and download the dataset from "
|
183
|
-
"https://www.objects365.org/download.html".format(name))
|
184
|
-
data_dir = osp.join(DATASET_HOME, name)
|
185
|
-
|
186
|
-
if name == 'mot':
|
187
|
-
if osp.exists(path) or osp.exists(data_dir):
|
188
|
-
return data_dir
|
189
|
-
else:
|
190
|
-
raise NotImplementedError(
|
191
|
-
"Dataset {} is not valid for download automatically. "
|
192
|
-
"Please apply and download the dataset following docs/tutorials/PrepareMOTDataSet.md".
|
193
|
-
format(name))
|
194
|
-
|
195
|
-
# For voc, only check dir VOCdevkit/VOC2012, VOCdevkit/VOC2007
|
196
|
-
if name in ['voc', 'fruit', 'roadsign_voc']:
|
197
|
-
exists = True
|
198
|
-
for sub_dir in dataset[1]:
|
199
|
-
check_dir = osp.join(data_dir, sub_dir)
|
200
|
-
if osp.exists(check_dir):
|
201
|
-
logger.info("Found {}".format(check_dir))
|
202
|
-
else:
|
203
|
-
exists = False
|
204
|
-
if exists:
|
205
|
-
return data_dir
|
206
|
-
|
207
|
-
# voc exist is checked above, voc is not exist here
|
208
|
-
check_exist = name != 'voc' and name != 'fruit' and name != 'roadsign_voc'
|
209
|
-
for url, md5sum in dataset[0]:
|
210
|
-
get_path(url, data_dir, md5sum, check_exist)
|
211
|
-
|
212
|
-
# voc should create list after download
|
213
|
-
if name == 'voc':
|
214
|
-
create_voc_list(data_dir)
|
215
|
-
return data_dir
|
216
|
-
|
217
|
-
# not match any dataset in DATASETS
|
218
|
-
raise ValueError(
|
219
|
-
"Dataset {} is not valid and cannot parse dataset type "
|
220
|
-
"'{}' for automaticly downloading, which only supports "
|
221
|
-
"'voc' , 'coco', 'wider_face', 'fruit', 'roadsign_voc' and 'mot' currently".
|
222
|
-
format(path, osp.split(path)[-1]))
|
223
|
-
|
224
|
-
|
225
|
-
def create_voc_list(data_dir, devkit_subdir='VOCdevkit'):
|
226
|
-
logger.debug("Create voc file list...")
|
227
|
-
devkit_dir = osp.join(data_dir, devkit_subdir)
|
228
|
-
years = ['2007', '2012']
|
229
|
-
|
230
|
-
# NOTE: since using auto download VOC
|
231
|
-
# dataset, VOC default label list should be used,
|
232
|
-
# do not generate label_list.txt here. For default
|
233
|
-
# label, see ../data/source/voc.py
|
234
|
-
create_list(devkit_dir, years, data_dir)
|
235
|
-
logger.debug("Create voc file list finished")
|
236
|
-
|
237
|
-
|
238
|
-
def map_path(url, root_dir):
|
239
|
-
# parse path after download to decompress under root_dir
|
240
|
-
fname = osp.split(url)[-1]
|
241
|
-
zip_formats = ['.zip', '.tar', '.gz']
|
242
|
-
fpath = fname
|
243
|
-
for zip_format in zip_formats:
|
244
|
-
fpath = fpath.replace(zip_format, '')
|
245
|
-
return osp.join(root_dir, fpath)
|
246
|
-
|
247
|
-
|
248
|
-
def get_path(url, root_dir, md5sum=None, check_exist=True):
|
249
|
-
""" Download from given url to root_dir.
|
250
|
-
if file or directory specified by url is exists under
|
251
|
-
root_dir, return the path directly, otherwise download
|
252
|
-
from url and decompress it, return the path.
|
253
|
-
|
254
|
-
url (str): download url
|
255
|
-
root_dir (str): root dir for downloading, it should be
|
256
|
-
WEIGHTS_HOME or DATASET_HOME
|
257
|
-
md5sum (str): md5 sum of download package
|
258
|
-
"""
|
259
|
-
# parse path after download to decompress under root_dir
|
260
|
-
fullpath = map_path(url, root_dir)
|
261
|
-
|
262
|
-
# For same zip file, decompressed directory name different
|
263
|
-
# from zip file name, rename by following map
|
264
|
-
decompress_name_map = {
|
265
|
-
"VOCtrainval_11-May-2012": "VOCdevkit/VOC2012",
|
266
|
-
"VOCtrainval_06-Nov-2007": "VOCdevkit/VOC2007",
|
267
|
-
"VOCtest_06-Nov-2007": "VOCdevkit/VOC2007",
|
268
|
-
"annotations_trainval": "annotations"
|
269
|
-
}
|
270
|
-
for k, v in decompress_name_map.items():
|
271
|
-
if fullpath.find(k) >= 0:
|
272
|
-
fullpath = osp.join(osp.split(fullpath)[0], v)
|
273
|
-
|
274
|
-
if osp.exists(fullpath) and check_exist:
|
275
|
-
if not osp.isfile(fullpath) or \
|
276
|
-
_check_exist_file_md5(fullpath, md5sum, url):
|
277
|
-
logger.debug("Found {}".format(fullpath))
|
278
|
-
return fullpath, True
|
279
|
-
else:
|
280
|
-
os.remove(fullpath)
|
281
|
-
|
282
|
-
fullname = _download(url, root_dir, md5sum)
|
283
|
-
|
284
|
-
# new weights format which postfix is 'pdparams' not
|
285
|
-
# need to decompress
|
286
|
-
if osp.splitext(fullname)[-1] not in ['.pdparams', '.yml']:
|
287
|
-
_decompress(fullname)
|
288
|
-
|
289
|
-
return fullpath, False
|
290
|
-
|
291
|
-
|
292
|
-
def download_dataset(path, dataset=None):
|
293
|
-
if dataset not in DATASETS.keys():
|
294
|
-
logger.error("Unknown dataset {}, it should be "
|
295
|
-
"{}".format(dataset, DATASETS.keys()))
|
296
|
-
return
|
297
|
-
dataset_info = DATASETS[dataset][0]
|
298
|
-
for info in dataset_info:
|
299
|
-
get_path(info[0], path, info[1], False)
|
300
|
-
logger.debug("Download dataset {} finished.".format(dataset))
|
301
|
-
|
302
|
-
|
303
|
-
def _dataset_exists(path, annotation, image_dir):
|
304
|
-
"""
|
305
|
-
Check if user define dataset exists
|
306
|
-
"""
|
307
|
-
if not osp.exists(path):
|
308
|
-
logger.warning("Config dataset_dir {} is not exits, "
|
309
|
-
"dataset config is not valid".format(path))
|
310
|
-
return False
|
311
|
-
|
312
|
-
if annotation:
|
313
|
-
annotation_path = osp.join(path, annotation)
|
314
|
-
if not osp.isfile(annotation_path):
|
315
|
-
logger.warning("Config annotation {} is not a "
|
316
|
-
"file, dataset config is not "
|
317
|
-
"valid".format(annotation_path))
|
318
|
-
return False
|
319
|
-
if image_dir:
|
320
|
-
image_path = osp.join(path, image_dir)
|
321
|
-
if not osp.isdir(image_path):
|
322
|
-
logger.warning("Config image_dir {} is not a "
|
323
|
-
"directory, dataset config is not "
|
324
|
-
"valid".format(image_path))
|
325
|
-
return False
|
326
|
-
return True
|
327
|
-
|
328
|
-
|
329
|
-
def _download(url, path, md5sum=None):
|
330
|
-
"""
|
331
|
-
Download from url, save to path.
|
332
|
-
|
333
|
-
url (str): download url
|
334
|
-
path (str): download to given path
|
335
|
-
"""
|
336
|
-
if not osp.exists(path):
|
337
|
-
os.makedirs(path)
|
338
|
-
|
339
|
-
fname = osp.split(url)[-1]
|
340
|
-
fullname = osp.join(path, fname)
|
341
|
-
retry_cnt = 0
|
342
|
-
|
343
|
-
while not (osp.exists(fullname) and _check_exist_file_md5(fullname, md5sum,
|
344
|
-
url)):
|
345
|
-
if retry_cnt < DOWNLOAD_RETRY_LIMIT:
|
346
|
-
retry_cnt += 1
|
347
|
-
else:
|
348
|
-
raise RuntimeError("Download from {} failed. "
|
349
|
-
"Retry limit reached".format(url))
|
350
|
-
|
351
|
-
logger.info("Downloading {} from {}".format(fname, url))
|
352
|
-
|
353
|
-
# NOTE: windows path join may incur \, which is invalid in url
|
354
|
-
if sys.platform == "win32":
|
355
|
-
url = url.replace('\\', '/')
|
356
|
-
|
357
|
-
req = requests.get(url, stream=True)
|
358
|
-
if req.status_code != 200:
|
359
|
-
raise RuntimeError("Downloading from {} failed with code "
|
360
|
-
"{}!".format(url, req.status_code))
|
361
|
-
|
362
|
-
# For protecting download interupted, download to
|
363
|
-
# tmp_fullname firstly, move tmp_fullname to fullname
|
364
|
-
# after download finished
|
365
|
-
tmp_fullname = fullname + "_tmp"
|
366
|
-
total_size = req.headers.get('content-length')
|
367
|
-
with open(tmp_fullname, 'wb') as f:
|
368
|
-
if total_size:
|
369
|
-
for chunk in tqdm.tqdm(
|
370
|
-
req.iter_content(chunk_size=1024),
|
371
|
-
total=(int(total_size) + 1023) // 1024,
|
372
|
-
unit='KB'):
|
373
|
-
f.write(chunk)
|
374
|
-
else:
|
375
|
-
for chunk in req.iter_content(chunk_size=1024):
|
376
|
-
if chunk:
|
377
|
-
f.write(chunk)
|
378
|
-
shutil.move(tmp_fullname, fullname)
|
379
|
-
return fullname
|
380
|
-
|
381
|
-
|
382
|
-
def _check_exist_file_md5(filename, md5sum, url):
|
383
|
-
# if md5sum is None, and file to check is weights file,
|
384
|
-
# read md5um from url and check, else check md5sum directly
|
385
|
-
return _md5check_from_url(filename, url) if md5sum is None \
|
386
|
-
and filename.endswith('pdparams') \
|
387
|
-
else _md5check(filename, md5sum)
|
388
|
-
|
389
|
-
|
390
|
-
def _md5check_from_url(filename, url):
|
391
|
-
# For weights in bcebos URLs, MD5 value is contained
|
392
|
-
# in request header as 'content_md5'
|
393
|
-
req = requests.get(url, stream=True)
|
394
|
-
content_md5 = req.headers.get('content-md5')
|
395
|
-
req.close()
|
396
|
-
if not content_md5 or _md5check(
|
397
|
-
filename,
|
398
|
-
binascii.hexlify(base64.b64decode(content_md5.strip('"'))).decode(
|
399
|
-
)):
|
400
|
-
return True
|
401
|
-
else:
|
402
|
-
return False
|
403
|
-
|
404
|
-
|
405
|
-
def _md5check(fullname, md5sum=None):
|
406
|
-
if md5sum is None:
|
407
|
-
return True
|
408
|
-
|
409
|
-
logger.debug("File {} md5 checking...".format(fullname))
|
410
|
-
md5 = hashlib.md5()
|
411
|
-
with open(fullname, 'rb') as f:
|
412
|
-
for chunk in iter(lambda: f.read(4096), b""):
|
413
|
-
md5.update(chunk)
|
414
|
-
calc_md5sum = md5.hexdigest()
|
415
|
-
|
416
|
-
if calc_md5sum != md5sum:
|
417
|
-
logger.warning("File {} md5 check failed, {}(calc) != "
|
418
|
-
"{}(base)".format(fullname, calc_md5sum, md5sum))
|
419
|
-
return False
|
420
|
-
return True
|
421
|
-
|
422
|
-
|
423
|
-
def _decompress(fname):
|
424
|
-
"""
|
425
|
-
Decompress for zip and tar file
|
426
|
-
"""
|
427
|
-
logger.info("Decompressing {}...".format(fname))
|
428
|
-
|
429
|
-
# For protecting decompressing interupted,
|
430
|
-
# decompress to fpath_tmp directory firstly, if decompress
|
431
|
-
# successed, move decompress files to fpath and delete
|
432
|
-
# fpath_tmp and remove download compress file.
|
433
|
-
fpath = osp.split(fname)[0]
|
434
|
-
fpath_tmp = osp.join(fpath, 'tmp')
|
435
|
-
if osp.isdir(fpath_tmp):
|
436
|
-
shutil.rmtree(fpath_tmp)
|
437
|
-
os.makedirs(fpath_tmp)
|
438
|
-
|
439
|
-
if fname.find('tar') >= 0:
|
440
|
-
with tarfile.open(fname) as tf:
|
441
|
-
tf.extractall(path=fpath_tmp)
|
442
|
-
elif fname.find('zip') >= 0:
|
443
|
-
with zipfile.ZipFile(fname) as zf:
|
444
|
-
zf.extractall(path=fpath_tmp)
|
445
|
-
else:
|
446
|
-
raise TypeError("Unsupport compress file type {}".format(fname))
|
447
|
-
|
448
|
-
for f in os.listdir(fpath_tmp):
|
449
|
-
src_dir = osp.join(fpath_tmp, f)
|
450
|
-
dst_dir = osp.join(fpath, f)
|
451
|
-
_move_and_merge_tree(src_dir, dst_dir)
|
452
|
-
|
453
|
-
shutil.rmtree(fpath_tmp)
|
454
|
-
os.remove(fname)
|
455
|
-
|
456
|
-
|
457
|
-
def _move_and_merge_tree(src, dst):
|
458
|
-
"""
|
459
|
-
Move src directory to dst, if dst is already exists,
|
460
|
-
merge src to dst
|
461
|
-
"""
|
462
|
-
if not osp.exists(dst):
|
463
|
-
shutil.move(src, dst)
|
464
|
-
elif osp.isfile(src):
|
465
|
-
shutil.move(src, dst)
|
466
|
-
else:
|
467
|
-
for fp in os.listdir(src):
|
468
|
-
src_fp = osp.join(src, fp)
|
469
|
-
dst_fp = osp.join(dst, fp)
|
470
|
-
if osp.isdir(src_fp):
|
471
|
-
if osp.isdir(dst_fp):
|
472
|
-
_move_and_merge_tree(src_fp, dst_fp)
|
473
|
-
else:
|
474
|
-
shutil.move(src_fp, dst_fp)
|
475
|
-
elif osp.isfile(src_fp) and \
|
476
|
-
not osp.isfile(dst_fp):
|
477
|
-
shutil.move(src_fp, dst_fp)
|