"convert output of retrain.py to tensorflow.js" Code Answer

1

as hinted by @ping yu in retrain image detection with mobilenet, you can use

python retrain.py --tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/2 
    --image_dir /tmp/flower_photos --saved_model_dir /tmp/saved_retrained_model
tensorflowjs_converter --input_format=tf_saved_model 
    --output_format=tfjs_graph_model 
    --saved_model_tags=serve 
    /tmp/saved_retrained_model/ /tmp/converted_model/

this saves the model using the saved model format.

By Briguy37 on May 10 2022

Answers related to “convert output of retrain.py to tensorflow.js”

Only authorized users can answer the Search term. Please sign in first, or register a free account.