Instalando o tensorflow_hub
A biblioteca tensorflow_hub
pode ser instalada junto com o TensorFlow 1 e o TensorFlow 2. Recomendamos que novos usuários comecem com o TensorFlow 2 imediatamente e os usuários atuais atualizem para ele.
Usar com o TensorFlow 2
Use pip para instalar o TensorFlow 2 como de costume. (Veja lá para obter instruções extras sobre suporte a GPU.) Em seguida, instale uma versão atual do tensorflow-hub
próximo a ele (deve ser 0.5.0 ou mais recente).
$ pip install "tensorflow>=2.0.0"
$ pip install --upgrade tensorflow-hub
A API estilo TF1 do TensorFlow Hub funciona com o modo de compatibilidade v1 do TensorFlow 2.
Uso legado com TensorFlow 1
O TensorFlow 1.15 é a única versão do TensorFlow 1.x ainda compatível com a biblioteca tensorflow_hub
(a partir da versão 0.11.0). O TensorFlow 1.15 tem como padrão o comportamento compatível com TF1, mas contém muitos recursos do TF2 sob o capô para permitir algum uso das APIs no estilo TF2 do TensorFlow Hub.
$ pip install "tensorflow>=1.15,<2.0"
$ pip install --upgrade tensorflow-hub
Uso de versões de pré-lançamento
Os pacotes pip tf-nightly
e tf-hub-nightly
são compilados automaticamente a partir do código-fonte no github, sem teste de lançamento. Isso permite que os desenvolvedores experimentem o código mais recente sem compilar a partir do código-fonte .
$ pip install tf-nightly
$ pip install --upgrade tf-hub-nightly
Próximos passos
- Visão geral da biblioteca
- Tutoriais:
- Classificação de texto
- Classificação de imagem
- Exemplos adicionais no GitHub
- Encontre modelos em tfhub.dev