Home
last modified time | relevance | path

Searched refs:colored_label (Results 1 – 3 of 3) sorted by relevance

/external/tflite-support/tensorflow_lite_support/examples/task/vision/desktop/
Dimage_segmenter_demo.cc84 Segmentation::ColoredLabel colored_label = in EncodeMaskToPngFile() local
86 pixel_data[3 * i] = colored_label.r(); in EncodeMaskToPngFile()
87 pixel_data[3 * i + 1] = colored_label.g(); in EncodeMaskToPngFile()
88 pixel_data[3 * i + 2] = colored_label.b(); in EncodeMaskToPngFile()
113 Segmentation::ColoredLabel colored_label = in DisplayColorLegend() local
116 colored_label.r(), colored_label.g(), in DisplayColorLegend()
117 colored_label.b()); in DisplayColorLegend()
119 if (colored_label.has_class_name()) { in DisplayColorLegend()
121 colored_label.class_name()); in DisplayColorLegend()
123 if (colored_label.has_display_name()) { in DisplayColorLegend()
[all …]
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/segmenter/
Dimage_segmenter_jni.cc143 for (const auto& colored_label : segmentation.colored_labels()) { in ConvertToSegmentationResults() local
144 jstring label = env->NewStringUTF(colored_label.class_name().c_str()); in ConvertToSegmentationResults()
146 env->NewStringUTF(colored_label.display_name().c_str()); in ConvertToSegmentationResults()
148 colored_label.r(), colored_label.g(), in ConvertToSegmentationResults()
149 colored_label.b()); in ConvertToSegmentationResults()
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_segmenter.cc290 Segmentation::ColoredLabel colored_label; in InitColoredLabels() local
291 colored_label.set_r(kColorMap[3 * i]); in InitColoredLabels()
292 colored_label.set_g(kColorMap[3 * i + 1]); in InitColoredLabels()
293 colored_label.set_b(kColorMap[3 * i + 2]); in InitColoredLabels()
296 colored_label.set_class_name(item.name); in InitColoredLabels()
299 colored_label.set_display_name(item.display_name); in InitColoredLabels()
301 colored_labels_.push_back(colored_label); in InitColoredLabels()