|
@@ -15,359 +15,307 @@
|
|
|
import os
|
|
import os
|
|
|
import shutil
|
|
import shutil
|
|
|
import tempfile
|
|
import tempfile
|
|
|
-from functools import lru_cache
|
|
|
|
|
|
|
+from abc import ABC, abstractmethod
|
|
|
from pathlib import Path
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
import huggingface_hub as hf_hub
|
|
import huggingface_hub as hf_hub
|
|
|
|
|
|
|
|
hf_hub.logging.set_verbosity_error()
|
|
hf_hub.logging.set_verbosity_error()
|
|
|
|
|
|
|
|
|
|
+import modelscope
|
|
|
import requests
|
|
import requests
|
|
|
|
|
|
|
|
|
|
+os.environ["AISTUDIO_LOG"] = "critical"
|
|
|
|
|
+from aistudio_sdk.snapshot_download import snapshot_download as aistudio_download
|
|
|
|
|
+
|
|
|
from ...utils import logging
|
|
from ...utils import logging
|
|
|
from ...utils.cache import CACHE_DIR
|
|
from ...utils.cache import CACHE_DIR
|
|
|
from ...utils.download import download_and_extract
|
|
from ...utils.download import download_and_extract
|
|
|
from ...utils.flags import MODEL_SOURCE
|
|
from ...utils.flags import MODEL_SOURCE
|
|
|
|
|
|
|
|
-OFFICIAL_MODELS = {
|
|
|
|
|
- "ResNet18": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet18_infer.tar",
|
|
|
|
|
- "ResNet18_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet18_vd_infer.tar",
|
|
|
|
|
- "ResNet34": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet34_infer.tar",
|
|
|
|
|
- "ResNet34_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet34_vd_infer.tar",
|
|
|
|
|
- "ResNet50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet50_infer.tar",
|
|
|
|
|
- "ResNet50_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet50_vd_infer.tar",
|
|
|
|
|
- "ResNet101": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet101_infer.tar",
|
|
|
|
|
- "ResNet101_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet101_vd_infer.tar",
|
|
|
|
|
- "ResNet152": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet152_infer.tar",
|
|
|
|
|
- "ResNet152_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet152_vd_infer.tar",
|
|
|
|
|
- "ResNet200_vd": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet200_vd_infer.tar",
|
|
|
|
|
- "PP-LCNet_x0_25": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x0_25_infer.tar",
|
|
|
|
|
- "PP-LCNet_x0_25_textline_ori": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x0_25_textline_ori_infer.tar",
|
|
|
|
|
- "PP-LCNet_x0_35": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x0_35_infer.tar",
|
|
|
|
|
- "PP-LCNet_x0_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x0_5_infer.tar",
|
|
|
|
|
- "PP-LCNet_x0_75": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x0_75_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_doc_ori": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_doc_ori_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_textline_ori": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_textline_ori_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_5_infer.tar",
|
|
|
|
|
- "PP-LCNet_x2_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x2_5_infer.tar",
|
|
|
|
|
- "PP-LCNet_x2_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x2_0_infer.tar",
|
|
|
|
|
- "PP-LCNetV2_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNetV2_small_infer.tar",
|
|
|
|
|
- "PP-LCNetV2_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNetV2_base_infer.tar",
|
|
|
|
|
- "PP-LCNetV2_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNetV2_large_infer.tar",
|
|
|
|
|
- "MobileNetV3_large_x0_35": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_large_x0_35_infer.tar",
|
|
|
|
|
- "MobileNetV3_large_x0_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_large_x0_5_infer.tar",
|
|
|
|
|
- "MobileNetV3_large_x0_75": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_large_x0_75_infer.tar",
|
|
|
|
|
- "MobileNetV3_large_x1_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_large_x1_0_infer.tar",
|
|
|
|
|
- "MobileNetV3_large_x1_25": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_large_x1_25_infer.tar",
|
|
|
|
|
- "MobileNetV3_small_x0_35": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_small_x0_35_infer.tar",
|
|
|
|
|
- "MobileNetV3_small_x0_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_small_x0_5_infer.tar",
|
|
|
|
|
- "MobileNetV3_small_x0_75": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_small_x0_75_infer.tar",
|
|
|
|
|
- "MobileNetV3_small_x1_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_small_x1_0_infer.tar",
|
|
|
|
|
- "MobileNetV3_small_x1_25": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV3_small_x1_25_infer.tar",
|
|
|
|
|
- "ConvNeXt_tiny": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_tiny_infer.tar",
|
|
|
|
|
- "ConvNeXt_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_small_infer.tar",
|
|
|
|
|
- "ConvNeXt_base_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_base_224_infer.tar",
|
|
|
|
|
- "ConvNeXt_base_384": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_base_384_infer.tar",
|
|
|
|
|
- "ConvNeXt_large_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_large_224_infer.tar",
|
|
|
|
|
- "ConvNeXt_large_384": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ConvNeXt_large_384_infer.tar",
|
|
|
|
|
- "MobileNetV2_x0_25": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV2_x0_25_infer.tar",
|
|
|
|
|
- "MobileNetV2_x0_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV2_x0_5_infer.tar",
|
|
|
|
|
- "MobileNetV2_x1_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV2_x1_0_infer.tar",
|
|
|
|
|
- "MobileNetV2_x1_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV2_x1_5_infer.tar",
|
|
|
|
|
- "MobileNetV2_x2_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV2_x2_0_infer.tar",
|
|
|
|
|
- "MobileNetV1_x0_25": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV1_x0_25_infer.tar",
|
|
|
|
|
- "MobileNetV1_x0_5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV1_x0_5_infer.tar",
|
|
|
|
|
- "MobileNetV1_x0_75": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV1_x0_75_infer.tar",
|
|
|
|
|
- "MobileNetV1_x1_0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-MobileNetV1_x1_0_infer.tar",
|
|
|
|
|
- "SwinTransformer_tiny_patch4_window7_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_tiny_patch4_window7_224_infer.tar",
|
|
|
|
|
- "SwinTransformer_small_patch4_window7_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_small_patch4_window7_224_infer.tar",
|
|
|
|
|
- "SwinTransformer_base_patch4_window7_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_base_patch4_window7_224_infer.tar",
|
|
|
|
|
- "SwinTransformer_base_patch4_window12_384": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_base_patch4_window12_384_infer.tar",
|
|
|
|
|
- "SwinTransformer_large_patch4_window7_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_large_patch4_window7_224_infer.tar",
|
|
|
|
|
- "SwinTransformer_large_patch4_window12_384": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-SwinTransformer_large_patch4_window12_384_infer.tar",
|
|
|
|
|
- "PP-HGNet_tiny": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNet_tiny_infer.tar",
|
|
|
|
|
- "PP-HGNet_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNet_small_infer.tar",
|
|
|
|
|
- "PP-HGNet_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNet_base_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B0_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B1": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B1_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B2": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B2_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B3": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B3_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B4": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B4_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B5_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B6": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B6_infer.tar",
|
|
|
|
|
- "FasterNet-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-L_infer.tar",
|
|
|
|
|
- "FasterNet-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-M_infer.tar",
|
|
|
|
|
- "FasterNet-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-S_infer.tar",
|
|
|
|
|
- "FasterNet-T0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-T0_infer.tar",
|
|
|
|
|
- "FasterNet-T1": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-T1_infer.tar",
|
|
|
|
|
- "FasterNet-T2": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterNet-T2_infer.tar",
|
|
|
|
|
- "StarNet-S1": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/StarNet-S1_infer.tar",
|
|
|
|
|
- "StarNet-S2": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/StarNet-S2_infer.tar",
|
|
|
|
|
- "StarNet-S3": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/StarNet-S3_infer.tar",
|
|
|
|
|
- "StarNet-S4": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/StarNet-S4_infer.tar",
|
|
|
|
|
- "MobileNetV4_conv_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV4_conv_small_infer.tar",
|
|
|
|
|
- "MobileNetV4_conv_medium": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV4_conv_medium_infer.tar",
|
|
|
|
|
- "MobileNetV4_conv_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV4_conv_large_infer.tar",
|
|
|
|
|
- "MobileNetV4_hybrid_medium": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV4_hybrid_medium_infer.tar",
|
|
|
|
|
- "MobileNetV4_hybrid_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileNetV4_hybrid_large_infer.tar",
|
|
|
|
|
- "CLIP_vit_base_patch16_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-CLIP_vit_base_patch16_224_infer.tar",
|
|
|
|
|
- "CLIP_vit_large_patch14_224": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-CLIP_vit_large_patch14_224_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_ML_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B0_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B0_ML_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B4_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B4_ML_infer.tar",
|
|
|
|
|
- "PP-HGNetV2-B6_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-HGNetV2-B6_ML_infer.tar",
|
|
|
|
|
- "ResNet50_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet50_ML_infer.tar",
|
|
|
|
|
- "CLIP_vit_base_patch16_448_ML": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/CLIP_vit_base_patch16_448_ML_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus-X": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus-X_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus-L_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus-M_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus-S_infer.tar",
|
|
|
|
|
- "RT-DETR-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-L_infer.tar",
|
|
|
|
|
- "RT-DETR-H": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-H_infer.tar",
|
|
|
|
|
- "RT-DETR-X": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-X_infer.tar",
|
|
|
|
|
- "YOLOv3-DarkNet53": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOv3-DarkNet53_infer.tar",
|
|
|
|
|
- "YOLOv3-MobileNetV3": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOv3-MobileNetV3_infer.tar",
|
|
|
|
|
- "YOLOv3-ResNet50_vd_DCN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOv3-ResNet50_vd_DCN_infer.tar",
|
|
|
|
|
- "YOLOX-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-L_infer.tar",
|
|
|
|
|
- "YOLOX-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-M_infer.tar",
|
|
|
|
|
- "YOLOX-N": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-N_infer.tar",
|
|
|
|
|
- "YOLOX-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-S_infer.tar",
|
|
|
|
|
- "YOLOX-T": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-T_infer.tar",
|
|
|
|
|
- "YOLOX-X": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLOX-X_infer.tar",
|
|
|
|
|
- "RT-DETR-R18": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-R18_infer.tar",
|
|
|
|
|
- "RT-DETR-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-R50_infer.tar",
|
|
|
|
|
- "PicoDet-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-S_infer.tar",
|
|
|
|
|
- "PicoDet-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-L_infer.tar",
|
|
|
|
|
- "Deeplabv3-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Deeplabv3-R50_infer.tar",
|
|
|
|
|
- "Deeplabv3-R101": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Deeplabv3-R101_infer.tar",
|
|
|
|
|
- "Deeplabv3_Plus-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-Deeplabv3_Plus-R50_infer.tar",
|
|
|
|
|
- "Deeplabv3_Plus-R101": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-Deeplabv3_Plus-R101_infer.tar",
|
|
|
|
|
- "PP-ShiTuV2_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-ShiTuV2_rec_infer.tar",
|
|
|
|
|
- "PP-ShiTuV2_rec_CLIP_vit_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-ShiTuV2_rec_CLIP_vit_base_infer.tar",
|
|
|
|
|
- "PP-ShiTuV2_rec_CLIP_vit_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-ShiTuV2_rec_CLIP_vit_large_infer.tar",
|
|
|
|
|
- "PP-LiteSeg-T": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LiteSeg-T_infer.tar",
|
|
|
|
|
- "PP-LiteSeg-B": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LiteSeg-B_infer.tar",
|
|
|
|
|
- "OCRNet_HRNet-W48": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/OCRNet_HRNet-W48_infer.tar",
|
|
|
|
|
- "OCRNet_HRNet-W18": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/OCRNet_HRNet-W18_infer.tar",
|
|
|
|
|
- "SegFormer-B0": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B0_infer.tar",
|
|
|
|
|
- "SegFormer-B1": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B1_infer.tar",
|
|
|
|
|
- "SegFormer-B2": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B2_infer.tar",
|
|
|
|
|
- "SegFormer-B3": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B3_infer.tar",
|
|
|
|
|
- "SegFormer-B4": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B4_infer.tar",
|
|
|
|
|
- "SegFormer-B5": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SegFormer-B5_infer.tar",
|
|
|
|
|
- "SeaFormer_tiny": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SeaFormer_tiny_infer.tar",
|
|
|
|
|
- "SeaFormer_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SeaFormer_small_infer.tar",
|
|
|
|
|
- "SeaFormer_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SeaFormer_base_infer.tar",
|
|
|
|
|
- "SeaFormer_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SeaFormer_large_infer.tar",
|
|
|
|
|
- "Mask-RT-DETR-H": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Mask-RT-DETR-H_infer.tar",
|
|
|
|
|
- "Mask-RT-DETR-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Mask-RT-DETR-L_infer.tar",
|
|
|
|
|
- "PP-OCRv4_server_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_server_rec_infer.tar",
|
|
|
|
|
- "Mask-RT-DETR-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Mask-RT-DETR-S_infer.tar",
|
|
|
|
|
- "Mask-RT-DETR-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Mask-RT-DETR-M_infer.tar",
|
|
|
|
|
- "Mask-RT-DETR-X": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Mask-RT-DETR-X_infer.tar",
|
|
|
|
|
- "SOLOv2": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SOLOv2_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNet50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNet50_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNet50-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNet50-FPN_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNet50-vd-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNet50-vd-FPN_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNet101-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNet101-FPN_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNet101-vd-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNet101-vd-FPN_infer.tar",
|
|
|
|
|
- "MaskRCNN-ResNeXt101-vd-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskRCNN-ResNeXt101-vd-FPN_infer.tar",
|
|
|
|
|
- "Cascade-MaskRCNN-ResNet50-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Cascade-MaskRCNN-ResNet50-FPN_infer.tar",
|
|
|
|
|
- "Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN_infer.tar",
|
|
|
|
|
- "PP-YOLOE_seg-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_seg-S_infer.tar",
|
|
|
|
|
- "PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "en_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-en_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "korean_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-korean_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "japan_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-japan_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "chinese_cht_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-chinese_cht_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "te_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-te_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "ka_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-ka_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "ta_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-ta_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "latin_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-latin_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "arabic_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-arabic_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "cyrillic_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-cyrillic_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "devanagari_PP-OCRv3_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-devanagari_PP-OCRv3_mobile_rec_infer.tar",
|
|
|
|
|
- "en_PP-OCRv4_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-en_PP-OCRv4_mobile_rec_infer.tar",
|
|
|
|
|
- "PP-OCRv4_server_rec_doc": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_server_rec_doc_infer.tar",
|
|
|
|
|
- "PP-OCRv4_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_mobile_rec_infer.tar",
|
|
|
|
|
- "PP-OCRv4_server_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_server_det_infer.tar",
|
|
|
|
|
- "PP-OCRv4_mobile_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_mobile_det_infer.tar",
|
|
|
|
|
- "PP-OCRv3_server_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv3_server_det_infer.tar",
|
|
|
|
|
- "PP-OCRv3_mobile_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv3_mobile_det_infer.tar",
|
|
|
|
|
- "PP-OCRv4_server_seal_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_server_seal_det_infer.tar",
|
|
|
|
|
- "PP-OCRv4_mobile_seal_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv4_mobile_seal_det_infer.tar",
|
|
|
|
|
- "ch_RepSVTR_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-ch_RepSVTR_rec_infer.tar",
|
|
|
|
|
- "ch_SVTRv2_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-ch_SVTRv2_rec_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_pedestrian_attribute": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-LCNet_x1_0_pedestrian_attribute_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_vehicle_attribute": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-LCNet_x1_0_vehicle_attribute_infer.tar",
|
|
|
|
|
- "PicoDet_layout_1x": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet_layout_1x_infer.tar",
|
|
|
|
|
- "PicoDet_layout_1x_table": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet_layout_1x_table_infer.tar",
|
|
|
|
|
- "SLANet": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SLANet_infer.tar",
|
|
|
|
|
- "SLANet_plus": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SLANet_plus_infer.tar",
|
|
|
|
|
- "LaTeX_OCR_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/LaTeX_OCR_rec_infer.tar",
|
|
|
|
|
- "UniMERNet": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/UniMERNet_infer.tar",
|
|
|
|
|
- "PP-FormulaNet-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet-S_infer.tar",
|
|
|
|
|
- "PP-FormulaNet-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet-L_infer.tar",
|
|
|
|
|
- "PP-FormulaNet_plus-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-S_infer.tar",
|
|
|
|
|
- "PP-FormulaNet_plus-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-M_infer.tar",
|
|
|
|
|
- "PP-FormulaNet_plus-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-L_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet34-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet34-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet50_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet50-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet50-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet50-vd-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet50-vd-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet50-vd-SSLDv2-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet50-vd-SSLDv2-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet101": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet101_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNet101-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNet101-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-ResNeXt101-vd-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-ResNeXt101-vd-FPN_infer.tar",
|
|
|
|
|
- "FasterRCNN-Swin-Tiny-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FasterRCNN-Swin-Tiny-FPN_infer.tar",
|
|
|
|
|
- "Cascade-FasterRCNN-ResNet50-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Cascade-FasterRCNN-ResNet50-FPN_infer.tar",
|
|
|
|
|
- "Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN_infer.tar",
|
|
|
|
|
- "UVDoc": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/UVDoc_infer.tar",
|
|
|
|
|
- "DLinear": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/DLinear_infer.tar",
|
|
|
|
|
- "NLinear": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/NLinear_infer.tar",
|
|
|
|
|
- "RLinear": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RLinear_infer.tar",
|
|
|
|
|
- "Nonstationary": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Nonstationary_infer.tar",
|
|
|
|
|
- "TimesNet": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/TimesNet_infer.tar",
|
|
|
|
|
- "TiDE": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/TiDE_infer.tar",
|
|
|
|
|
- "PatchTST": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PatchTST_infer.tar",
|
|
|
|
|
- "DLinear_ad": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/DLinear_ad_infer.tar",
|
|
|
|
|
- "AutoEncoder_ad": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/AutoEncoder_ad_infer.tar",
|
|
|
|
|
- "Nonstationary_ad": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Nonstationary_ad_infer.tar",
|
|
|
|
|
- "PatchTST_ad": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PatchTST_ad_infer.tar",
|
|
|
|
|
- "TimesNet_ad": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/TimesNet_ad_infer.tar",
|
|
|
|
|
- "TimesNet_cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/TimesNet_cls_infer.tar",
|
|
|
|
|
- "STFPM": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/STFPM_infer.tar",
|
|
|
|
|
- "FCOS-ResNet50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/FCOS-ResNet50_infer.tar",
|
|
|
|
|
- "DETR-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/DETR-R50_infer.tar",
|
|
|
|
|
- "PP-YOLOE-L_vehicle": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE-L_vehicle_infer.tar",
|
|
|
|
|
- "PP-YOLOE-S_vehicle": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE-S_vehicle_infer.tar",
|
|
|
|
|
- "PP-ShiTuV2_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-ShiTuV2_det_infer.tar",
|
|
|
|
|
- "PP-YOLOE-S_human": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE-S_human_infer.tar",
|
|
|
|
|
- "PP-YOLOE-L_human": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE-L_human_infer.tar",
|
|
|
|
|
- "PicoDet-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-M_infer.tar",
|
|
|
|
|
- "PicoDet-XS": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-XS_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus_SOD-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus_SOD-L_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus_SOD-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus_SOD-S_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus_SOD-largesize-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus_SOD-largesize-L_infer.tar",
|
|
|
|
|
- "CenterNet-DLA-34": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/CenterNet-DLA-34_infer.tar",
|
|
|
|
|
- "CenterNet-ResNet50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/CenterNet-ResNet50_infer.tar",
|
|
|
|
|
- "PicoDet-S_layout_3cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-S_layout_3cls_infer.tar",
|
|
|
|
|
- "PicoDet-S_layout_17cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-S_layout_17cls_infer.tar",
|
|
|
|
|
- "PicoDet-L_layout_3cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-L_layout_3cls_infer.tar",
|
|
|
|
|
- "PicoDet-L_layout_17cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-L_layout_17cls_infer.tar",
|
|
|
|
|
- "RT-DETR-H_layout_3cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-H_layout_3cls_infer.tar",
|
|
|
|
|
- "RT-DETR-H_layout_17cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-H_layout_17cls_infer.tar",
|
|
|
|
|
- "PicoDet_LCNet_x2_5_face": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet_LCNet_x2_5_face_infer.tar",
|
|
|
|
|
- "BlazeFace": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/BlazeFace_infer.tar",
|
|
|
|
|
- "BlazeFace-FPN-SSH": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/BlazeFace-FPN-SSH_infer.tar",
|
|
|
|
|
- "PP-YOLOE_plus-S_face": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE_plus-S_face_infer.tar",
|
|
|
|
|
- "MobileFaceNet": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MobileFaceNet_infer.tar",
|
|
|
|
|
- "ResNet50_face": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/ResNet50_face_infer.tar",
|
|
|
|
|
- "PP-YOLOE-R-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-YOLOE-R-L_infer.tar",
|
|
|
|
|
- "Co-Deformable-DETR-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Co-Deformable-DETR-R50_infer.tar",
|
|
|
|
|
- "Co-Deformable-DETR-Swin-T": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Co-Deformable-DETR-Swin-T_infer.tar",
|
|
|
|
|
- "Co-DINO-R50": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Co-DINO-R50_infer.tar",
|
|
|
|
|
- "Co-DINO-Swin-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/Co-DINO-Swin-L_infer.tar",
|
|
|
|
|
- "whisper_large": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/whisper_large.tar",
|
|
|
|
|
- "whisper_base": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/whisper_base.tar",
|
|
|
|
|
- "whisper_medium": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/whisper_medium.tar",
|
|
|
|
|
- "whisper_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/whisper_small.tar",
|
|
|
|
|
- "whisper_tiny": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/whisper_tiny.tar",
|
|
|
|
|
- "PP-TSM-R50_8frames_uniform": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-TSM-R50_8frames_uniform_infer.tar",
|
|
|
|
|
- "PP-TSMv2-LCNetV2_8frames_uniform": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-TSMv2-LCNetV2_8frames_uniform_infer.tar",
|
|
|
|
|
- "PP-TSMv2-LCNetV2_16frames_uniform": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-TSMv2-LCNetV2_16frames_uniform_infer.tar",
|
|
|
|
|
- "MaskFormer_tiny": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskFormer_tiny_infer.tar",
|
|
|
|
|
- "MaskFormer_small": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/MaskFormer_small_infer.tar",
|
|
|
|
|
- "PP-LCNet_x1_0_table_cls": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_table_cls_infer.tar",
|
|
|
|
|
- "SLANeXt_wired": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SLANeXt_wired_infer.tar",
|
|
|
|
|
- "SLANeXt_wireless": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SLANeXt_wireless_infer.tar",
|
|
|
|
|
- "RT-DETR-L_wired_table_cell_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-L_wired_table_cell_det_infer.tar",
|
|
|
|
|
- "RT-DETR-L_wireless_table_cell_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-L_wireless_table_cell_det_infer.tar",
|
|
|
|
|
- "YOWO": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOWO_infer.tar",
|
|
|
|
|
- "PP-TinyPose_128x96": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-TinyPose_128x96_infer.tar",
|
|
|
|
|
- "PP-TinyPose_256x192": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-TinyPose_256x192_infer.tar",
|
|
|
|
|
- "GroundingDINO-T": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/GroundingDINO-T_infer.tar",
|
|
|
|
|
- "SAM-H_box": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SAM-H_box_infer.tar",
|
|
|
|
|
- "SAM-H_point": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/SAM-H_point_infer.tar",
|
|
|
|
|
- "PP-DocLayout-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-L_infer.tar",
|
|
|
|
|
- "PP-DocLayout-M": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-M_infer.tar",
|
|
|
|
|
- "PP-DocLayout-S": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-S_infer.tar",
|
|
|
|
|
- "PP-DocLayout_plus-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout_plus-L_infer.tar",
|
|
|
|
|
- "PP-DocBlockLayout": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocBlockLayout_infer.tar",
|
|
|
|
|
- "BEVFusion": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/BEVFusion_infer.tar",
|
|
|
|
|
- "YOLO-Worldv2-L": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/YOLO-Worldv2-L_infer.tar",
|
|
|
|
|
- "PP-DocBee-2B": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocBee-2B_infer.tar",
|
|
|
|
|
- "PP-DocBee-7B": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocBee-7B_infer.tar",
|
|
|
|
|
- "PP-Chart2Table": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-Chart2Table_infer.tar",
|
|
|
|
|
- "PP-OCRv5_server_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-OCRv5_server_det_infer.tar",
|
|
|
|
|
- "PP-OCRv5_mobile_det": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-OCRv5_mobile_det_infer.tar",
|
|
|
|
|
- "PP-OCRv5_server_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv5_server_rec_infer.tar",
|
|
|
|
|
- "PP-OCRv5_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-PP-OCRv5_mobile_rec_infer.tar",
|
|
|
|
|
- "eslav_PP-OCRv5_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/\
|
|
|
|
|
-eslav_PP-OCRv5_mobile_rec_infer.tar",
|
|
|
|
|
- "PP-DocBee2-3B": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocBee2-3B_infer.tar",
|
|
|
|
|
- "latin_PP-OCRv5_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/latin_PP-OCRv5_mobile_rec_infer.tar",
|
|
|
|
|
- "korean_PP-OCRv5_mobile_rec": "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/korean_PP-OCRv5_mobile_rec_infer.tar",
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-
|
|
|
|
|
-HUGGINGFACE_MODELS = [
|
|
|
|
|
|
|
+ALL_MODELS = [
|
|
|
|
|
+ "ResNet18",
|
|
|
|
|
+ "ResNet18_vd",
|
|
|
|
|
+ "ResNet34",
|
|
|
|
|
+ "ResNet34_vd",
|
|
|
|
|
+ "ResNet50",
|
|
|
|
|
+ "ResNet50_vd",
|
|
|
|
|
+ "ResNet101",
|
|
|
|
|
+ "ResNet101_vd",
|
|
|
|
|
+ "ResNet152",
|
|
|
|
|
+ "ResNet152_vd",
|
|
|
|
|
+ "ResNet200_vd",
|
|
|
|
|
+ "PP-LCNet_x0_25",
|
|
|
|
|
+ "PP-LCNet_x0_25_textline_ori",
|
|
|
|
|
+ "PP-LCNet_x0_35",
|
|
|
|
|
+ "PP-LCNet_x0_5",
|
|
|
|
|
+ "PP-LCNet_x0_75",
|
|
|
|
|
+ "PP-LCNet_x1_0",
|
|
|
|
|
+ "PP-LCNet_x1_0_doc_ori",
|
|
|
|
|
+ "PP-LCNet_x1_0_textline_ori",
|
|
|
|
|
+ "PP-LCNet_x1_5",
|
|
|
|
|
+ "PP-LCNet_x2_5",
|
|
|
|
|
+ "PP-LCNet_x2_0",
|
|
|
|
|
+ "PP-LCNetV2_small",
|
|
|
|
|
+ "PP-LCNetV2_base",
|
|
|
|
|
+ "PP-LCNetV2_large",
|
|
|
|
|
+ "MobileNetV3_large_x0_35",
|
|
|
|
|
+ "MobileNetV3_large_x0_5",
|
|
|
|
|
+ "MobileNetV3_large_x0_75",
|
|
|
|
|
+ "MobileNetV3_large_x1_0",
|
|
|
|
|
+ "MobileNetV3_large_x1_25",
|
|
|
|
|
+ "MobileNetV3_small_x0_35",
|
|
|
|
|
+ "MobileNetV3_small_x0_5",
|
|
|
|
|
+ "MobileNetV3_small_x0_75",
|
|
|
|
|
+ "MobileNetV3_small_x1_0",
|
|
|
|
|
+ "MobileNetV3_small_x1_25",
|
|
|
|
|
+ "ConvNeXt_tiny",
|
|
|
|
|
+ "ConvNeXt_small",
|
|
|
|
|
+ "ConvNeXt_base_224",
|
|
|
|
|
+ "ConvNeXt_base_384",
|
|
|
|
|
+ "ConvNeXt_large_224",
|
|
|
|
|
+ "ConvNeXt_large_384",
|
|
|
|
|
+ "MobileNetV2_x0_25",
|
|
|
|
|
+ "MobileNetV2_x0_5",
|
|
|
|
|
+ "MobileNetV2_x1_0",
|
|
|
|
|
+ "MobileNetV2_x1_5",
|
|
|
|
|
+ "MobileNetV2_x2_0",
|
|
|
|
|
+ "MobileNetV1_x0_25",
|
|
|
|
|
+ "MobileNetV1_x0_5",
|
|
|
|
|
+ "MobileNetV1_x0_75",
|
|
|
|
|
+ "MobileNetV1_x1_0",
|
|
|
|
|
+ "SwinTransformer_tiny_patch4_window7_224",
|
|
|
|
|
+ "SwinTransformer_small_patch4_window7_224",
|
|
|
|
|
+ "SwinTransformer_base_patch4_window7_224",
|
|
|
|
|
+ "SwinTransformer_base_patch4_window12_384",
|
|
|
|
|
+ "SwinTransformer_large_patch4_window7_224",
|
|
|
|
|
+ "SwinTransformer_large_patch4_window12_384",
|
|
|
|
|
+ "PP-HGNet_tiny",
|
|
|
|
|
+ "PP-HGNet_small",
|
|
|
|
|
+ "PP-HGNet_base",
|
|
|
|
|
+ "PP-HGNetV2-B0",
|
|
|
|
|
+ "PP-HGNetV2-B1",
|
|
|
|
|
+ "PP-HGNetV2-B2",
|
|
|
|
|
+ "PP-HGNetV2-B3",
|
|
|
|
|
+ "PP-HGNetV2-B4",
|
|
|
|
|
+ "PP-HGNetV2-B5",
|
|
|
|
|
+ "PP-HGNetV2-B6",
|
|
|
|
|
+ "FasterNet-L",
|
|
|
|
|
+ "FasterNet-M",
|
|
|
|
|
+ "FasterNet-S",
|
|
|
|
|
+ "FasterNet-T0",
|
|
|
|
|
+ "FasterNet-T1",
|
|
|
|
|
+ "FasterNet-T2",
|
|
|
|
|
+ "StarNet-S1",
|
|
|
|
|
+ "StarNet-S2",
|
|
|
|
|
+ "StarNet-S3",
|
|
|
|
|
+ "StarNet-S4",
|
|
|
|
|
+ "MobileNetV4_conv_small",
|
|
|
|
|
+ "MobileNetV4_conv_medium",
|
|
|
|
|
+ "MobileNetV4_conv_large",
|
|
|
|
|
+ "MobileNetV4_hybrid_medium",
|
|
|
|
|
+ "MobileNetV4_hybrid_large",
|
|
|
|
|
+ "CLIP_vit_base_patch16_224",
|
|
|
|
|
+ "CLIP_vit_large_patch14_224",
|
|
|
|
|
+ "PP-LCNet_x1_0_ML",
|
|
|
|
|
+ "PP-HGNetV2-B0_ML",
|
|
|
|
|
+ "PP-HGNetV2-B4_ML",
|
|
|
|
|
+ "PP-HGNetV2-B6_ML",
|
|
|
|
|
+ "ResNet50_ML",
|
|
|
|
|
+ "CLIP_vit_base_patch16_448_ML",
|
|
|
|
|
+ "PP-YOLOE_plus-X",
|
|
|
|
|
+ "PP-YOLOE_plus-L",
|
|
|
|
|
+ "PP-YOLOE_plus-M",
|
|
|
|
|
+ "PP-YOLOE_plus-S",
|
|
|
|
|
+ "RT-DETR-L",
|
|
|
|
|
+ "RT-DETR-H",
|
|
|
|
|
+ "RT-DETR-X",
|
|
|
|
|
+ "YOLOv3-DarkNet53",
|
|
|
|
|
+ "YOLOv3-MobileNetV3",
|
|
|
|
|
+ "YOLOv3-ResNet50_vd_DCN",
|
|
|
|
|
+ "YOLOX-L",
|
|
|
|
|
+ "YOLOX-M",
|
|
|
|
|
+ "YOLOX-N",
|
|
|
|
|
+ "YOLOX-S",
|
|
|
|
|
+ "YOLOX-T",
|
|
|
|
|
+ "YOLOX-X",
|
|
|
|
|
+ "RT-DETR-R18",
|
|
|
|
|
+ "RT-DETR-R50",
|
|
|
|
|
+ "PicoDet-S",
|
|
|
|
|
+ "PicoDet-L",
|
|
|
|
|
+ "Deeplabv3-R50",
|
|
|
|
|
+ "Deeplabv3-R101",
|
|
|
|
|
+ "Deeplabv3_Plus-R50",
|
|
|
|
|
+ "Deeplabv3_Plus-R101",
|
|
|
|
|
+ "PP-ShiTuV2_rec",
|
|
|
|
|
+ "PP-ShiTuV2_rec_CLIP_vit_base",
|
|
|
|
|
+ "PP-ShiTuV2_rec_CLIP_vit_large",
|
|
|
|
|
+ "PP-LiteSeg-T",
|
|
|
|
|
+ "PP-LiteSeg-B",
|
|
|
|
|
+ "OCRNet_HRNet-W48",
|
|
|
|
|
+ "OCRNet_HRNet-W18",
|
|
|
|
|
+ "SegFormer-B0",
|
|
|
|
|
+ "SegFormer-B1",
|
|
|
|
|
+ "SegFormer-B2",
|
|
|
|
|
+ "SegFormer-B3",
|
|
|
|
|
+ "SegFormer-B4",
|
|
|
|
|
+ "SegFormer-B5",
|
|
|
|
|
+ "SeaFormer_tiny",
|
|
|
|
|
+ "SeaFormer_small",
|
|
|
|
|
+ "SeaFormer_base",
|
|
|
|
|
+ "SeaFormer_large",
|
|
|
|
|
+ "Mask-RT-DETR-H",
|
|
|
|
|
+ "Mask-RT-DETR-L",
|
|
|
|
|
+ "PP-OCRv4_server_rec",
|
|
|
|
|
+ "Mask-RT-DETR-S",
|
|
|
|
|
+ "Mask-RT-DETR-M",
|
|
|
|
|
+ "Mask-RT-DETR-X",
|
|
|
|
|
+ "SOLOv2",
|
|
|
|
|
+ "MaskRCNN-ResNet50",
|
|
|
|
|
+ "MaskRCNN-ResNet50-FPN",
|
|
|
|
|
+ "MaskRCNN-ResNet50-vd-FPN",
|
|
|
|
|
+ "MaskRCNN-ResNet101-FPN",
|
|
|
|
|
+ "MaskRCNN-ResNet101-vd-FPN",
|
|
|
|
|
+ "MaskRCNN-ResNeXt101-vd-FPN",
|
|
|
|
|
+ "Cascade-MaskRCNN-ResNet50-FPN",
|
|
|
|
|
+ "Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN",
|
|
|
|
|
+ "PP-YOLOE_seg-S",
|
|
|
|
|
+ "PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "en_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "korean_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "japan_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "chinese_cht_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "te_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "ka_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "ta_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "latin_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "arabic_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "cyrillic_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "devanagari_PP-OCRv3_mobile_rec",
|
|
|
|
|
+ "en_PP-OCRv4_mobile_rec",
|
|
|
|
|
+ "PP-OCRv4_server_rec_doc",
|
|
|
|
|
+ "PP-OCRv4_mobile_rec",
|
|
|
|
|
+ "PP-OCRv4_server_det",
|
|
|
|
|
+ "PP-OCRv4_mobile_det",
|
|
|
|
|
+ "PP-OCRv3_server_det",
|
|
|
|
|
+ "PP-OCRv3_mobile_det",
|
|
|
|
|
+ "PP-OCRv4_server_seal_det",
|
|
|
|
|
+ "PP-OCRv4_mobile_seal_det",
|
|
|
|
|
+ "ch_RepSVTR_rec",
|
|
|
|
|
+ "ch_SVTRv2_rec",
|
|
|
|
|
+ "PP-LCNet_x1_0_pedestrian_attribute",
|
|
|
|
|
+ "PP-LCNet_x1_0_vehicle_attribute",
|
|
|
|
|
+ "PicoDet_layout_1x",
|
|
|
|
|
+ "PicoDet_layout_1x_table",
|
|
|
|
|
+ "SLANet",
|
|
|
|
|
+ "SLANet_plus",
|
|
|
|
|
+ "LaTeX_OCR_rec",
|
|
|
|
|
+ "UniMERNet",
|
|
|
|
|
+ "PP-FormulaNet-S",
|
|
|
|
|
+ "PP-FormulaNet-L",
|
|
|
|
|
+ "PP-FormulaNet_plus-S",
|
|
|
|
|
+ "PP-FormulaNet_plus-M",
|
|
|
|
|
+ "PP-FormulaNet_plus-L",
|
|
|
|
|
+ "FasterRCNN-ResNet34-FPN",
|
|
|
|
|
+ "FasterRCNN-ResNet50",
|
|
|
|
|
+ "FasterRCNN-ResNet50-FPN",
|
|
|
|
|
+ "FasterRCNN-ResNet50-vd-FPN",
|
|
|
|
|
+ "FasterRCNN-ResNet50-vd-SSLDv2-FPN",
|
|
|
|
|
+ "FasterRCNN-ResNet101",
|
|
|
|
|
+ "FasterRCNN-ResNet101-FPN",
|
|
|
|
|
+ "FasterRCNN-ResNeXt101-vd-FPN",
|
|
|
|
|
+ "FasterRCNN-Swin-Tiny-FPN",
|
|
|
|
|
+ "Cascade-FasterRCNN-ResNet50-FPN",
|
|
|
|
|
+ "Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN",
|
|
|
|
|
+ "UVDoc",
|
|
|
|
|
+ "DLinear",
|
|
|
|
|
+ "NLinear",
|
|
|
|
|
+ "RLinear",
|
|
|
|
|
+ "Nonstationary",
|
|
|
|
|
+ "TimesNet",
|
|
|
|
|
+ "TiDE",
|
|
|
|
|
+ "PatchTST",
|
|
|
|
|
+ "DLinear_ad",
|
|
|
|
|
+ "AutoEncoder_ad",
|
|
|
|
|
+ "Nonstationary_ad",
|
|
|
|
|
+ "PatchTST_ad",
|
|
|
|
|
+ "TimesNet_ad",
|
|
|
|
|
+ "TimesNet_cls",
|
|
|
|
|
+ "STFPM",
|
|
|
|
|
+ "FCOS-ResNet50",
|
|
|
|
|
+ "DETR-R50",
|
|
|
|
|
+ "PP-YOLOE-L_vehicle",
|
|
|
|
|
+ "PP-YOLOE-S_vehicle",
|
|
|
|
|
+ "PP-ShiTuV2_det",
|
|
|
|
|
+ "PP-YOLOE-S_human",
|
|
|
|
|
+ "PP-YOLOE-L_human",
|
|
|
|
|
+ "PicoDet-M",
|
|
|
|
|
+ "PicoDet-XS",
|
|
|
|
|
+ "PP-YOLOE_plus_SOD-L",
|
|
|
|
|
+ "PP-YOLOE_plus_SOD-S",
|
|
|
|
|
+ "PP-YOLOE_plus_SOD-largesize-L",
|
|
|
|
|
+ "CenterNet-DLA-34",
|
|
|
|
|
+ "CenterNet-ResNet50",
|
|
|
|
|
+ "PicoDet-S_layout_3cls",
|
|
|
|
|
+ "PicoDet-S_layout_17cls",
|
|
|
|
|
+ "PicoDet-L_layout_3cls",
|
|
|
|
|
+ "PicoDet-L_layout_17cls",
|
|
|
|
|
+ "RT-DETR-H_layout_3cls",
|
|
|
|
|
+ "RT-DETR-H_layout_17cls",
|
|
|
|
|
+ "PicoDet_LCNet_x2_5_face",
|
|
|
|
|
+ "BlazeFace",
|
|
|
|
|
+ "BlazeFace-FPN-SSH",
|
|
|
|
|
+ "PP-YOLOE_plus-S_face",
|
|
|
|
|
+ "MobileFaceNet",
|
|
|
|
|
+ "ResNet50_face",
|
|
|
|
|
+ "PP-YOLOE-R-L",
|
|
|
|
|
+ "Co-Deformable-DETR-R50",
|
|
|
|
|
+ "Co-Deformable-DETR-Swin-T",
|
|
|
|
|
+ "Co-DINO-R50",
|
|
|
|
|
+ "Co-DINO-Swin-L",
|
|
|
|
|
+ "whisper_large",
|
|
|
|
|
+ "whisper_base",
|
|
|
|
|
+ "whisper_medium",
|
|
|
|
|
+ "whisper_small",
|
|
|
|
|
+ "whisper_tiny",
|
|
|
|
|
+ "PP-TSM-R50_8frames_uniform",
|
|
|
|
|
+ "PP-TSMv2-LCNetV2_8frames_uniform",
|
|
|
|
|
+ "PP-TSMv2-LCNetV2_16frames_uniform",
|
|
|
|
|
+ "MaskFormer_tiny",
|
|
|
|
|
+ "MaskFormer_small",
|
|
|
|
|
+ "PP-LCNet_x1_0_table_cls",
|
|
|
|
|
+ "SLANeXt_wired",
|
|
|
|
|
+ "SLANeXt_wireless",
|
|
|
|
|
+ "RT-DETR-L_wired_table_cell_det",
|
|
|
|
|
+ "RT-DETR-L_wireless_table_cell_det",
|
|
|
|
|
+ "YOWO",
|
|
|
|
|
+ "PP-TinyPose_128x96",
|
|
|
|
|
+ "PP-TinyPose_256x192",
|
|
|
|
|
+ "GroundingDINO-T",
|
|
|
|
|
+ "SAM-H_box",
|
|
|
|
|
+ "SAM-H_point",
|
|
|
|
|
+ "PP-DocLayout-L",
|
|
|
|
|
+ "PP-DocLayout-M",
|
|
|
|
|
+ "PP-DocLayout-S",
|
|
|
|
|
+ "PP-DocLayout_plus-L",
|
|
|
|
|
+ "PP-DocBlockLayout",
|
|
|
|
|
+ "BEVFusion",
|
|
|
|
|
+ "YOLO-Worldv2-L",
|
|
|
|
|
+ "PP-DocBee-2B",
|
|
|
|
|
+ "PP-DocBee-7B",
|
|
|
|
|
+ "PP-Chart2Table",
|
|
|
|
|
+ "PP-OCRv5_server_det",
|
|
|
|
|
+ "PP-OCRv5_mobile_det",
|
|
|
|
|
+ "PP-OCRv5_server_rec",
|
|
|
|
|
+ "PP-OCRv5_mobile_rec",
|
|
|
|
|
+ "eslav_PP-OCRv5_mobile_rec",
|
|
|
|
|
+ "PP-DocBee2-3B",
|
|
|
|
|
+ "latin_PP-OCRv5_mobile_rec",
|
|
|
|
|
+ "korean_PP-OCRv5_mobile_rec",
|
|
|
|
|
+]
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+OCR_MODELS = [
|
|
|
"arabic_PP-OCRv3_mobile_rec",
|
|
"arabic_PP-OCRv3_mobile_rec",
|
|
|
"chinese_cht_PP-OCRv3_mobile_rec",
|
|
"chinese_cht_PP-OCRv3_mobile_rec",
|
|
|
"ch_RepSVTR_rec",
|
|
"ch_RepSVTR_rec",
|
|
@@ -437,63 +385,190 @@ HUGGINGFACE_MODELS = [
|
|
|
]
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
|
-@lru_cache(1)
|
|
|
|
|
-def is_huggingface_accessible():
|
|
|
|
|
- try:
|
|
|
|
|
- response = requests.get("https://huggingface.co", timeout=1)
|
|
|
|
|
- return response.ok == True
|
|
|
|
|
- except requests.exceptions.RequestException as e:
|
|
|
|
|
- return False
|
|
|
|
|
|
|
+class _BaseModelHoster(ABC):
|
|
|
|
|
+ alias = ""
|
|
|
|
|
+ model_list = []
|
|
|
|
|
+ healthcheck_url = None
|
|
|
|
|
+ _healthcheck_timeout = 1
|
|
|
|
|
+
|
|
|
|
|
+ def __init__(self, save_dir):
|
|
|
|
|
+ self._save_dir = save_dir
|
|
|
|
|
+
|
|
|
|
|
+ def get_model(self, model_name):
|
|
|
|
|
+ assert (
|
|
|
|
|
+ model_name in self.model_list
|
|
|
|
|
+ ), f"The model {model_name} is not supported on hosting {self.__class__.__name__}!"
|
|
|
|
|
+ model_dir = self._save_dir / f"{model_name}"
|
|
|
|
|
+ self._download(model_name, model_dir)
|
|
|
|
|
+ return model_dir
|
|
|
|
|
+
|
|
|
|
|
+ @abstractmethod
|
|
|
|
|
+ def _download(self):
|
|
|
|
|
+ raise NotImplementedError
|
|
|
|
|
+
|
|
|
|
|
+ @classmethod
|
|
|
|
|
+ def is_available(cls):
|
|
|
|
|
+ if cls.healthcheck_url is None:
|
|
|
|
|
+ return True
|
|
|
|
|
+ try:
|
|
|
|
|
+ response = requests.head(
|
|
|
|
|
+ cls.healthcheck_url, timeout=cls._healthcheck_timeout
|
|
|
|
|
+ )
|
|
|
|
|
+ return response.ok == True
|
|
|
|
|
+ except Exception:
|
|
|
|
|
+ logging.debug(f"The model hosting platform({cls.__name__}) is unreachable!")
|
|
|
|
|
+ return False
|
|
|
|
|
|
|
|
|
|
|
|
|
-class OfficialModelsDict(dict):
|
|
|
|
|
- """Official Models Dict"""
|
|
|
|
|
|
|
+class _BosModelHoster(_BaseModelHoster):
|
|
|
|
|
+ model_list = ALL_MODELS
|
|
|
|
|
+ alias = "bos"
|
|
|
|
|
+ healthcheck_url = "https://paddle-model-ecology.bj.bcebos.com"
|
|
|
|
|
|
|
|
|
|
+ version = "paddle3.0.0"
|
|
|
|
|
+ base_url = (
|
|
|
|
|
+ "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model"
|
|
|
|
|
+ )
|
|
|
|
|
+ special_model_fn = {
|
|
|
|
|
+ "whisper_large": "whisper_large.tar",
|
|
|
|
|
+ "whisper_base": "whisper_base.tar",
|
|
|
|
|
+ "whisper_medium": "whisper_medium.tar",
|
|
|
|
|
+ "whisper_small": "whisper_small.tar",
|
|
|
|
|
+ "whisper_tiny": "whisper_tiny.tar",
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ def _download(self, model_name, save_dir):
|
|
|
|
|
+ if model_name in self.special_model_fn:
|
|
|
|
|
+ fn = self.special_model_fn[model_name]
|
|
|
|
|
+ else:
|
|
|
|
|
+ fn = f"{model_name}_infer.tar"
|
|
|
|
|
+ url = f"{self.base_url}/{self.version}/{fn}"
|
|
|
|
|
+ download_and_extract(url, save_dir.parent, model_name, overwrite=False)
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class _HuggingFaceModelHoster(_BaseModelHoster):
|
|
|
|
|
+ model_list = OCR_MODELS
|
|
|
|
|
+ alias = "huggingface"
|
|
|
|
|
+ healthcheck_url = "https://huggingface.co"
|
|
|
|
|
+
|
|
|
|
|
+ def _download(self, model_name, save_dir):
|
|
|
|
|
+ def _clone(local_dir):
|
|
|
|
|
+ hf_hub.snapshot_download(
|
|
|
|
|
+ repo_id=f"PaddlePaddle/{model_name}", local_dir=local_dir
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ if os.path.exists(save_dir):
|
|
|
|
|
+ _clone(save_dir)
|
|
|
|
|
+ else:
|
|
|
|
|
+ with tempfile.TemporaryDirectory() as td:
|
|
|
|
|
+ temp_dir = os.path.join(td, "temp_dir")
|
|
|
|
|
+ _clone(temp_dir)
|
|
|
|
|
+ shutil.move(temp_dir, save_dir)
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class _ModelScopeModelHoster(_BaseModelHoster):
|
|
|
|
|
+ model_list = OCR_MODELS
|
|
|
|
|
+ alias = "modelscope"
|
|
|
|
|
+ healthcheck_url = "https://modelscope.cn"
|
|
|
|
|
+
|
|
|
|
|
+ def _download(self, model_name, save_dir):
|
|
|
|
|
+ def _clone(local_dir):
|
|
|
|
|
+ modelscope.snapshot_download(
|
|
|
|
|
+ repo_id=f"PaddlePaddle/{model_name}", local_dir=local_dir
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ if os.path.exists(save_dir):
|
|
|
|
|
+ _clone(save_dir)
|
|
|
|
|
+ else:
|
|
|
|
|
+ with tempfile.TemporaryDirectory() as td:
|
|
|
|
|
+ temp_dir = os.path.join(td, "temp_dir")
|
|
|
|
|
+ _clone(temp_dir)
|
|
|
|
|
+ shutil.move(temp_dir, save_dir)
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class _AIStudioModelHoster(_BaseModelHoster):
|
|
|
|
|
+ model_list = OCR_MODELS
|
|
|
|
|
+ alias = "aistudio"
|
|
|
|
|
+ healthcheck_url = "https://aistudio.baidu.com"
|
|
|
|
|
+
|
|
|
|
|
+ def _download(self, model_name, save_dir):
|
|
|
|
|
+ def _clone(local_dir):
|
|
|
|
|
+ aistudio_download(repo_id=f"PaddleX/{model_name}", local_dir=local_dir)
|
|
|
|
|
+
|
|
|
|
|
+ if os.path.exists(save_dir):
|
|
|
|
|
+ _clone(save_dir)
|
|
|
|
|
+ else:
|
|
|
|
|
+ with tempfile.TemporaryDirectory() as td:
|
|
|
|
|
+ temp_dir = os.path.join(td, "temp_dir")
|
|
|
|
|
+ _clone(temp_dir)
|
|
|
|
|
+ shutil.move(temp_dir, save_dir)
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class _ModelManager:
|
|
|
|
|
+ model_list = ALL_MODELS
|
|
|
_save_dir = Path(CACHE_DIR) / "official_models"
|
|
_save_dir = Path(CACHE_DIR) / "official_models"
|
|
|
|
|
|
|
|
- def __getitem__(self, key):
|
|
|
|
|
- def _download_from_bos():
|
|
|
|
|
- url = super(OfficialModelsDict, self).__getitem__(key)
|
|
|
|
|
- download_and_extract(url, self._save_dir, f"{key}", overwrite=False)
|
|
|
|
|
- return self._save_dir / f"{key}"
|
|
|
|
|
-
|
|
|
|
|
- def _download_from_hf():
|
|
|
|
|
- local_dir = self._save_dir / f"{key}"
|
|
|
|
|
- try:
|
|
|
|
|
- if os.path.exists(local_dir):
|
|
|
|
|
- hf_hub.snapshot_download(
|
|
|
|
|
- repo_id=f"PaddlePaddle/{key}", local_dir=local_dir
|
|
|
|
|
- )
|
|
|
|
|
- else:
|
|
|
|
|
- with tempfile.TemporaryDirectory() as td:
|
|
|
|
|
- temp_dir = os.path.join(td, "temp_dir")
|
|
|
|
|
- hf_hub.snapshot_download(
|
|
|
|
|
- repo_id=f"PaddlePaddle/{key}", local_dir=temp_dir
|
|
|
|
|
- )
|
|
|
|
|
- shutil.move(temp_dir, local_dir)
|
|
|
|
|
- except Exception as e:
|
|
|
|
|
- logging.warning(
|
|
|
|
|
- f"Encounter exception when download model from huggingface: \n{e}.\nPaddleX would try to download from BOS."
|
|
|
|
|
- )
|
|
|
|
|
- return _download_from_bos()
|
|
|
|
|
- return local_dir
|
|
|
|
|
|
|
+ def __init__(self) -> None:
|
|
|
|
|
+ self._hosters = self._build_hosters()
|
|
|
|
|
|
|
|
|
|
+ def _build_hosters(self):
|
|
|
|
|
+ hosters = []
|
|
|
|
|
+ for hoster_cls in [
|
|
|
|
|
+ _HuggingFaceModelHoster,
|
|
|
|
|
+ _AIStudioModelHoster,
|
|
|
|
|
+ _ModelScopeModelHoster,
|
|
|
|
|
+ _BosModelHoster,
|
|
|
|
|
+ ]:
|
|
|
|
|
+ if hoster_cls.alias == MODEL_SOURCE:
|
|
|
|
|
+ if hoster_cls.is_available():
|
|
|
|
|
+ hosters.insert(0, hoster_cls(self._save_dir))
|
|
|
|
|
+ else:
|
|
|
|
|
+ if hoster_cls.is_available():
|
|
|
|
|
+ hosters.append(hoster_cls(self._save_dir))
|
|
|
|
|
+ if len(hosters) == 0:
|
|
|
|
|
+ logging.warning(
|
|
|
|
|
+ f"""No model hoster is available! Please check your network connection to one of the following model hosts:
|
|
|
|
|
+HuggingFace ({_HuggingFaceModelHoster.healthcheck_url}),
|
|
|
|
|
+ModelScope ({_ModelScopeModelHoster.healthcheck_url}),
|
|
|
|
|
+AIStudio ({_AIStudioModelHoster.healthcheck_url}), or
|
|
|
|
|
+BOS ({_BosModelHoster.healthcheck_url}).
|
|
|
|
|
+Otherwise, only local models can be used."""
|
|
|
|
|
+ )
|
|
|
|
|
+ return hosters
|
|
|
|
|
+
|
|
|
|
|
+ def _get_model_local_path(self, model_name):
|
|
|
logging.info(
|
|
logging.info(
|
|
|
- f"Using official model ({key}), the model files will be automatically downloaded and saved in {self._save_dir}."
|
|
|
|
|
|
|
+ f"Using official model ({model_name}), the model files will be automatically downloaded and saved in {self._save_dir}."
|
|
|
)
|
|
)
|
|
|
|
|
+ if len(self._hosters) == 0:
|
|
|
|
|
+ msg = "No available model hosting platforms detected. Please check your network connection."
|
|
|
|
|
+ logging.error(msg)
|
|
|
|
|
+ raise Exception(msg)
|
|
|
|
|
+ return self._download_from_hoster(self._hosters, model_name)
|
|
|
|
|
|
|
|
- if (
|
|
|
|
|
- MODEL_SOURCE.lower() == "huggingface"
|
|
|
|
|
- and is_huggingface_accessible()
|
|
|
|
|
- and key in HUGGINGFACE_MODELS
|
|
|
|
|
- ):
|
|
|
|
|
- return _download_from_hf()
|
|
|
|
|
- elif MODEL_SOURCE.lower() == "modelscope":
|
|
|
|
|
- raise Exception(
|
|
|
|
|
- f"ModelScope is not supported! Please use `HuggingFace` or `BOS`."
|
|
|
|
|
- )
|
|
|
|
|
- else:
|
|
|
|
|
- return _download_from_bos()
|
|
|
|
|
|
|
+ def _download_from_hoster(self, hosters, model_name):
|
|
|
|
|
+ for idx, hoster in enumerate(hosters):
|
|
|
|
|
+ if model_name in hoster.model_list:
|
|
|
|
|
+ try:
|
|
|
|
|
+ return hoster.get_model(model_name)
|
|
|
|
|
+ except Exception as e:
|
|
|
|
|
+ logging.warning(
|
|
|
|
|
+ f"Encounter exception when download model from {hoster.alias}: \n{e}."
|
|
|
|
|
+ )
|
|
|
|
|
+ if len(hosters) <= 1:
|
|
|
|
|
+ raise Exception(
|
|
|
|
|
+ f"No model source is available! Please check network or use local model files!"
|
|
|
|
|
+ )
|
|
|
|
|
+ logging.warning(
|
|
|
|
|
+ f"PaddleX would try to download from other model sources."
|
|
|
|
|
+ )
|
|
|
|
|
+ return self._download_from_hoster(hosters[idx + 1 :], model_name)
|
|
|
|
|
+
|
|
|
|
|
+ def __contains__(self, model_name):
|
|
|
|
|
+ return model_name in self.model_list
|
|
|
|
|
+
|
|
|
|
|
+ def __getitem__(self, model_name):
|
|
|
|
|
+ return self._get_model_local_path(model_name)
|
|
|
|
|
|
|
|
|
|
|
|
|
-official_models = OfficialModelsDict(OFFICIAL_MODELS)
|
|
|
|
|
|
|
+official_models = _ModelManager()
|