10 行Python 代碼,實現 AI 目標檢測技術,真給力!

只需10行Python代碼,我們就能實現計算機視覺中目標檢測。

from imageai.Detection import ObjectDetection
import os

execution_path = os.getcwd()

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))

for eachObject in detections:
print(eachObject["name"] + " : " + eachObject["percentage_probability"] )

沒錯,用這寥寥10行代碼,就能實現目前AI產品中應用廣泛的目標檢測技術。

看完了代碼,下面容我們聊聊目標檢測背後的技術背景,並解讀這10行Python代碼的由來和實現原理。

目標檢測簡介

人工智能的一個重要領域就是計算機視覺,它是指計算機及軟件系統識別和理解圖像與視頻的科學。計算機視覺包含很多細分方向,比如圖像識別、目標檢測、圖像生成和圖像超分辨率等。其中目標檢測由於用途廣泛,在計算機視覺領域的意義最爲深遠。

目標檢測是指計算機和軟件系統能夠定位出圖像/畫面中的物體,並識別出它們。目標檢測技術已經廣泛應用於人臉檢測、車輛檢測、人流量統計、網絡圖像、安防系統和無人車等領域。和其它計算機視覺技術一樣,目標檢測未來會進一步成爲人工智能的重要組成部分,有着廣闊的發展前景。

不過,在軟件應用和系統中使用現代目標檢測方法以及根據這些方法創建應用,並非簡單直接。早期的目標檢測實現主要是應用一些經典算法,比如OpenCV中支持的算法。然而這些算法的表現並不穩定,在不同情況下差異巨大。

2012年深度學習技術的突破性進展,催生了一大批高度精準的目標檢測算法,比如R-CNN,Fast-RCNN,Faster-RCNN,RetinaNet和既快又準的SSD及YOLO。使用這些基於深度學習的方法和算法,需要理解大量的數學和深度學習框架。現在全世界有數以百萬計的開發者在藉助目標檢測技術創造新產品新項目,但由於理解和使用較爲複雜困難,仍有很多人不得要領。

爲了解決這個困擾開發者們的問題,計算機視覺專家Moses Olafenwa帶領團隊推出了Python庫ImageAI,能讓開發人員只需寥寥數行代碼就能很容易的將最先進的計算機視覺技術應用到自己的項目和產品中。

我們開頭所示的10行代碼實現,就是要用到ImageAI。

如何藉助ImageAI輕鬆實現目標檢測

使用ImageAI執行目標檢測,你只需以下4步:

1.在電腦上安裝Python

2.安裝ImageAI及其環境依賴

3.下載目標檢測模塊文件

4.運行示例代碼,就是我們展示的那10行

下面我們一步步詳細講解。

1)從Python官網下載和安裝Python 3

python.org/

2)通過pip安裝如下環境依賴

1.Tensorflow

pip install tensorflow

2.Numpy

pip install numpy

3.SciPy

pip install scipy

4.OpenCV

pip install opencv-python

5.Pillow

pip install pillow

6.Matplotlib

pip install matplotlib

7.H5py

pip install h5py

8.Keras

pip install keras

9.ImageAI

pip install

github.com

3)通過該鏈接下載RetinaNet 模型文件用於目標檢測。

到了這裏我們已經安裝好了所有依賴,就可以準備寫自己的首個目標檢測代碼了。 創建一個Python文件,爲其命名(比如FirstDetection.py),然後將如下代碼寫到文件中,再把RetinaNet模型文件以及你想檢測的圖像拷貝到包含該Python文件的文件夾裏。

FirstDetection.py

from imageai.Detection import ObjectDetection
import os

execution_path = os.getcwd()

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))

for eachObject in detections:
print(eachObject["name"] + " : " + eachObject["percentage_probability"] )

然後運行代碼,等待控制檯打印結果。等控制檯打印出結果後,就可以打開FirstDetection.py所在的文件夾,你就會發現有新的圖像保存在了裏面。比如下面兩張示例圖像,以及執行目標檢測後保存的兩張新圖像。

大家在學python的時候肯定會遇到很多難題,以及對於新技術的追求,這裏推薦一下我們的Python學習扣qun:784758214,這裏是python學習者聚集地!!同時,自己是一名高級python開發工程師,從基礎的python腳本到web開發、爬蟲、django、數據挖掘等,零基礎到項目實戰的資料都有整理。送給每一位python的小夥伴!每日分享一些學習的方法和需要注意的小細節

點擊:python技術分享

目標檢測之前:

目標檢測之後:

我們可以看到圖像上顯示了檢測出的物體名稱及概率。

解讀10行代碼

下面我們解釋一下這10行代碼的工作原理。

from imageai.Detection import ObjectDetection
import os

execution_path = os.getcwd()

在上面3行代碼中,我們在第一行導入了ImageAI目標檢測類,在第二行導入Python os類,在第三行定義了一個變量,獲取通往我們的Python文件、RetinaNet模型文件和圖像所在文件夾的路徑。

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))

在上面5行代碼中,我們在第一行定義我們的目標檢測類,在第二行設定RetinaNet的模型類型,在第三行將模型路徑設置爲RetinaNet模型的路徑,在第四行將模型加載到目標檢測類中,然後我們在第五行調用檢測函數,並在輸入和輸出圖像路徑中進行解析。

for eachObject in detections:
print(eachObject["name"] + " : " + eachObject["percentage_probability"] )

在上面兩行代碼中,我們迭代了第一行中detector.detectObjectFromImage函數返回的所有結果,然後打印出第二行中模型對圖像上每個物體的檢測結果(名稱和概率)。

ImageAI支持很多強大的目標檢測自定義功能,其中一項就是能夠提取在圖像上檢測到的每個物體的圖像。只需將附加參數extract_detected_objects=True解析爲detectObjectsFromImage函數,如下所示,目標檢測類就會爲圖像物體創建一個文件夾,提取每張圖像,將它們保存在新創建的文件夾中,並返回一個包含通過每張圖像的路徑的額外數組。

detections, extracted_images = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"), extract_detected_objects=True)

我們用前面的第一張圖像作爲例子,可以得到圖像中檢測到的各個物體的單獨圖像:

ImageAI提供了很多功能,能夠用於各類目標檢測任務的自定義和生產部署。包括:

-調整最小概率:默認概率小於50%的物體不會顯示,如有需要,你可以自行調整這個數字。

-自定義目標檢測:使用提供的CustomObject類,你可以檢測一個或多個特定物體。

-調整檢測速度:可以通過將檢測速度設爲“快”“更快”“最快”三個不同等級,調整檢測速度。

-輸入輸出類型:你可以自定義圖像的路徑,Numpy數組或圖像的文件流爲輸入輸出。

誠然,單看這10行代碼每一行,談不上驚豔,也藉助了不少庫,但是僅用10行代碼就能讓我們輕鬆實現之前很麻煩的目標檢測,還是能談得上“給力”二字。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章