本文邏輯：

1. 我從網上下載了十幾張貓和狗的圖片，用於檢驗我們訓練好的模型。
2. 處理我們下載的圖片
3. 載入模型
4. 將圖片輸入模型進行檢驗

程式碼如下：

#coding=utf-8  
import tensorflow as tf 
from PIL import Image  
import matplotlib.pyplot as plt
import input_data 
import numpy as np
import model
import os 
  
#從指定目錄中選取一張圖片 
def get_one_image(train): 
    files = os.listdir(train)
    n = len(files)
    ind = np.random.randint(0,n)
    img_dir = os.path.join(train,files[ind])  
    image = Image.open(img_dir)  
    plt.imshow(image)
    plt.show()
    image = image.resize([208, 208])  
    image = np.array(image)
    return image  
  
  
def evaluate_one_image():  
	#存放的是我從百度下載的貓狗圖片路徑
    train = '/Users/yangyibo/GitWork/pythonLean/AI/貓狗識別/testImg/' 
    image_array = get_one_image(train)  
      
    with tf.Graph().as_default():  
        BATCH_SIZE = 1  # 因為只讀取一副圖片 所以batch 設定為1
        N_CLASSES = 2  # 2個輸出神經元，［1，0］ 或者 ［0，1］貓和狗的概率
        # 轉化圖片格式
        image = tf.cast(image_array, tf.float32)  
        # 圖片標準化
        image = tf.image.per_image_standardization(image)
        # 圖片原來是三維的 [208, 208, 3] 重新定義圖片形狀 改為一個4D  四維的 tensor
        image = tf.reshape(image, [1, 208, 208, 3])  
        logit = model.inference(image, BATCH_SIZE, N_CLASSES)  
        # 因為 inference 的返回沒有用啟用函式，所以在這裡對結果用softmax 啟用
        logit = tf.nn.softmax(logit)  
        
        # 用最原始的輸入資料的方式向模型輸入資料 placeholder
        x = tf.placeholder(tf.float32, shape=[208, 208, 3])  
        
        # 我門存放模型的路徑
        logs_train_dir = '/Users/yangyibo/GitWork/pythonLean/AI/貓狗識別/saveNet/'   
        # 定義saver 
        saver = tf.train.Saver()  
          
        with tf.Session() as sess:  
              
            print("從指定的路徑中載入模型。。。。")
            # 將模型載入到sess 中 
            ckpt = tf.train.get_checkpoint_state(logs_train_dir)  
            if ckpt and ckpt.model_checkpoint_path:  
                global_step = ckpt.model_checkpoint_path.split('/')[-1].split('-')[-1]  
                saver.restore(sess, ckpt.model_checkpoint_path)  
                print('模型載入成功, 訓練的步數為 %s' % global_step)  
            else:  
                print('模型載入失敗，，，檔案沒有找到')  
            # 將圖片輸入到模型計算
            prediction = sess.run(logit, feed_dict={x: image_array})
            # 獲取輸出結果中最大概率的索引
            max_index = np.argmax(prediction)  
            if max_index==0:  
                print('貓的概率 %.6f' %prediction[:, 0])  
            else:  
                print('狗的概率 %.6f' %prediction[:, 1]) 
# 測試
evaluate_one_image()

---------------------

本文來自 雙斜槓少年 的CSDN 部落格 ，全文地址請點選：https://blog.csdn.net/u012373815/article/details/79222121?utm_source=copy 
 

/Users/yangyibo/GitWork/pythonLean/AI/貓狗識別/testImg/ 存放的是我從百度下載的貓狗圖片

執行結果： 因為從testimg 中選取圖片是隨機的，所以每次執行的結果不同

從指定的路徑中載入模型。。。。
模型載入成功, 訓練的步數為 11999
狗的概率 0.964047
[Finished in 6.8s]