文章目錄

• • 1. tfhub
  • 2. 例子：神經(jīng)風(fēng)格轉(zhuǎn)換
  • 3. retrain 例子

學(xué)習(xí)于：簡單粗暴 TensorFlow 2

1. tfhub

網(wǎng)址：
https://hub.tensorflow.google.cn/
https://tfhub.dev/

• 可以搜索，下載模型
  

• 安裝包 pip install tensorflow-hub

import tensorflow_hub as hubhub_url = 'https://hub.tensorflow.google.cn/google/magenta/arbitrary-image-stylization-v1-256/2' hub_model = hub.load(hub_url) # 加載模型 outputs = hub_model(inputs) # 調(diào)用模型

2. 例子：神經(jīng)風(fēng)格轉(zhuǎn)換

Ng 課也講過這個例子

import tensorflow_hub as hub import matplotlib.pyplot as plt import numpy as np import tensorflow as tf# 歸一化，resize def load_image_local(img_path, img_size=(256, 256)):# png 4 通道轉(zhuǎn) jpg 3通道if 'png' in img_path:img = Image.open(img_path)img = img.convert('RGB')img.save("temp.jpg")img = plt.imread("temp.jpg").astype(np.float32)[np.newaxis, :, :, :]else:# 添加一個 batch_size 軸img = plt.imread(img_path).astype(np.float32)[np.newaxis, :, :, :]if img.max() > 1.0:img = img / 255.img = tf.image.resize(img, img_size, preserve_aspect_ratio=True)return img# 繪制圖片 def show_image(img, title, save=False, fig_dpi=300):plt.imshow(img, aspect='equal')plt.axis('off')plt.show()if save:plt.imsave(title + '.jpg', img.numpy())# 圖片路徑 content_image_path = "pic1.jpg" style_image_path = "pic2.jpg"# 處理圖片 content_image = load_image_local(content_image_path) style_image = load_image_local(style_image_path)# 展示圖片 show_image(content_image[0], "Content Image") show_image(style_image[0], "Style Image")# 加載模型 hub_url = 'https://hub.tensorflow.google.cn/google/magenta/arbitrary-image-stylization-v1-256/2' hub_model = hub.load(hub_url)# 調(diào)用模型 outputs = hub_model(tf.constant(content_image), tf.constant(style_image)) stylized_image = outputs[0] # 取出第一個樣本預(yù)測值 [ :, :, 3]# 展示預(yù)測圖片 show_image(stylized_image[0], "Stylized Image", True)

內(nèi)容圖片：

風(fēng)格圖片：

轉(zhuǎn)換后的圖片：

3. retrain 例子

https://hub.tensorflow.google.cn/google/imagenet/inception_v3/feature_vector/4

• hub.KerasLayer(url) 封裝一個layer到模型當(dāng)中，可以設(shè)置是否 finetune

num_classes = 10 model = tf.keras.Sequential([hub.KerasLayer("https://hub.tensorflow.google.cn/google/imagenet/inception_v3/feature_vector/4",trainable=False), # 可以設(shè)為True，微調(diào)tf.keras.layers.Dense(num_classes, activation='softmax') ]) model.build([None, 299, 299, 3]) # Batch input shape model.summary()

模型結(jié)構(gòu)

Model: "sequential" _________________________________________________________________ Layer (type) Output Shape Param # ================================================================= keras_layer (KerasLayer) (None, 2048) 21802784 _________________________________________________________________ dense (Dense) (None, 10) 20490 ================================================================= Total params: 21,823,274 Trainable params: 20,490 Non-trainable params: 21,802,784 _________________________________________________________________

總結(jié)

以上是生活随笔為你收集整理的TensorFlow 2.0 - Hub 模型复用的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。