Как сделать label.txt из модели keras

Question

Обучал модель на керасе на двух классах. Затем я преобразовал его в .tflite с помощью этого кода:

# Converting a SavedModel to a TensorFlow Lite model. 
saved_model_dir = r'C:\Users\Munib\New folder\my_model.h5' 
loaded_model = tf.keras.models.load_model(saved_model_dir)
converter = tf.lite.TFLiteConverter.from_keras_model(loaded_model)# .from_saved_model(saved_model_dir) 
tflite_model = converter.convert()
open("my_model_converted_model.tflite", "wb").write(tflite_model)

Когда я изучал подробнее inte rnet о том, как импортировать модель .tflite в android studio. Они сказали, что у меня должно быть два файла:

1. my_model_converted_model.tflite

2. label.txt

Я сейчас не понимаю, как делать этикетки из кераса, может кто-нибудь подскажет Спасибо

Answer 1

Если ваша модель имеет 2 класса, выходные данные вашей модели могут выглядеть следующим образом - например: [0.12, 0.88]. Если первый результат, то есть 0,12, соответствует вероятности того, что прогноз «Да», а второй результат, то есть 0,88, соответствует вероятности того, что прогноз равен «Сейчас», тогда создайте новый файл с именем «labels.txt» "с содержанием:

Yes
No

Вот и все! Это так просто :)

Отметьте это как принятый ответ , если это решило вашу проблему :)