Class that predicts the language of an input text.
mp
.
tasks
.
text
.
LanguageDetector
(
graph_config
:
mp
.
calculators
.
core
.
constant_side_packet_calculator_pb2
.
mediapipe_dot_framework_dot_calculator__pb2
.
CalculatorGraphConfig
)
->
None
This API expects a TFLite model with TFLite Model Metadata that contains the mandatory (described below) input tensors, output tensor, and the language codes in an AssociatedFile.
Input tensors
(kTfLiteString)
- 1 input tensor that is scalar or has shape [1] containing the input string.
Output tensor
(kTfLiteFloat32)
- 1 output tensor of shape
[1 x N]whereNis the number of languages.
Methods
close
close
()
->
None
Shuts down the mediapipe text task instance.
Raises
RuntimeError
If the mediapipe text task failed to close.
create_from_model_path
@classmethodcreate_from_model_path ( model_path : str ) -> 'LanguageDetector'
Creates an LanguageDetector
object from a TensorFlow Lite model and the default LanguageDetectorOptions
.
Args
model_path
Path to the model.
Returns
LanguageDetector
object that's created from the model file and the
default LanguageDetectorOptions
.
Raises
ValueError
If failed to create
LanguageDetector
object from the
provided
file such as invalid file path.RuntimeError
If other types of error occurred.
create_from_options
@classmethodcreate_from_options ( options :mp . tasks . text . LanguageDetectorOptions) -> 'LanguageDetector'
Creates the LanguageDetector
object from language detector options.
Args
options
Options for the language detector task.
Returns
LanguageDetector
object that's created from options
.
Raises
ValueError
If failed to create
LanguageDetector
object from LanguageDetectorOptions
such as missing the model.RuntimeError
If other types of error occurred.
detect
detect
(
text
:
str
)
->
mp
.
tasks
.
text
.
LanguageDetectorResult
Predicts the language of the input text
.
Returns
A
LanguageDetectorResult
object that contains a list of languages and
scores.
Raises
ValueError
If any of the input arguments is invalid.
RuntimeError
If language detection failed to run.
__enter__
__enter__
()
Returns self
upon entering the runtime context.
__exit__
__exit__
(
unused_exc_type
,
unused_exc_value
,
unused_traceback
)
Shuts down the mediapipe text task instance on exit of the context manager.
Raises
RuntimeError
If the mediapipe text task failed to close.
