TensorFlow Hub 安裝及使用

2021-08-20 07:19:09 字數 803 閱讀 2699

tensorflow-hub 是 google 提供的機器學習模組打包函式庫,幫開發者把tensorflow的訓練模型發布成模組,方便再次使用或是與社交共享。

打包用來發布機器學習中可重複使用的部分,包括tensorflow_graph、權重以及外部檔案等所需資源。這些模組包含了大量資料集預訓練後取得的變數,可以供類似任務使用。這樣的成果可以使後續開發者,用更少的資料集訓練精準模型、改進泛化或是大幅提公升模型訓練速度。

和其他包沒什麼區別,可以直接 pip 安裝,但注意要求已安裝 tensorflow 包且版本在1.7以上。

pip install tensorflow-hub
可惜在 anaconda 下安裝沒有成功- 貌似還沒有加入 channels。

import tensorflow as tf

import tensorflow_hub as hub

deftest

(): embed = hub.module("")

embeddings = embed(["the quick brown fox jumps over the lazy dog."])

sess = tf.session()

sess.run(tf.initialize_all_variables())

sess.run(tf.initialize_all_tables())

print(sess.run(embeddings))

if __name__ == "__main__":

test()

使用TensorFlow Hub進行神經風格遷移

作者 richmond alake 編譯 flin towardsdatascience 我不會繪畫,但是機器學習可以 神經風格遷移 nst 是一種技術,涉及利用深度卷積神經網路和演算法從一幅影象中提取內容資訊並從另一幅參考影象中提取風格資訊。在提取風格和內容之後,將生成乙個組合影象,其中生成的影象...

安裝Docker(yum安裝)

確定你是centos7及以上版本 root 192 desktop cat etc redhat release centos linux release 7.2.1511 core 3.yum安裝gcc相關 yum y install gcc yum y install gcc c 4.解除安裝舊...

安裝 python python安裝

pyenv 多版本管理工具 1.linux安裝pyenv方式 git 安裝 1 安裝git yum install git y 2 安裝python編譯依賴 yum y install gcc make patch gdbm devel openssl devel sqlite devel read...