Загрузка 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 имеет два типа моделей:
- Модель слоев, которая позволяет обучать, вы можете загрузить их с помощью tf.loadModel (...)
- Модели, которые конвертированы из сгенерированных моделей TensorFlow, что не позволяет проводить обучение. Это то, что у вас есть, вы должны использовать tf.loadFrozenModel (...)
Вот пример для загрузки замороженной модели и выполнения прогноза на изображении. https://github.com/tensorflow/tfjs-converter/tree/master/demo/mobilenet