PiperOrigin-RevId: 491434987
This commit is contained in:
Sebastian Schmidt 2022-11-28 13:51:59 -08:00 committed by Copybara-Service
parent c48ca1f674
commit 342f95fa20
2 changed files with 2 additions and 2 deletions

View File

@ -98,7 +98,7 @@ struct ImageSegmenterOptions {
// - list of segmented masks. // - list of segmented masks.
// - if `output_type` is CATEGORY_MASK, uint8 Image, Image vector of size 1. // - if `output_type` is CATEGORY_MASK, uint8 Image, Image vector of size 1.
// - if `output_type` is CONFIDENCE_MASK, float32 Image list of size // - if `output_type` is CONFIDENCE_MASK, float32 Image list of size
// `cahnnels`. // `channels`.
// - batch is always 1 // - batch is always 1
// An example of such model can be found at: // An example of such model can be found at:
// https://tfhub.dev/tensorflow/lite-model/deeplabv3/1/metadata/2 // https://tfhub.dev/tensorflow/lite-model/deeplabv3/1/metadata/2

View File

@ -110,7 +110,7 @@ class ImageSegmenter(base_vision_task_api.BaseVisionTaskApi):
- list of segmented masks. - list of segmented masks.
- if `output_type` is CATEGORY_MASK, uint8 Image, Image vector of size 1. - if `output_type` is CATEGORY_MASK, uint8 Image, Image vector of size 1.
- if `output_type` is CONFIDENCE_MASK, float32 Image list of size - if `output_type` is CONFIDENCE_MASK, float32 Image list of size
`cahnnels`. `channels`.
- batch is always 1 - batch is always 1
An example of such model can be found at: An example of such model can be found at: