隨著深度學習算法的興起和普及，人工智能領(lǐng)域取得了令人矚目的進步，特別是在計算機視覺領(lǐng)域。21世紀的第二個十年迅速采用卷積神經(jīng)網(wǎng)絡，發(fā)明了最先進的算法，大量訓練數(shù)據(jù)的可用性以及高性能和高性價比計算的發(fā)明。計算機視覺中的一個關(guān)鍵概念是圖像分類; 這是軟件系統(tǒng)正確標記圖像中主導對象的能力。

　　ImageAI是一個Python庫，旨在幫助開發(fā)人員構(gòu)建具有自包含計算機視覺功能的應用程序和系統(tǒng)。

　　1. 安裝Python 3.5.1或更高版本和pip

　?。ㄈ绻呀?jīng)安裝了Python 3.5.1或更高版本，請?zhí)^本節(jié)）

　　2. 安裝ImageAI依賴項

　　3. 安裝ImageAI庫

　　pip3 install https://github.com/OlafenwaMoses/ImageAI/raw/master/dist/imageai-1.0.2-py3-none-any.whl

　　4. 下載經(jīng)過ImageNet-1000數(shù)據(jù)集訓練的ResNet Model文件，并將文件復制到您的python項目文件夾。

　　https://github.com/fchollet/deep-learning-models/releases/download/v0.2/resnet50_weights_tf_dim_ordering_tf_kernels.h5

　　5. 創(chuàng)建一個名稱為python的文件（例如“FirstPrediction.py”），并將下面的代碼寫入其中。

　　sample.jpg

　　代碼結(jié)果：

　　代碼說明

　　現(xiàn)在讓我們分解代碼，以便了解它是如何工作的。上面的代碼工作如下：

　　from imageai.Prediction import ImagePrediction import os

　　上面的代碼導入了ImageAI ImagePrediction類和python os類。

　　execution_path = os.getcwd（）

　　上面的代碼創(chuàng)建一個變量，它保存對包含python文件（在本例中為FirstPrediction.py）和ResNet模型文件的路徑的引用。

　　在上面的代碼中，我們在第一行創(chuàng)建了一個ImagePrediction（）類的實例，然后通過在第二行中調(diào)用.setModelTypeAsResNet（），將預測對象的模型類型設置為ResNet ，然后設置模型路徑將預測對象復制到模型文件（resnet50_weights_tf_dim_ordering_tf_kernels.h5）的路徑中，并將其復制到第三行的項目文件夾文件夾中。

　　predictions， percentage_probabilities = prediction.predictImage（“C:UsersMyUserDownloadssample.jpg”， result_count=5）

　　在上面的行中，我們定義了2個變量，它等于被調(diào)用來預測圖像的函數(shù)，這個函數(shù)是 .predictImage（）函數(shù)，我們在其中解析了圖像的路徑，并且還指出了我們想要的預測結(jié)果的數(shù)量有（從1到1000的值）解析result_count = 5 。所述 .predictImage（）函數(shù)將返回與所述第一（2級陣列的對象的預測）是預測和所述第二（陣列percentage_probabilities）是相應的百分比概率為每個預測的陣列。

　　for index in range（len（predictions））： print（predictions［index］ + “ ： ” + percentage_probabilities［index］）

　　上述行獲取中的每個對象的預測陣列，并且還獲得從相應百分比概率percentage_probabilities，最后打印二者的結(jié)果到控制臺。

　　該 .predictImage（）函數(shù)將在路徑中的圖像，也可以說明我們預計函數(shù)返回預測的數(shù)量（可選，默認為5）。ImageNet-1000數(shù)據(jù)集中有1000個項目，ResNet模型在該數(shù)據(jù)集上進行了訓練，這意味著 .predictImage函數(shù)將返回1000個可能的預測值，并按其概率排列。

　　借助ImageAI，您可以輕松方便地將圖像預測代碼集成到您在python中構(gòu)建的任何應用程序，網(wǎng)站或系統(tǒng)中。ImageAI庫支持其他算法和模型類型，其中一些針對速度進行了優(yōu)化，另一些針對精度進行了優(yōu)化。借助ImageAI，我們希望支持計算機視覺的更多專業(yè)方面，包括但不限于特殊環(huán)境和特殊領(lǐng)域的圖像識別以及自定義圖像預測。