如何使用Tensorflow模型實現(xiàn)預測？針對這個問題，這篇文章詳細介紹了相對應(yīng)的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。

為武安等地區(qū)用戶提供了全套網(wǎng)頁設(shè)計制作服務(wù)，及武安網(wǎng)站建設(shè)行業(yè)解決方案。主營業(yè)務(wù)為網(wǎng)站制作、做網(wǎng)站、武安網(wǎng)站設(shè)計，以傳統(tǒng)方式定制建設(shè)網(wǎng)站，并提供域名空間備案等一條龍服務(wù)，秉承以專業(yè)、用心的態(tài)度為用戶提供真誠的服務(wù)。我們深信只要達到每一位用戶的要求，就會得到認可，從而選擇與我們長期合作。這樣，我們也可以走得更遠！

模型文件：

預測圖片：

這里直接貼代碼，都有注釋，應(yīng)該很好理解

import tensorflow as tf
import inference
 
image_size = 128 # 輸入層圖片大小
 
# 模型保存的路徑和文件名
MODEL_SAVE_PATH = "model/"
MODEL_NAME = "model.ckpt"
 
# 加載需要預測的圖片
image_data = tf.gfile.FastGFile("./data/test/d.png", 'rb').read()
 
# 將圖片格式轉(zhuǎn)換成我們所需要的矩陣格式，第二個參數(shù)為1，代表1維
decode_image = tf.image.decode_png(image_data, 1)
 
# 再把數(shù)據(jù)格式轉(zhuǎn)換成能運算的float32
decode_image = tf.image.convert_image_dtype(decode_image, tf.float32)
 
# 轉(zhuǎn)換成指定的輸入格式形狀
image = tf.reshape(decode_image, [-1, image_size, image_size, 1])
 
# 定義預測結(jié)果為logit值大的分類，這里是前向傳播算法，也就是卷積層、池化層、全連接層那部分
test_logit = inference.inference(image, train=False, regularizer=None)
 
# 利用softmax來獲取概率
probabilities = tf.nn.softmax(test_logit)
 
# 獲取大概率的標簽位置
correct_prediction = tf.argmax(test_logit, 1)
 
# 定義Savar類
saver = tf.train.Saver()
 
with tf.Session() as sess:
  sess.run((tf.global_variables_initializer(), tf.local_variables_initializer()))
 
  # 加載檢查點狀態(tài)，這里會獲取最新訓練好的模型
  ckpt = tf.train.get_checkpoint_state(MODEL_SAVE_PATH)
  if ckpt and ckpt.model_checkpoint_path:
    # 加載模型和訓練好的參數(shù)
    saver.restore(sess, ckpt.model_checkpoint_path)
    print("加載模型成功：" + ckpt.model_checkpoint_path)
 
    # 通過文件名得到模型保存時迭代的輪數(shù).格式：model.ckpt-6000.data-00000-of-00001
    global_step = ckpt.model_checkpoint_path.split('/')[-1].split('-')[-1]
 
    # 獲取預測結(jié)果
    probabilities, label = sess.run([probabilities, correct_prediction])
 
    # 獲取此標簽的概率
    probability = probabilities[0][label]
 
    print("After %s training step(s),validation label = %d, has %g probability" % (global_step, label, probability))
  else:
    print("模型加載失?。?quot; + ckpt.model_checkpoint_path)

運行輸出結(jié)果：

（標簽為3，概率為0.984478）

標簽字典：

3對應(yīng)小寫d，識別正確。

其他的圖片的預測結(jié)果：

預測圖片1：

標簽字典：

關(guān)于如何使用Tensorflow模型實現(xiàn)預測問題的解答就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道了解更多相關(guān)知識。

新聞標題：如何使用Tensorflow模型實現(xiàn)預測-創(chuàng)新互聯(lián)
本文網(wǎng)址：http://www.rwnh.cn/article42/ppoec.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供外貿(mào)建站、軟件開發(fā)、Google、網(wǎng)站內(nèi)鏈、關(guān)鍵詞優(yōu)化、虛擬主機

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)