Browse Source

add CLI pipline params for ml_cls and attr rec (#2844)

zhangyubo0722 10 months ago
parent
commit
62f36b9e08
1 changed files with 31 additions and 0 deletions
  1. 31 0
      paddlex/utils/pipeline_arguments.py

+ 31 - 0
paddlex/utils/pipeline_arguments.py

@@ -84,5 +84,36 @@ PIPELINE_ARGUMENTS = {
             "help": "Sets the Top-K value for image classification.",
         },
     ],
+    "image_multilabel_classification": [
+        {
+            "name": "--threshold",
+            "type": float,
+            "help": "Sets the threshold for image multilabel classification.",
+        },
+    ],
+    "pedestrian_attribute_recognition": [
+        {
+            "name": "--det_threshold",
+            "type": float,
+            "help": "Sets the threshold for human detection.",
+        },
+        {
+            "name": "--cls_threshold",
+            "type": float,
+            "help": "Sets the threshold for pedestrian attribute recognition.",
+        },
+    ],
+    "vehicle_attribute_recognition": [
+        {
+            "name": "--det_threshold",
+            "type": float,
+            "help": "Sets the threshold for vehicle detection.",
+        },
+        {
+            "name": "--cls_threshold",
+            "type": float,
+            "help": "Sets the threshold for vehicle attribute recognition.",
+        },
+    ],
     "ts_classification": None,
 }