AI從入門到放棄：CNN的導火索，用MLP做影像分類識別？

歡迎大家前往騰訊雲+社群，獲取更多騰訊海量技術實踐乾貨哦~

作者：鄭善友 騰訊MIG後臺開發工程師

導語：在沒有CNN以及更先進的神經網路的時代，樸素的想法是用多層感知機（MLP）做圖片分類的識別；但殘酷的現實是，MLP做這事的效果並不理想。本文通過使用MLP做圖片分類識別的嘗試作為思路指引，實為下一篇CNN相關筆記的引子文章。 本文的文件和程式碼，傳送門： github專案地址

一. 用MLP做影像分類識別？

1. 在沒有CNN以及更先進的神經網路的時代，樸素的想法是用多層感知機（MLP）做圖片分類的識別，沒毛病
2. 作為上篇筆記學習的延續，以及下一篇CNN的藥引，使用MLP來做圖片分類識別，實在是個不錯的過度例子。通過這個例子，從思路上引出一系列的問題，我不賣關子，自問自答吧，即：
   • MLP能做圖片分類識別嗎？—> 答案是是可以的，上一篇我們是擬合非線性分類函式，這裡是擬合影像特徵，數學本質沒區別。
   • MLP做這個事情效果如何？—> 個人認知內，只能說一般一般。
   • MLP在這一領域效果一般，是有什麼缺陷嗎？ —> 缺陷是有的，下文會詳細說。
   • 有更好的解決方案嗎？ —> 那也是必須有的，地球人火星人喵星人都知道有CNN等等更先進的東東；但是在沒有這些東西存在的時代，你發明出來了，那才真是666。

二. 先上車

1. 資料來源

• 資料來源當然是圖片，但是是經過資料化處理的圖片，使用的是h5檔案。h5檔案簡單說就是把資料按索引固化起來，挺簡單的不多說，度度一下 —> h5py入門講解
• 我們有3個h5檔案，存著不重複的圖片資料，分別是：
  • train_catvnoncat.h5 (用來訓練模型用的，一共有209張，其中有貓也有不是貓的圖片，尺寸64*64畫素)
  • test_catvnoncat.h5 （用來測試模型準確度的，一共有50張圖片，，其中有貓也有不是貓的圖片，尺寸64*64畫素）
  • my_cat_misu.h5 （用來玩的，我家貓主子的1張照騙，尺寸64*64畫素）

2. 資料結構

• 拿train_catvnoncat.h5舉例，這個檔案有2個索引：
  • train_set_x：這是一個陣列，因為是209張圖片，所以陣列長度是209。陣列中的元素是一個 64*64*3 的矩陣。64*64是圖片畫素尺寸，3是什麼鬼？別忘了這是彩色圖片，3就是代表RGB這3個顏色通道的值。
  • train_set_y：圖片標籤陣列，長度也是209，是209張圖片的標籤（label），對應陣列下標的值是1時，代表這張圖片是喵星人，0則代表不是。
  • 同理，test_catvnoncat.h5 中有 test_set_x 和 test_set_y；my_cat_misu.h5 中有 mycat_set_x 和 mycat_set_y

3. 告訴你怎麼製作圖片的h5檔案，以後做cnn等模型訓練時，非常有用

• 以我主子為例子：
  • 原圖：

• 自己處理成64*64的圖片，當然你也可以寫程式碼做圖片處理，我懶，交給你實現了：

• python程式碼，用到h5py庫：

def save_imgs_to_h5file(h5_fname, x_label, y_label, img_paths_list, img_label_list):
   # 構造n張圖片的隨機矩陣
   data_imgs = np.random.rand(len(img_paths_list), 64, 64, 3).astype('int')   
   label_imgs = np.random.rand(len(img_paths_list), 1).astype('int')
   
   # plt.imread可以把圖片以多維陣列形式讀出來，然後我們存成 n*n*3 的矩陣   
   for i in range(len(img_paths_list)):       
       data_imgs[i] = np.array(plt.imread(img_paths_list[i]))   
       label_imgs[i] = np.array(img_label_list[i])    
       
   # 建立h5檔案，按照指定的索引label存到檔案中，完事了   
   f = h5py.File(h5_fname, 'w')   
   f.create_dataset(x_label, data=data_imgs)   
   f.create_dataset(y_label, data=label_imgs)   
   f.close()    return data_imgs, label_imgs    
   
   #用法   
   # 圖片label為1代表這是一張喵星人的圖片，0代表不是   
   save_imgs_to_h5file('datasets/my_cat_misu.h5', 'mycat_set_x', 'mycat_set_y', ['misu.jpg'],[1])
複製程式碼

4. 看看我的資料來源的樣子

• 用來訓練的圖片集合，209張：

• 用來校驗模型準確度的圖片集合， 50張

• 用來玩的，主子照騙，1張：

三. 開車了

1. 如何設計模型：

• 輸入層： 我們的圖片是64*64的畫素尺寸，那麼算上RGB三個通道的資料，我們把三維矩陣拉成麵條 64*64*3 = 12288。 也就是我們輸入層的資料長度是12288。
• 隱藏層： 使用多層隱藏層，可以自行多嘗試一下不同的結構。這裡我使用3個隱藏層，隱藏層神經元個數分別是20，7，5
• 輸出層： 我們的目標就是判斷某張圖片是否是貓而已，所以輸出層1個神經元，輸出概率大於0.5認為是貓，小於等於0.5認為不是。

【插播】：有人會想，第一層隱藏層的神經元和輸入層數量一致是不是會好點？理論上會好點，但是這涉及到MLP的一個缺陷，因為全連線情況下，這樣做，第一層的權重w引數就有1228的平方個，約為1.5個億。如果圖片更大呢？引數會成指數級膨脹，後果盡情想象。

2. 如何訓練模型

• 還用說，把209張圖片的資料扔到神經網路，完成一次迭代，然後訓練1萬次，可自行嘗試迭不同代次數觀察效果。

3. 如何衡量模型的準確度

• 大神吳恩達（Andrew Ng）提到的方法之一，就是劃分不同集合，一部分用來訓練，一部分用來驗證模型效果，這樣可以達到衡量你所訓練的模型的效果如何。所以我們訓練使用209張圖片，最終使用50張測試模型效果。
• 為了好玩，可以自己用不同圖片通過模型去做分類識別。

四. 老規矩，甩程式碼

還是說明一下程式碼流程吧：

• 程式碼使用到的 NeuralNetwork 是我上一篇筆記的程式碼，實現了BP神經網路，import進來直接用即可。
• 程式碼做的事情就是：
  1. 從h5檔案載入圖片資料
  2. 把原始圖片顯示出來，同時也儲存成圖片檔案
  3. 訓練神經網路模型
  4. 驗證模型準確度
  5. 把識別結果標註到原始圖片上，同時也儲存成圖片檔案

#coding:utf-8
import h5py
import matplotlib.font_manager as fm
import matplotlib.pyplot as plt
import numpy as np
from NeuralNetwork import *

font = fm.FontProperties(fname='/System/Library/Fonts/STHeiti Light.ttc')

def load_Cat_dataset():
    train_dataset = h5py.File('datasets/train_catvnoncat.h5', "r")
    train_set_x_orig = np.array(train_dataset["train_set_x"][:])
    train_set_y_orig = np.array(train_dataset["train_set_y"][:])

    test_dataset = h5py.File('datasets/test_catvnoncat.h5', "r")
    test_set_x_orig = np.array(test_dataset["test_set_x"][:])
    test_set_y_orig = np.array(test_dataset["test_set_y"][:])

    mycat_dataset = h5py.File('datasets/my_cat_misu.h5', "r")
    mycat_set_x_orig = np.array(mycat_dataset["mycat_set_x"][:])
    mycat_set_y_orig = np.array(mycat_dataset["mycat_set_y"][:])

    classes = np.array(test_dataset["list_classes"][:])

    train_set_y_orig = train_set_y_orig.reshape((1, train_set_y_orig.shape[0]))
    test_set_y_orig = test_set_y_orig.reshape((1, test_set_y_orig.shape[0]))
    mycat_set_y_orig = mycat_set_y_orig.reshape((1, mycat_set_y_orig.shape[0]))

    return train_set_x_orig, train_set_y_orig, test_set_x_orig, test_set_y_orig, mycat_set_x_orig, mycat_set_y_orig,classes


def predict_by_modle(x, y, nn):

    m = x.shape[1]
    p = np.zeros((1,m))

    output, caches = nn.forward_propagation(x)

    for i in range(0, output.shape[1]):
        if output[0,i] > 0.5:
            p[0,i] = 1
        else:
            p[0,i] = 0

    # 預測出來的結果和期望的結果比對，看看準確率多少：
    # 比如100張預測圖片裡有50張貓的圖片，只識別出40張，那麼識別率就是80%
    print(u"識別率: "  + str(np.sum((p == y)/float(m))))
    return np.array(p[0], dtype=np.int), (p==y)[0], np.sum((p == y)/float(m))*100


def save_imgs_to_h5file(h5_fname, x_label, y_label, img_paths_list, img_label_list):
    data_imgs = np.random.rand(len(img_paths_list), 64, 64, 3).astype('int')
    label_imgs = np.random.rand(len(img_paths_list), 1).astype('int')

    for i in range(len(img_paths_list)):
        data_imgs[i] = np.array(plt.imread(img_paths_list[i]))
        label_imgs[i] = np.array(img_label_list[i])

    f = h5py.File(h5_fname, 'w')
    f.create_dataset(x_label, data=data_imgs)
    f.create_dataset(y_label, data=label_imgs)
    f.close()

    return data_imgs, label_imgs

if __name__ == "__main__":
    # 圖片label為1代表這是一張喵星人的圖片，0代表不是
    #save_imgs_to_h5file('datasets/my_cat_misu.h5', 'mycat_set_x', 'mycat_set_y', ['misu.jpg'],[1])

    train_set_x_orig, train_set_y_orig, test_set_x_orig, test_set_y_orig, mycat_set_x_orig, mycat_set_y_orig, classes = load_Cat_dataset()

    train_x_flatten = train_set_x_orig.reshape(train_set_x_orig.shape[0], -1).T
    test_x_flatten = test_set_x_orig.reshape(test_set_x_orig.shape[0], -1).T
    mycat_x_flatten = mycat_set_x_orig.reshape(mycat_set_x_orig.shape[0], -1).T
    train_set_x = train_x_flatten / 255.
    test_set_x = test_x_flatten / 255.
    mycat_set_x = mycat_x_flatten / 255.

    print(u"訓練圖片數量: %d" % len(train_set_x_orig))
    print(u"測試圖片數量: %d" % len(test_set_x_orig))

    plt.figure(figsize=(10, 20))
    plt.subplots_adjust(wspace=0,hspace=0.15)
    for i in range(len(train_set_x_orig)):
        plt.subplot(21,10, i+1)
        plt.imshow(train_set_x_orig[i],interpolation='none',cmap='Reds_r',vmin=0.6,vmax=.9)
        plt.xticks([])
        plt.yticks([])
    plt.savefig("cat_pics_train.png")
    plt.show()

    plt.figure(figsize=(8, 8))
    plt.subplots_adjust(wspace=0, hspace=0.1)
    for i in range(len(test_set_x_orig)):
        ax = plt.subplot(8, 8, i + 1)
        im = ax.imshow(test_set_x_orig[i], interpolation='none', cmap='Reds_r', vmin=0.6, vmax=.9)
        plt.xticks([])
        plt.yticks([])

    plt.savefig("cat_pics_test.png")
    plt.show()

    plt.figure(figsize=(2, 2))
    plt.subplots_adjust(wspace=0, hspace=0)
    for i in range(len(mycat_set_x_orig)):
        ax = plt.subplot(1, 1, i + 1)
        im = ax.imshow(mycat_set_x_orig[i], interpolation='none', cmap='Reds_r', vmin=0.6, vmax=.9)
        plt.xticks([])
        plt.yticks([])

    plt.savefig("cat_pics_my.png")
    plt.show()

    # 用訓練圖片集訓練模型
    layers_dims = [12288, 20, 7, 5, 1]
    nn = NeuralNetwork(layers_dims, True)
    nn.set_xy(train_set_x, train_set_y_orig)
    nn.set_num_iterations(10000)
    nn.set_learning_rate(0.0075)
    nn.training_modle()

    # 結果展示說明：
    # 【識別正確】：
    #   1.原圖是貓，識別為貓  --> 原圖顯示
    #   2.原圖不是貓，識別為不是貓 --> 降低顯示亮度

    # 【識別錯誤】：
    #   1.原圖是貓，但是識別為不是貓 --> 標紅顯示
    #   2.原圖不是貓， 但是識別成貓 --> 標紅顯示

    # 訓練用的圖片走一遍模型，觀察其識別率
    plt.figure(figsize=(10, 20))
    plt.subplots_adjust(wspace=0, hspace=0.15)

    pred_train, true, accuracy = predict_by_modle(train_set_x, train_set_y_orig, nn)

    for i in range(len(train_set_x_orig)):
        ax = plt.subplot(21, 10, i + 1)

        x_data = train_set_x_orig[i]
        if pred_train[i] == 0 and train_set_y_orig[0][i] == 0:
            x_data = x_data/5

        if true[i] == False:
            x_data[:, :, 0] = x_data[:, :, 0] + (255 - x_data[:, :, 0])

        im = plt.imshow(x_data,interpolation='none',cmap='Reds_r',vmin=0.6,vmax=.9)

        plt.xticks([])
        plt.yticks([])

    plt.suptitle(u"Num Of Pictrues: %d\n Accuracy: %.2f%%" % (len(train_set_x_orig), accuracy), y=0.92, fontsize=20)
    plt.savefig("cat_pics_train_predict.png")
    plt.show()

    # 不屬於訓練圖片集合的測試圖片，走一遍模型，觀察其識別率
    plt.figure(figsize=(8, 8))
    plt.subplots_adjust(wspace=0, hspace=0.1)

    pred_test, true, accuracy = predict_by_modle(test_set_x, test_set_y_orig, nn)

    for i in range(len(test_set_x_orig)):
        ax = plt.subplot(8, 8, i + 1)

        x_data = test_set_x_orig[i]
        if pred_test[i] == 0 and test_set_y_orig[0][i] == 0:
            x_data = x_data/5

        if true[i] == False:
            x_data[:, :, 0] = x_data[:, :, 0] + (255 - x_data[:, :, 0])

        im = ax.imshow(x_data, interpolation='none', cmap='Reds_r', vmin=0.6, vmax=.9)

        plt.xticks([])
        plt.yticks([])

        plt.suptitle(u"Num Of Pictrues: %d\n Accuracy: %.2f%%" % (len(mycat_set_x_orig), accuracy), fontsize=20)
    plt.savefig("cat_pics_test_predict.png")
    plt.show()

    # 用我家主子的照騙，走一遍模型，觀察其識別率，因為只有一張圖片，所以識別率要麼 100% 要麼 0%
    plt.figure(figsize=(2, 2.6))
    plt.subplots_adjust(wspace=0, hspace=0.1)

    pred_mycat, true, accuracy = predict_by_modle(mycat_set_x, mycat_set_y_orig, nn)
    for i in range(len(mycat_set_x_orig)):
        ax = plt.subplot(1, 1, i+1)

        x_data = mycat_set_x_orig[i]
        if pred_mycat[i] == 0 and mycat_set_y_orig[0][i] == 0:
            x_data = x_data/5

        if true[i] == False:
            x_data[:, :, 0] = x_data[:, :, 0] + (255 - x_data[:, :, 0])

        im = ax.imshow(x_data, interpolation='none', cmap='Reds_r', vmin=0.6, vmax=.9)

        plt.xticks([])
        plt.yticks([])

        if pred_mycat[i] == 1:
            plt.suptitle(u"我：'我主子是喵星人嗎？'\nA I :'是滴'", fontproperties = font)
        else:
            plt.suptitle(u"我：'我主子是喵星人嗎？'\nA I :'唔系~唔系~'", fontproperties = font)
    plt.savefig("cat_pics_my_predict.png")
    plt.show()
複製程式碼

五.結論

1. 神經網路模型的輸出結果，標註到了圖片上並展示出來，規則是：

結果展示說明：

【識別正確】：

1. 原圖是貓，識別為貓 —> 原圖顯示
2. 原圖不是貓，識別為不是貓 —> 降低顯示亮度

【識別錯誤】：

1. 原圖是貓，但是識別為不是貓 —> 標紅顯示
2. 原圖不是貓， 但是識別成貓 —> 標紅顯示

圖片標題會顯示Accuracy（準確度），準確度的計算公式是： 識別正確圖片數/圖片總數。

2. 模型訓練完成後，把訓練用的209張圖片用訓練好的模型識別一遍，觀察結果：可以看到，迭代1w次的模型，識別訓練圖集，準確度是 100% 的：

3. 模型訓練完成後，使用測試圖集用訓練好的模型識別一遍，觀察結果：可以看到，迭代1w次的模型，識別訓練圖集，準確度只有 78%：

4. 看看模型能不能認出我主子是喵星人，看樣子，它是認出來了：

六.對結果進一步分析，引出一系列問題

1. 丟擲一個問題： 為什麼用測試圖集驗證模型，識別率只有78%？在我嘗試過改變神經網路結構設計，引數調參後，仍然無法提高識別率，為什麼呢？
2. 不算徹底的解答：
   • 也許是我水平有限，調參姿勢不對？姿勢帥並不是萬能的，我們應該從更深層次的原理進行分析。
   • 有人說，你訓練資料少了，好像有那麼些道理。其實是可以給模型輸入更多圖片的特徵是個不錯的辦法，比如旋轉一下，圖片內容放大縮小，挪挪位置等。但是Andrew Ng也說過，過分追求訓練資料收集是一條不歸路。在同等訓練資料集下，有更好的辦法嗎？由此引出下一個問題。
3. 刨根問底： 想要知道為什麼MLP識別度難以做到很高，撇開網路結構，調參，訓練資料先不談。我們應該從MLP身上找找茬。搞清楚我們目標，是提高對圖片進行分類識別，那麼在使用MLP實現這個目標時，它自身是否有缺陷，導致實現這個目標遇到了困難。那麼解決了這些困難，就找到了解決問題的方法。
4. MLP在做圖片分類識別的缺陷：
   • 神經元是全連線的方式構成的神經網路，全連線情況下，假設圖片是1k*1k畫素大小，那麼隱藏層個數和輸入層尺寸一致時，不考慮RGB顏色通道，單通道下，權重w引數個數會是：$(10^3∗10^3)^2=10^{12}$=1千億（沒數錯0的話） 如果圖片再大點，引數膨脹到不可想象，直接導致的負面效果是：
     • 引數過多，計算量龐大
     • 全連線情況下，過深的網路容易導致梯度消失，模型難以訓練
   • MLP全連線的情況下，無法做到圖片的形變識別。怎麼理解這個詞呢，拿手寫數字舉個例子，比如寫8，每個人書寫習慣不一樣，有的人寫的很正，但有的人寫歪了點，上半部分小，下半部分大，等等。這時候，MLP的缺點就顯現出來了，同一張圖片，旋轉，或者稍微平移形變一下，它無法識別。你可以通過增加更多特徵給模型，但這不是本質上的解決該問題的方法，而是對訓練的優化手段。

七. 總結要解決的問題，離下一個坑就不遠了

上面已經列舉了要解決的幾個問題，這裡總結一下：

• 我們要解決引數膨脹帶來的計算量龐大的問題
• 優化引數量之後，如何在同等訓練資料集不變的情況下，如何提取更多特徵
• 在輸入有一定的旋轉平移伸縮時，仍能正確識別

能解決以上問題的，眾所周知，就是CNN以及眾多更先進的神經網路模型了。本文作為一篇引子文章，也是CNN的導火索。 程式碼在你手中，把第一層隱藏層設計成和輸入層一樣大，即 layers_dims = [12288, 12288, 20, 7, 5, 1]。還只是64*64的小圖片而已，那龜速，我和我的小破筆記本都不能忍啊。這也是為什麼大神們發明CNN的原因之一吧！

---

問答

TensorFlow：如何進行影像識別？

相關閱讀

人臉識別技術發展及實用方案設計

影像分析那些事︱AI來了

影像識別基本演算法之SURF

---

此文已由作者授權騰訊雲+社群釋出，原文連結：https://cloud.tencent.com/developer/article/1150162?fromSource=waitui

歡迎大家前往騰訊雲+社群或關注雲加社群微信公眾號（QcloudCommunity），第一時間獲取更多海量技術實踐乾貨哦~