隨着深度學習算法的興起和普及，人工智能領域取得了使人矚目的進步，特別是在計算機視覺領域。21世紀的第二個十年迅速採用卷積神經網絡，發明了最早進的算法，大量訓練數據的可用性以及高性能和高性價比計算的發明。計算機視覺中的一個關鍵概念是圖像分類; 這是軟件系統正確標記圖像中主導對象的能力。python

ImageAI是一個Python庫，旨在幫助開發人員構建具備自包含計算機視覺功能的應用程序和系統。git

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

（若是您已經安裝了Python 3.5.1或更高版本，請跳過本節）github

https://www.python.org/downloads/算法

2. 安裝ImageAI依賴項

- Tensorflow網絡

pip3 install --upgrade tensorflow函數

- Numpy性能

pip3 install numpy學習

- SciPy優化

pip3 install scipy網站

- OpenCV

pip3 install opencv-python

- Matplotlib

pip3 install matplotlib

- h5py

pip3 install h5py

- Keras

pip3 install keras

3. 安裝ImageAI庫

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

4. 下載通過ImageNet-1000數據集訓練的ResNet Model文件，並將文件複製到您的python項目文件夾。

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

5. 建立一個名稱爲python的文件（例如「FirstPrediction.py」），並將下面的代碼寫入其中。

from imageai.Prediction import ImagePrediction

import os

execution_path = os.getcwd()

prediction = ImagePrediction()

prediction.setModelTypeAsResNet()

prediction.setModelPath( execution_path + " esnet50_weights_tf_dim_ordering_tf_kernels.h5")

prediction.loadModel()

predictions, percentage_probabilities = prediction.predictImage("C:UsersMyUserDownloadssample.jpg", result_count=5)

for index in range(len(predictions)):

print(predictions[index] + " : " + percentage_probabilities[index])

sample.jpg

代碼結果：

sports_car : 90.61029553413391

car_wheel : 5.9294357895851135

racer : 0.9972884319722652

convertible : 0.8457873947918415

grille : 0.581052340567112

代碼說明

如今讓咱們分解代碼，以便了解它是如何工做的。上面的代碼工做以下：

from imageai.Prediction import ImagePrediction

import os

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

execution_path = os.getcwd（）

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

prediction = ImagePrediction（）

prediction.setModelTypeAsResNet（）

prediction.setModelPath（execution_path +「 resnet50_weights_tf_dim_ordering_tf_kernels.h5」）

在上面的代碼中，咱們在第一行建立了一個ImagePrediction（）類的實例，而後經過在第二行中調用.setModelTypeAsResNet（），將預測對象的模型類型設置爲ResNet ，而後設置模型路徑將預測對象複製到模型文件（resnet50_weights_tf_dim_ordering_tf_kernels.h5）的路徑中，並將其複製到第三行的項目文件夾文件夾中。

predictions, percentage_probabilities = prediction.predictImage("C:UsersMyUserDownloadssample.jpg", result_count=5)

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

for index in range(len(predictions)):

print(predictions[index] + " : " + percentage_probabilities[index])

上述行獲取中的每一個對象的預測陣列，而且還得到從相應百分比機率percentage_probabilities，最後打印兩者的結果到控制檯。

該 .predictImage（）函數將在路徑中的圖像，也能夠說明咱們預計函數返回預測的數量（可選，默認爲5）。ImageNet-1000數據集中有1000個項目，ResNet模型在該數據集上進行了訓練，這意味着 .predictImage函數將返回1000個可能的預測值，並按其機率排列。

藉助ImageAI，您能夠輕鬆方便地將圖像預測代碼集成到您在python中構建的任何應用程序，網站或系統中。ImageAI庫支持其餘算法和模型類型，其中一些針對速度進行了優化，另外一些針對精度進行了優化。藉助ImageAI，咱們但願支持計算機視覺的更多專業方面，包括但不限於特殊環境和特殊領域的圖像識別以及自定義圖像預測