Можно ли создать label.txt вручную?

Question

Я недавно конвертировал свою модель в программу tenorflow lite, но у меня есть только файл .tflite, а не label.txt для моего Android проекта. Так возможно ли создать мой собственный label.txt, используя классы, которые я использовал для классификации? Если нет, то как сгенерировать label.txt?

Answer 1

Запустите следующий код после установки библиотеки TFLite Model Maker и передайте набор данных для классификации:

data = ImageClassifierDataLoader.from_folder('folder/')
train_data, test_data = data.split(0.8)
model = image_classifier.create(train_data)
loss, accuracy = model.evaluate(test_data)
model.export('image_classifier.tflite', 'imageLabels.txt')

При запуске его в Colab или локально файлы меток будут автоматически созданы с категориями из каждая из подпапок.

Answer 2

Вы должны иметь возможность создавать и использовать свой собственный файл label.txt. Файл должен включать имена меток в том порядке, в котором вы указали их при обучении, по одному имени в строке.