TensorFlow Hub
TensorFlow Hub is a repository of TensorFlow modules shared by publishers such as Google, NVIDIA, and Kaggle. TensorFlow modules are self-contained models built on TensorFlow that can be reused for different tasks. Put simply, it is an external collection of published TensorFlow modules for transfer learning and fine-tuning. With TensorFlow Hub, you can access different deep learning models or weights than the ones provided directly from TensorFlow's core API.
Note
You can find more information about TensorFlow Hub here: https://tfhub.dev/.
In order to use it, you first need to install it:
pip install tensorflow-hub
Once it's installed, you can load available classification models with the load()
method by specifying the link to a module:
import tensorflow_hub as hub MODULE_HANDLE = 'https://tfhub.dev/tensorflow/efficientnet'\ '/b0...