Google Coral TPU Edge開發板 – 執行影像分類與物件辨識範例

延續上一篇[Google Coral TPU Edge開發板 – 硬體介紹與開機],現在要示範基本的影像分類與物件偵測範例。

原廠網站也提供了很多範例,都可以做做看

Edge TPU Python API – 影像分類範例

預設的 edgetpu_demo sh是透過 Edge TPU Python library (edgetpu Python module) 來對影像中的每一個 frame 進行推論,可以看到效能還不賴。本函式庫包含了各種API,幫助您快速進行影像分類或物件辨識等相關視覺推論。後續還有其他範例來說明如何使用本API。

首先,請在 Coral 開發板的 terminal 中,回到 home 目錄並下載鳥類分類模型、標籤檔與分類用的照片:

[pastacode lang=”bash” manual=”cd%20~%0A%0Awget%20https%3A%2F%2Fdl.google.com%2Fcoral%2Fcanned_models%2Fmobilenet_v2_1.0_224_inat_bird_quant_edgetpu.tflite%20%5C%0Ahttps%3A%2F%2Fdl.google.com%2Fcoral%2Fcanned_models%2Finat_bird_labels.txt%20%5C%0Ahttps%3A%2F%2Fcoral.withgoogle.com%2Fstatic%2Fdocs%2Fimages%2Fparrot.jpg” message=”” highlight=”” provider=”manual”/]

現在切換到範例目錄來對鸚鵡的照片進行影像分類,先來看看這隻鸚鵡:

https://coral.withgoogle.com/static/docs/images/parrot.jpg
[pastacode lang=”bash” manual=”cd%20%2Fusr%2Flib%2Fpython3%2Fdist-packages%2Fedgetpu%2Fdemo%2F%0A%0Apython3%20classify_image.py%20%5C%0A–model%20~%2Fmobilenet_v2_1.0_224_inat_bird_quant_edgetpu.tflite%20%5C%0A–label%20~%2Finat_bird_labels.txt%20%5C%0A–image%20~%2Fparrot.jpg” message=”” highlight=”” provider=”manual”/]

應該會看到類似的結果:

物件辨識範例

由此取得本範例程式碼object_detection.py 會透過 DetectionEngine API 進行物件辨識,這會用到偵測模型、標籤檔與要被偵測的影像。如果不指定標籤檔的話,預設會偵測臉部 (當然需要提供 face detection model).

本範例會使用 MobileNet SSD,已預先訓練好可偵測1,000種不同類型的物件 (或單純人臉)。

先取得檔案:

[pastacode lang=”bash” manual=”” message=”” highlight=”” provider=”manual”/]

範例的貓與人臉

https://coral.withgoogle.com/static/docs/images/cat.jpg
https://coral.withgoogle.com/static/docs/images/face.jpg

執行範例:

[pastacode lang=”bash” manual=”cd%20%2Fusr%2Flib%2Fpython3%2Fdist-packages%2Fedgetpu%2Fdemo%0A%0Apython3%20object_detection.py%20%5C%0A–model%20%24EXAMPLE_DIR%2Fmobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite%20%5C%0A–label%20%24EXAMPLE_DIR%2Fcoco_labels.txt%20%5C%0A–input%20%24EXAMPLE_DIR%2Fcat.jpg%20%5C%0A–output%20%24EXAMPLE_DIR%2Fobject_detection_results.jpg” message=”” highlight=”” provider=”manual”/]

應可看到以下結果,並會在 –output 資料夾把 cat 框起來

https://coral.withgoogle.com/static/docs/images/examples/object_detection_results.jpg

換一張圖試試看 (face.jpg)

[pastacode lang=”bash” manual=”python3%20object_detection.py%20%5C%0A–model%20%24EXAMPLE_DIR%2Fmobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite%20%5C%0A–input%20%24EXAMPLE_DIR%2Fface.jpg%20%5C%0A–output%20%24EXAMPLE_DIR%2Fface_detection_results.jpg” message=”” highlight=”” provider=”manual”/]

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *