What I would, while it’s not available on the ImageClassifier, is to copy what AudioClassifier is doing: https://github.com/tensorflow/examples/blob/1ff6d7aa9bdf9749ada28bf67ad435ca518b6232/tensorflow_examples/lite/model_maker/core/task/audio_classifier.py#L86 3
the code might be easy to copy/adapt