使用TensorFlow，TensorFlow Lite和TensorRT模型（圖像，視頻，網絡攝像頭）進行YOLOv4對象檢測

該視頻將逐步介紹設置代碼，安裝依賴項，將YOLO Darknet樣式權重轉換爲已保存的TensorFlow模型以及運行模型的步驟。利用YOLOv4做爲TensorFlow Lite模型的優點，它的小巧輕巧的尺寸使其很是適合移動和邊緣設備（如樹莓派）。想要利用GPU的所有功能？而後使用TensorFlow TensorRT運行YOLOv4，以將性能提升多達8倍。

 

在這裏獲取代碼：

https://github.com/theAIGuysCode/tensorflow-yolov4-tflite

在此視頻中，介紹了：

1.克隆或下載代碼

2.安裝CPU或GPU的必需依賴項

3.下載並將YOLOv4權重轉換爲已保存的TensorFlow

4.使用TensorFlow對圖像，視頻和網絡攝像頭執行YOLOv4對象檢測

5.將TensorFlow模型轉換爲TensorFlow Lite .tflite模型

6.將TensorFlow模型轉換爲TensorFlow TensorRT模型

7.使用TensorFlow Lite運行YOLOv4對象檢測

 

YOLOv4官方論文：

https://arxiv.org/abs/2004.10934

推薦閱讀

30個最大的機器學習TensorFlow數據集

點擊「閱讀原文」圖書配套資源

本文分享自微信公衆號 - 相約機器人（xiangyuejiqiren）。
若有侵權，請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」，歡迎正在閱讀的你也加入，一塊兒分享。