File tree Expand file tree Collapse file tree 1 file changed +2
-20
lines changed
tensorflow_lite_support/c/task/processor Expand file tree Collapse file tree 1 file changed +2
-20
lines changed Original file line number Diff line number Diff line change @@ -15,32 +15,14 @@ limitations under the License.
1515#ifndef TENSORFLOW_LITE_SUPPORT_C_TASK_PROCESSOR_CLASSIFICATION_RESULT_H_
1616#define TENSORFLOW_LITE_SUPPORT_C_TASK_PROCESSOR_CLASSIFICATION_RESULT_H_
1717
18+ #include "tensorflow_lite_support/c/task/processor/category.h"
19+
1820// Defines C structure for Classification Results and associated helper methods.
1921
2022#ifdef __cplusplus
2123extern "C" {
2224#endif // __cplusplus
2325
24- // A single predicted class.
25- typedef struct TfLiteCategory {
26- // The index of the class in the corresponding label map, usually packed in
27- // the TFLite Model Metadata [1].
28- //
29- // [1]: https://www.tensorflow.org/lite/convert/metadata
30- int index ;
31-
32- // The score for this class e.g. (but not necessarily) a probability in [0,1].
33- float score ;
34-
35- // A human readable name of the class filled from the label map.
36- char * display_name ;
37- // An ID for the class, not necessarily human-readable (e.g. a Google
38- // Knowledge Graph ID [1]), filled from the label map.
39- //
40- // [1]: https://developers.google.com/knowledge-graph
41- char * label ;
42- } TfLiteCategory ;
43-
4426// List of predicted classes (aka labels) for a given image classifier head.
4527typedef struct TfLiteClassifications {
4628 // The index of the image classifier head these classes refer to. This is
You can’t perform that action at this time.
0 commit comments