|
@@ -23,7 +23,6 @@ from .config import ClsConfig
|
|
|
|
|
|
|
|
REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLECLAS_PATH")
|
|
REPO_ROOT_PATH = os.environ.get("PADDLE_PDX_PADDLECLAS_PATH")
|
|
|
PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))
|
|
PDX_CONFIG_DIR = osp.abspath(osp.join(osp.dirname(__file__), "..", "configs"))
|
|
|
-HPI_CONFIG_DIR = Path(__file__).parent.parent.parent.parent / "utils" / "hpi_configs"
|
|
|
|
|
|
|
|
|
|
register_suite_info(
|
|
register_suite_info(
|
|
|
{
|
|
{
|
|
@@ -45,8 +44,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_tiny_patch4_window7_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -59,8 +56,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_small_patch4_window7_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -72,8 +67,6 @@ register_model_info(
|
|
|
PDX_CONFIG_DIR, "SwinTransformer_base_patch4_window7_224.yaml"
|
|
PDX_CONFIG_DIR, "SwinTransformer_base_patch4_window7_224.yaml"
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_base_patch4_window7_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -86,8 +79,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_base_patch4_window12_384.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -100,8 +91,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_large_patch4_window7_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -114,8 +103,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR
|
|
|
|
|
- / "SwinTransformer_large_patch4_window12_384.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -125,7 +112,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_25.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_25.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x0_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -135,7 +121,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_35.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_35.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x0_35.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -145,7 +130,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x0_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -155,7 +139,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_75.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x0_75.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x0_75.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -165,7 +148,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x1_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -175,7 +157,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_doc_ori.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_doc_ori.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x1_0_doc_ori.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -185,7 +166,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x1_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -195,7 +175,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x2_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x2_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x2_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -205,7 +184,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x2_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x2_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNet_x2_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -215,7 +193,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_small.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_small.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNetV2_small.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -225,7 +202,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_base.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_base.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNetV2_base.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -235,7 +211,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_large.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNetV2_large.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-LCNetV2_large.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -245,7 +220,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_base_patch16_224.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_base_patch16_224.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "CLIP_vit_base_patch16_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -255,7 +229,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_large_patch14_224.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_large_patch14_224.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "CLIP_vit_large_patch14_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -265,7 +238,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_tiny.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_tiny.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNet_tiny.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -275,7 +247,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_small.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_small.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNet_small.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -285,7 +256,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_base.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNet_base.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNet_base.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -295,7 +265,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -305,7 +274,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B1.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B1.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B1.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -315,7 +283,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B2.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B2.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B2.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -325,7 +292,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B3.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B3.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B3.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -335,7 +301,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B4.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B4.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B4.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -345,7 +310,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -355,7 +319,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B6.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B6.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "PP-HGNetV2-B6.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -365,7 +328,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet18.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet18.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet18.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -375,7 +337,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet18_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet18_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet18_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -385,7 +346,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet34.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet34.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet34.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -395,7 +355,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet34_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet34_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet34_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -405,7 +364,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet50.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -415,7 +373,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet50_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -425,7 +382,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet101.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet101.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet101.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -435,7 +391,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet101_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet101_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet101_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -445,7 +400,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet152.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet152.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet152.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -455,7 +409,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet152_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet152_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet152_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -465,7 +418,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet200_vd.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet200_vd.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ResNet200_vd.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -476,7 +428,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_25.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_25.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV1_x0_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -487,7 +438,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV1_x0_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -498,7 +448,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_75.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x0_75.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV1_x0_75.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -509,7 +458,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x1_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV1_x1_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV1_x1_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -519,7 +467,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x0_25.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x0_25.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV2_x0_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -529,7 +476,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x0_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x0_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV2_x0_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -539,7 +485,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x1_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x1_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV2_x1_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -549,7 +494,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x1_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x1_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV2_x1_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -559,7 +503,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x2_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV2_x2_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV2_x2_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -569,7 +512,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_35.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_35.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x0_35.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -579,7 +521,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x0_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -589,7 +530,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_75.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x0_75.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x0_75.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -599,7 +539,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x1_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x1_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x1_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -609,7 +548,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x1_25.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_large_x1_25.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x1_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -619,7 +557,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_35.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_35.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_large_x1_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -629,7 +566,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_5.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_5.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_small_x0_5.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -639,7 +575,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_75.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x0_75.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_small_x0_75.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -649,7 +584,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x1_0.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x1_0.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_small_x1_0.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -659,7 +593,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x1_25.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "MobileNetV3_small_x1_25.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "MobileNetV3_small_x1_25.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -714,7 +647,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_tiny.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_tiny.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_tiny.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -724,7 +656,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_small.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_small.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_small.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -734,7 +665,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_base_224.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_base_224.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_base_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -744,7 +674,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_base_384.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_base_384.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_base_384.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -754,7 +683,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_large_224.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_large_224.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_large_224.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -764,7 +692,6 @@ register_model_info(
|
|
|
"suite": "Cls",
|
|
"suite": "Cls",
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_large_384.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ConvNeXt_large_384.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export"],
|
|
|
- "hpi_config_path": HPI_CONFIG_DIR / "ConvNeXt_large_384.yaml",
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -865,7 +792,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -876,7 +802,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "ResNet50_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -887,7 +812,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B0_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B0_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -898,7 +822,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B4_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B4_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -909,7 +832,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B6_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-HGNetV2-B6_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -920,7 +842,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_base_patch16_448_ML.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "CLIP_vit_base_patch16_448_ML.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -933,7 +854,6 @@ register_model_info(
|
|
|
),
|
|
),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -944,7 +864,6 @@ register_model_info(
|
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_vehicle_attribute.yaml"),
|
|
"config_path": osp.join(PDX_CONFIG_DIR, "PP-LCNet_x1_0_vehicle_attribute.yaml"),
|
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
"supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
"infer_config": "deploy/configs/inference_cls.yaml",
|
|
|
- "hpi_config_path": None,
|
|
|
|
|
}
|
|
}
|
|
|
)
|
|
)
|
|
|
|
|
|