PiperOrigin-RevId: 491434987
This commit is contained in:
Sebastian Schmidt
2022-11-28 13:54:09 -08:00
committed by Copybara-Service
parent c48ca1f674
commit 342f95fa20
2 changed files with 2 additions and 2 deletions
@@ -110,7 +110,7 @@ class ImageSegmenter(base_vision_task_api.BaseVisionTaskApi):
- list of segmented masks.
- if `output_type` is CATEGORY_MASK, uint8 Image, Image vector of size 1.
- if `output_type` is CONFIDENCE_MASK, float32 Image list of size
`cahnnels`.
`channels`.
- batch is always 1
An example of such model can be found at: