Запуск приложения сегментации образцов tflite с другой моделью - PullRequest
0 голосов
/ 25 февраля 2020

Я пытаюсь запустить пример приложения из tenorflow для сегментации изображения с другой моделью. Я хотел бы запустить его с моделью shufflenetv2 с dp c. Поэтому я скопировал модель и изменил imageSize на 225 в ImageSegmentationModelExecutor.kt. Затем я получаю сообщение об ошибке

что-то пошло не так: у + высота должна быть <= bitmap.height () </p>

Выполнение некоторых небольших настроек в функции scaleBitmapAndKeepRatio для ImageUtils .kt решает проблему. (Просто изменил targetBmp.width на высоту дважды, один раз в матрице и второй раз в возвращении.) Это приводит к следующей ошибке

что-то пошло не так: невозможно преобразовать буфер TensorFlowLite с 202500 байтами и a Java Буфер с 4252500 байтами.

Соотношение этих двух чисел равно NUM_CLASSES. Не уверен, что это правильный способ запустить его или как продолжить отсюда. Есть идеи или предложения?

...