talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

API for c++ ImageSegmenter to get labels

Browse Source 
PiperOrigin-RevId: 516714139
This commit is contained in:
MediaPipe Team
2023-03-14 21:13:08 -07:00
committed by Copybara-Service
parent cafff14135
commit f517eddce1
4 changed files with 95 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/cc/vision/image_segmenter/image_segmenter.h
+12
View File
@@ -189,6 +189,18 @@ class ImageSegmenter : tasks::vision::core::BaseVisionTaskApi {
// Shuts down the ImageSegmenter when all works are done.
absl::Status Close() { return runner_->Close(); }
// Get the category label list of the ImageSegmenter can recognize. For
// CATEGORY_MASK type, the index in the category mask corresponds to the
// category in the label list. For CONFIDENCE_MASK type, the output mask list
// at index corresponds to the category in the label list.
//
// If there is no labelmap provided in the model file, empty label list is
// returned.
std::vector<std::string> GetLabels() { return labels_; }
private:
std::vector<std::string> labels_;
};
} // namespace image_segmenter