tflite_support.task.audio.AudioClassifier

Class that performs classification on audio.

tflite_support.task.audio.AudioClassifier(
    options: tflite_support.task.audio.AudioClassifierOptions,
    classifier: _CppAudioClassifier
) -> None

required_audio_format Gets the required audio format for the model.
required_input_buffer_size Gets the required input buffer size for the model.

Methods

classify

View source

classify(
    audio: tflite_support.task.audio.TensorAudio
) -> tflite_support.task.processor.ClassificationResult

Performs classification on the provided TensorAudio.

Args
audio Tensor audio, used to extract the feature vectors.

Returns
classification result.

Raises
ValueError If any of the input arguments is invalid.
RuntimeError If failed to run audio classification.

create_audio_record

View source

create_audio_record() -> tflite_support.task.audio.AudioRecord

Creates an AudioRecord instance to record audio.

Returns
An AudioRecord instance.

create_from_file

View source

@classmethod
create_from_file(
    file_path: str
) -> 'AudioClassifier'

Creates the AudioClassifier object from a TensorFlow Lite model.

Args
file_path Path to the model.

Returns
AudioClassifier object that's created from options.

Raises
ValueError If failed to create AudioClassifier object from the provided file such as invalid file.
RuntimeError If other types of error occurred.

create_from_options

View source

@classmethod
create_from_options(
    options: tflite_support.task.audio.AudioClassifierOptions
) -> 'AudioClassifier'

Creates the AudioClassifier object from audio classifier options.

Args
options Options for the audio classifier task.

Returns
AudioClassifier object that's created from options.

Raises
ValueError If failed to create AudioClassifier object from AudioClassifierOptions such as missing the model.
RuntimeError If other types of error occurred.

create_input_tensor_audio

View source

create_input_tensor_audio() -> tflite_support.task.audio.TensorAudio

Creates a TensorAudio instance to store the audio input.

Returns
A TensorAudio instance.