如何使用TensorFlow驗證儲存的模型可以重新載入並給出相同的結果?


可以使用TensorFlow的`load_model`方法載入儲存的模型,並使用`predict`方法進行預測來驗證儲存的模型是否可以重新載入。重新載入的模型可以用來預測資料。

閱讀更多: 什麼是TensorFlow以及Keras如何與TensorFlow一起建立神經網路?

包含至少一層卷積層的神經網路被稱為卷積神經網路。我們可以使用卷積神經網路來構建學習模型。

影像分類遷移學習背後的直覺是,如果一個模型在一個大型且通用的資料集上進行訓練,那麼這個模型可以有效地作為一個通用的視覺世界模型。它已經學習了特徵圖,這意味著使用者不必從頭開始在一個大型資料集上訓練一個大型模型。

TensorFlow Hub是一個包含預訓練TensorFlow模型的倉庫。TensorFlow可以用來微調學習模型。

我們將瞭解如何使用來自TensorFlow Hub的模型與tf.keras,使用來自TensorFlow Hub的影像分類模型。完成此操作後,可以執行遷移學習以微調自定義影像類別的模型。這是透過使用預訓練的分類器模型來獲取影像並預測它是哪個來完成的。這可以在不需要任何訓練的情況下完成。

我們使用Google Colaboratory執行以下程式碼。Google Colab或Colaboratory幫助在瀏覽器上執行Python程式碼,無需任何配置即可免費訪問GPU(圖形處理單元)。Colaboratory構建在Jupyter Notebook之上。

示例

print("Check to see if it gives the same results")
reloaded = tf.keras.models.load_model(export_path)
result_batch = model.predict(image_batch)
reloaded_result_batch = reloaded.predict(image_batch)
abs(reloaded_result_batch - result_batch).max()

程式碼來源 −https://www.tensorflow.org/tutorials/images/transfer_learning_with_hub

輸出

Check to see if it gives the same results
0.0

解釋

  • 我們首先必須確認儲存的模型可以重新載入。

  • 完成此操作後,我們必須確認儲存的模型獲得了相同的結果。

更新於:2021年2月25日

瀏覽量:109

啟動你的職業生涯

透過完成課程獲得認證

開始學習
廣告