mumutu66 9 kuukautta sitten
vanhempi
commit
4c0bd6f8b4
1 muutettua tiedostoa jossa 5 lisäystä ja 5 poistoa
  1. 5 5
      paddlex/utils/pipeline_arguments.py

+ 5 - 5
paddlex/utils/pipeline_arguments.py

@@ -93,7 +93,7 @@ PIPELINE_ARGUMENTS = {
     "object_detection": [
         {
             "name": "--threshold",
-            "type": custom_type(Optional[Union[float, dict[int, float]]]),
+            "type": custom_type(Optional[Union[float, Dict[int, float]]]),
             "help": "Sets the threshold for object detection.",
         },
     ],
@@ -604,7 +604,7 @@ PIPELINE_ARGUMENTS = {
     "small_object_detection": [
         {
             "name": "--threshold",
-            "type": custom_type(Optional[Union[float, dict[int, float]]]),
+            "type": custom_type(Optional[Union[float, Dict[int, float]]]),
             "help": "Sets the threshold for small object detection.",
         },
     ],
@@ -643,14 +643,14 @@ PIPELINE_ARGUMENTS = {
     "rotated_object_detection": [
         {
             "name": "--threshold",
-            "type": custom_type(Optional[Union[float, dict[int, float]]]),
+            "type": custom_type(Optional[Union[float, Dict[int, float]]]),
             "help": "Sets the threshold for rotated object detection.",
         },
     ],
     "open_vocabulary_detection": [
         {
             "name": "--thresholds",
-            "type": custom_type(dict[str, float]),
+            "type": custom_type(Dict[str, float]),
             "help": "Sets the thresholds for open vocabulary detection.",
         },
         {
@@ -667,7 +667,7 @@ PIPELINE_ARGUMENTS = {
         },
         {
             "name": "--prompt",
-            "type": custom_type(list[list[float]]),
+            "type": custom_type(List[List[float]]),
             "help": "Sets the prompt for open vocabulary segmentation.",
         },
     ],