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"/] 現在切換到範例目錄來對鸚鵡的照片進行影像分類，先來看看這隻鸚鵡： [caption id="" align="aligncenter" width="166"] https://coral.withgoogle.com/static/docs/images/parrot.jpg[/caption] [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"/] 範例的貓與人臉 [caption id="" align="aligncenter" width="513"] https://coral.withgoogle.com/static/docs/images/cat.jpg[/caption] [caption id="" align="aligncenter" width="381"] https://coral.withgoogle.com/static/docs/images/face.jpg[/caption] 執行範例： [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 框起來 [caption id="" align="aligncenter" width="485"] https://coral.withgoogle.com/static/docs/images/examples/object_detection_results.jpg[/caption] 換一張圖試試看 (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"/]