Loosen the constraint for score_threshold in ImageClassifier

PiperOrigin-RevId: 373805511

Author: lu-wang-g

tensorflow_lite_support/cc/task/vision/image_classifier.cc

@@ -103,14 +103,6 @@ absl::Status ImageClassifier::SanityCheckOptions(
         "Invalid `max_results` option: value must be != 0",
         TfLiteSupportStatus::kInvalidArgumentError);
   }
-  if (options.score_threshold() < 0 || options.score_threshold() >= 1) {
-    return CreateStatusWithPayload(
-        StatusCode::kInvalidArgument,
-        absl::StrFormat(
-            "`score_threshold` out of range: %f. Valid range is [0,1[.",
-            options.score_threshold()),
-        TfLiteSupportStatus::kInvalidArgumentError);
-  }
   if (options.class_name_whitelist_size() > 0 &&
       options.class_name_blacklist_size() > 0) {
     return CreateStatusWithPayload(

tensorflow_lite_support/cc/task/vision/proto/image_classifier_options.proto

@@ -43,7 +43,7 @@ message ImageClassifierOptions {
   // returned.
   optional int32 max_results = 2 [default = -1];
 
-  // Score threshold in [0,1), overrides the ones provided in the model metadata
+  // Score threshold, overrides the ones provided in the model metadata
   // (if any). Results below this value are rejected.
   optional float score_threshold = 3;
 

tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/classifier/ImageClassifier.java

@@ -275,7 +275,7 @@ public Builder setMaxResults(int maxResults) {
       }
 
       /**
-       * Sets the score threshold in [0,1).
+       * Sets the score threshold.
        *
        * <p>It overrides the one provided in the model metadata (if any). Results below this value
        * are rejected.