hub

hub copied to clipboard

What happened?

Using below URL for pretrained MobileNet model, it throws me an error : OSError: SavedModel file does not exist at: /var/folders/9y/yyk41v6n0fn2plkwb8wc5tdw0000gn/T/tfhub_modules/145bb06ec3b59b08fb564ab752bd5aa222bfb50a/{saved_model.pbtxt|saved_model.pb}

URL = "https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4

I've now downgraded to below link and it works ok : URL = "https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/3

Relevant code

#import tensorflow 
import tensorflow as tf
import tensorflow_datasets as tfds
import tensorflow_hub as hub

#Load the MobileNet pre-trained network from TensorFlow Hub.
URL = "https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4"  
feature_extractor = hub.KerasLayer(URL, input_shape=(image_size, image_size, 3))
feature_extractor.trainable = False

Relevant log output

URL: https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4

---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
/var/folders/9y/yyk41v6n0fn2plkwb8wc5tdw0000gn/T/ipykernel_49683/687692077.py in <module>
      2 URL = "https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4"
      3 
----> 4 feature_extractor = hub.KerasLayer(URL, input_shape=(image_size, image_size, 3))
      5 
      6 #setting the pre-trained weights and biases non trainable, so we can freeze them.

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow_hub/keras_layer.py in __init__(self, handle, trainable, arguments, _sentinel, tags, signature, signature_outputs_as_dict, output_key, output_shape, load_options, **kwargs)
    151 
    152     self._load_options = load_options
--> 153     self._func = load_module(handle, tags, self._load_options)
    154     self._has_training_argument = func_has_training_argument(self._func)
    155     self._is_hub_module_v1 = getattr(self._func, "_is_hub_module_v1", False)

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow_hub/keras_layer.py in load_module(handle, tags, load_options)
    447       except ImportError:  # Expected before TF2.4.
    448         set_load_options = load_options
--> 449     return module_v2.load(handle, tags=tags, options=set_load_options)
    450 
    451 

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow_hub/module_v2.py in load(handle, tags, options)
    104         module_path, tags=tags, options=options)
    105   else:
--> 106     obj = tf.compat.v1.saved_model.load_v2(module_path, tags=tags)
    107   obj._is_hub_module_v1 = is_hub_module_v1  # pylint: disable=protected-access
    108   return obj

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow/python/saved_model/load.py in load(export_dir, tags, options)
    867     ValueError: If `tags` don't match a MetaGraph in the SavedModel.
    868   """
--> 869   return load_internal(export_dir, tags, options)["root"]
    870 
    871 

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow/python/saved_model/load.py in load_internal(export_dir, tags, options, loader_cls, filters)
    879     tags = nest.flatten(tags)
    880   saved_model_proto, debug_info = (
--> 881       loader_impl.parse_saved_model_with_debug_info(export_dir))
    882 
    883   if (len(saved_model_proto.meta_graphs) == 1 and

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model_with_debug_info(export_dir)
     54     parsed. Missing graph debug info file is fine.
     55   """
---> 56   saved_model = _parse_saved_model(export_dir)
     57 
     58   debug_info_path = os.path.join(

~/opt/anaconda3/envs/tensorflow_env/lib/python3.7/site-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model(export_dir)
    114         "SavedModel file does not exist at: %s%s{%s|%s}" %
    115         (export_dir, os.path.sep, constants.SAVED_MODEL_FILENAME_PBTXT,
--> 116          constants.SAVED_MODEL_FILENAME_PB))
    117 
    118 

OSError: SavedModel file does not exist at: /var/folders/9y/yyk41v6n0fn2plkwb8wc5tdw0000gn/T/tfhub_modules/145bb06ec3b59b08fb564ab752bd5aa222bfb50a/{saved_model.pbtxt|saved_model.pb}

tensorflow_hub Version

0.12.0 (latest stable release)

TensorFlow Version

other (please specify)

Other libraries

tensorflow==2.5.0

Python Version

3.x

OS

macOS