как использовать предварительно обученную модель в мобильном андроид приложении Tensorflow? - PullRequest
0 голосов
/ 25 апреля 2019

Я пытаюсь использовать мобильное демо-приложение tenor flow на Android для обнаружения и классификации изображений, следуя этому учебнику . Я думаю, что приложение использует начальную модель, и я хочу использовать другую предварительно обученную модель "ssd_mobilenetv2_oidv4" из этой ссылки . У меня нет большого опыта работы с машинным обучением, но я хочу заменить модель, которая использовалась в приложении, моделью mobilenetv2, так как она обучалась в большем количестве классов.

Я пытался скопировать файлы в предварительно обученной модели в папку ресурсов в приложении и отрегулировать путь в приложении, но проблема в том, что две модели имеют разные форматы (я думаю), поэтому приложение вылетает раньше это даже начинается.

файлы в папке активов (начальная модель): imagenet_comp_graph_label_strings.txt multibox_location_priors.txt tensorflow_inception_graph.pb tiny-yolo.pb

MObilenetV2 файлы в модели, которую я хочу использовать: checkpoint frozen_inference_graph.pb model.ckpt.data-00000-of-00001 model.ckpt.index model.ckpt.meta pipeline.config and folder "saved_model" containing: a- saved_model.pb b- an empty folder named "variables"

как же превратить модель, которую я хочу использовать, в тот же формат, что и модель в папке ресурсов (начальная модель)

1 Ответ

0 голосов
/ 25 апреля 2019

Обратите внимание, что при восстановлении файлов контрольных точек вам не нужно предварительно их инициализировать.

tf.train.Saver также можно восстановить (tf.train.Saver.restore):

https://www.tensorflow.org/api_docs/python/tf/train/Saver#restore

...