Восстановить классификатор изображений в tenorflow js

Question

Любая идея преобразовать классификатор изображений для переобучения для использования с тензорным потоком js с https://www.tensorflow.org/hub/tutorials/image_retraining

mkdir ~/example_code
cd ~/example_code
curl -LO https://github.com/tensorflow/hub/raw/r0.1/examples/image_retraining/retrain.py

python retrain.py --image_dir ~/flower_photos

попробуйте конвертировать модель с помощью tenorflowjs_converter https://github.com/tensorflow/tfjs-converter

tensorflowjs_converter \
    --input_format=tf_frozen_model \
    --output_node_names='MobilenetV1/Predictions/Reshape_1' \
    --saved_model_tags=serve \
    /tmp/output_graph.pb \
    /tmp/web_model

получать эту ошибку

"graph." % repr(name))
KeyError: "The name 'MobilenetV1/Predictions/Reshape_1' refers to an Operation not in the graph."

также сбой при создании модели mobilenet v1 с помощью команды

python retrain.py \
    --image_dir ~/flower_photos \
    --tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v1_100_224/feature_vector/1

Спасибо

Answer 1

@Mustafa Я думаю, вы неправильно оцениваете --output_node_names , попробуйте пройтись по модели с помощью тензорной доски, и вы найдете значение, которое должно быть здесь указано, это должно быть что-то вроде final_result(это в моем случае).