Загрузка mobilenet v2 работает, но предварительно обученная mobilenet v2 не работает

Я переучиваю модель mobilenet v2, используя свои собственные изображения, и могу пометить новые изображения выводом в python ( https://www.tensorflow.org/hub/tutorials/image_retraining). Загрузка файла работает, но во время предсказания это не удается (concole.log из Firefox и Chromium):

The dict provided in model.execute(dict) has keys: [images] not part of   model graph.

Я переучиваю модель, используя предоставленный retrain.py

python retrain.py --image_dir flower_photos/ --tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/classification/2 --random_brightness 10 --how_many_training_steps 100

внутри flower_photos есть папки с названиями изображений и внутри соответствующих изображений.

flower_photos

--- Хуфлаттих

------- 1.jpg

------- 2.jpg

....

--- Buschwindröschen

------- 1.jpg

------- 2.jpg

Я могу конвертировать эту модель, используя

tensorflowjs_converter --input_format=tf_frozen_model --output_node_names='module_apply_default/MobilenetV2/Logits/output' /tmp/output_graph.pb   Mobilenetv2/web_model

но это не работает внутри предоставленного примера с https://github.com/tensorflow/tfjs-examples/tree/master/mobilenet

Если я конвертирую оригинальный mobilenet v2, используя

tensorflowjs_converter --input_format=tf_hub 'https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/classification/2' mobilenetv2/web_model

я могу загрузить внутри предоставленного примера.

В конце концов, программа должна обнаружить различные цветы раннего цветения, показанные веб-камерой, и классифицировать их. Это должно быть PWA для студентов и мотивировать их, чтобы испытать природу.

1 ответ

В настоящее время Tensorflow.js имеет два типа моделей:

  1. Модель слоев, которая позволяет обучать, вы можете загрузить их с помощью tf.loadModel (...)
  2. Модели, которые конвертированы из сгенерированных моделей TensorFlow, что не позволяет проводить обучение. Это то, что у вас есть, вы должны использовать tf.loadFrozenModel (...)

Вот пример для загрузки замороженной модели и выполнения прогноза на изображении. https://github.com/tensorflow/tfjs-converter/tree/master/demo/mobilenet

Другие вопросы по тегам