Класс-оболочка для модели TFLite и интерпретатора TFLite.
Примечание. Model
может одновременно содержать только одну модель TFLite и всегда содержит экземпляр интерпретатора TFLite для ее запуска.
Вложенные классы
сорт | Модель.Строитель | Этот класс устарел. Пожалуйста, используйте Model.createModel(Context, String, Options) . | |
перечисление | Модель.Устройство | Тип устройства времени выполнения, используемый для выполнения классификации. | |
сорт | Модель.Параметры | Варианты запуска модели. |
Публичные методы
пустота | закрывать () |
статическая модель | createModel (контекст контекста, String modelPath, параметры Model.Options ) Загружает модель из ресурсов и инициализирует интерпретатор TFLite с заданными параметрами. |
статическая модель | createModel (контекст контекста, String modelPath) Загружает модель из ресурсов и инициализирует интерпретатор TFLite. |
MappedByteBuffer | получить данные () Возвращает данные модели, отображенные в памяти. |
Тензор | getInputTensor (int inputIndex) Получает тензор, связанный с предоставленным входным индексом. |
Тензор | getOutputTensor (int выходной индекс) Получает тензор, связанный с предоставленным выходным индексом. |
интервал [] | getOutputTensorShape (int выходной индекс) Возвращает выходную форму. |
Нить | получитьПуть () Возвращает путь к файлу модели, хранящемуся в Assets. |
пустота |
Унаследованные методы
Публичные методы
public void закрыть ()
общедоступная статическая модель createModel (контекст контекста, строка modelPath, параметры Model.Options )
Загружает модель из ресурсов и инициализирует интерпретатор TFLite с заданными параметрами.
Параметры
контекст | Контекст приложения. |
---|---|
модельпуть | Путь к файлу модели. |
параметры | Варианты запуска модели. |
Броски
Исключение IO | если при открытии файла модели возникает какое-либо исключение. |
---|
Смотрите также
общедоступная статическая модель createModel (контекстный контекст, String modelPath)
Загружает модель из ресурсов и инициализирует интерпретатор TFLite.
Варианты по умолчанию: (1) устройство ЦП; (2) одна нить.
Параметры
контекст | Контекст приложения. |
---|---|
модельпуть | Путь к файлу модели. |
Броски
Исключение IO | если при открытии файла модели возникает какое-либо исключение. |
---|
общедоступный тензор getInputTensor (int inputIndex)
Получает тензор, связанный с предоставленным входным индексом.
Параметры
инпутиндекс |
---|
Броски
IllegalStateException | если интерпретатор закрыт. |
---|
общедоступный тензор getOutputTensor (int outputIndex)
Получает тензор, связанный с предоставленным выходным индексом.
Параметры
выходной индекс |
---|
Броски
IllegalStateException | если интерпретатор закрыт. |
---|
public int[] getOutputTensorShape (int outputIndex)
Возвращает выходную форму. Полезно, если форма вывода определяется только при создании графика.
Параметры
выходной индекс |
---|
Броски
IllegalStateException | если интерпретатор закрыт. |
---|
public void run (входные данные Object[] , Map < Integer , Object > выходные данные)
Выполняет вывод модели на нескольких входных данных и возвращает несколько выходных данных.
Параметры
входы | массив входных данных. Входные данные должны быть в том же порядке, что и входные данные модели. Каждый вход может быть массивом или многомерным массивом или ByteBuffer примитивных типов, включая int, float, long и byte. ByteBuffer — предпочтительный способ передачи больших входных данных, тогда как строковые типы требуют использования входного пути (многомерного) массива. Когда используется ByteBuffer , его содержимое должно оставаться неизменным до тех пор, пока не будет выполнен вывод модели. |
---|---|
результаты | карта, отображающая выходные индексы в многомерные массивы выходных данных или ByteBuffer примитивных типов, включая int, float, long и byte. Необходимо только сохранять записи для используемых выходов. |