|
|
@@ -377,15 +377,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet34-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet34-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet34-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet34-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -393,15 +393,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet50',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet50.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet50",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet50.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -409,15 +409,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet50-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet50-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet50-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet50-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -425,15 +425,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet50-vd-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet50-vd-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet50-vd-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet50-vd-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -441,15 +441,17 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet50-vd-SSLDv2-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet50-vd-SSLDv2-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(
|
|
|
+ PDX_CONFIG_DIR, "FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml"
|
|
|
+ ),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -457,15 +459,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet101',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet101.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet101",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet101.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -473,15 +475,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNet101-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNet101-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNet101-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNet101-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -489,15 +491,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-ResNeXt101-vd-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-ResNeXt101-vd-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-ResNeXt101-vd-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-ResNeXt101-vd-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -505,15 +507,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FasterRCNN-Swin-Tiny-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FasterRCNN-Swin-Tiny-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FasterRCNN-Swin-Tiny-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FasterRCNN-Swin-Tiny-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -521,15 +523,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'Cascade-FasterRCNN-ResNet50-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'Cascade-FasterRCNN-ResNet50-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "Cascade-FasterRCNN-ResNet50-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "Cascade-FasterRCNN-ResNet50-FPN.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -537,15 +539,17 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(
|
|
|
+ PDX_CONFIG_DIR, "Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml"
|
|
|
+ ),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -553,15 +557,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PicoDet-XS',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PicoDet-XS.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PicoDet-XS",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PicoDet-XS.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -569,15 +573,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PicoDet-M',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PicoDet-M.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PicoDet-M",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PicoDet-M.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -585,15 +589,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'FCOS-ResNet50',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'FCOS-ResNet50.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "FCOS-ResNet50",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "FCOS-ResNet50.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -601,31 +605,31 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'DETR-R50',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'DETR-R50.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "DETR-R50",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "DETR-R50.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
|
|
|
|
|
|
register_model_info(
|
|
|
- {
|
|
|
- 'model_name': 'PP-YOLOE-L_vehicle',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE-L_vehicle.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ {
|
|
|
+ "model_name": "PP-YOLOE-L_vehicle",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE-L_vehicle.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -633,15 +637,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE-S_vehicle',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE-S_vehicle.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE-S_vehicle",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE-S_vehicle.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -649,15 +653,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-ShiTuV2_det',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-ShiTuV2_det.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-ShiTuV2_det",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-ShiTuV2_det.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -665,15 +669,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE-L_human',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE-L_human.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE-L_human",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE-L_human.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -681,31 +685,31 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE-S_human',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE-S_human.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE-S_human",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE-S_human.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
-)
|
|
|
+)
|
|
|
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'CenterNet-DLA-34',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'CenterNet-DLA-34.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "CenterNet-DLA-34",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "CenterNet-DLA-34.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -713,15 +717,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'CenterNet-ResNet50',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'CenterNet-ResNet50.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "CenterNet-ResNet50",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "CenterNet-ResNet50.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -729,15 +733,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE_plus_SOD-L',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE_plus_SOD-L.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE_plus_SOD-L",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE_plus_SOD-L.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -745,15 +749,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE_plus_SOD-S',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE_plus_SOD-S.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE_plus_SOD-S",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE_plus_SOD-S.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|
|
|
@@ -761,15 +765,15 @@ register_model_info(
|
|
|
|
|
|
register_model_info(
|
|
|
{
|
|
|
- 'model_name': 'PP-YOLOE_plus_SOD-largesize-L',
|
|
|
- 'suite': 'Det',
|
|
|
- 'config_path': osp.join(PDX_CONFIG_DIR, 'PP-YOLOE_plus_SOD-largesize-L.yaml'),
|
|
|
- 'supported_apis': ['train', 'evaluate', 'predict', 'export', 'infer'],
|
|
|
- 'supported_dataset_types': ['COCODetDataset'],
|
|
|
- 'supported_train_opts': {
|
|
|
- 'device': ['cpu', 'gpu_nxcx', 'xpu', 'npu', 'mlu'],
|
|
|
- 'dy2st': False,
|
|
|
- 'amp': ['OFF']
|
|
|
+ "model_name": "PP-YOLOE_plus_SOD-largesize-L",
|
|
|
+ "suite": "Det",
|
|
|
+ "config_path": osp.join(PDX_CONFIG_DIR, "PP-YOLOE_plus_SOD-largesize-L.yaml"),
|
|
|
+ "supported_apis": ["train", "evaluate", "predict", "export", "infer"],
|
|
|
+ "supported_dataset_types": ["COCODetDataset"],
|
|
|
+ "supported_train_opts": {
|
|
|
+ "device": ["cpu", "gpu_nxcx", "xpu", "npu", "mlu"],
|
|
|
+ "dy2st": False,
|
|
|
+ "amp": ["OFF"],
|
|
|
},
|
|
|
}
|
|
|
)
|