| /* Copyright 2020 The TensorFlow Authors. All Rights Reserved. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| ==============================================================================*/ |
| |
| syntax = "proto2"; |
| |
| package tflite.task.vision; |
| |
| import "tensorflow_lite_support/cc/task/core/proto/external_file.proto"; |
| |
| // Options for setting up an ImageClassifier. |
| // Next Id: 14 |
| message ImageClassifierOptions { |
| // The external model file, as a single standalone TFLite file. If it is |
| // packed with TFLite Model Metadata [1], those are used to populate e.g. the |
| // label map, score calibration and recommended score thresholds. Models |
| // without any such metadata or partial metadata are supported, but may result |
| // in the image classifier providing degraded functionality; typically, a |
| // model that doesn't contain any label map won't be able to return any class |
| // or display names but will be limited to returning class indices. |
| // |
| // [1]: https://www.tensorflow.org/lite/convert/metadata |
| optional core.ExternalFile model_file_with_metadata = 10; |
| |
| // The locale to use for display names specified through the TFLite Model |
| // Metadata, if any. Defaults to English. |
| optional string display_names_locale = 11 [default = "en"]; |
| |
| // The maximum number of top-scored classification results to return. If < 0, |
| // all available results will be returned. If 0, an invalid argument error is |
| // returned. |
| optional int32 max_results = 2 [default = -1]; |
| |
| // Score threshold in [0,1), overrides the ones provided in the model metadata |
| // (if any). Results below this value are rejected. |
| optional float score_threshold = 3; |
| |
| // Optional whitelist of class names. If non-empty, classifications whose |
| // class name is not in this set will be filtered out. Duplicate or unknown |
| // class names are ignored. Mutually exclusive with class_name_blacklist. |
| repeated string class_name_whitelist = 4; |
| |
| // Optional blacklist of class names. If non-empty, classifications whose |
| // class name is in this set will be filtered out. Duplicate or unknown |
| // class names are ignored. Mutually exclusive with class_name_whitelist. |
| repeated string class_name_blacklist = 5; |
| |
| // The number of threads to be used for TFLite ops that support |
| // multi-threading when running inference with CPU. |
| // num_threads should be greater than 0 or equal to -1. Setting num_threads to |
| // -1 has the effect to let TFLite runtime set the value. |
| optional int32 num_threads = 13 [default = -1]; |
| |
| // Reserved tags. |
| reserved 1, 6, 7, 8, 9, 12; |
| } |