TensorFlow 電腦視覺

TensorFlow 提供多種電腦視覺 (CV) 和圖片分類工具。本文件將介紹其中一些工具,並概述相關資源,協助您開始執行常見的 CV 工作。

視覺程式庫與工具

TensorFlow 透過高階 Keras 程式庫和低階 tf.image 模組提供 CV 工具。在大多數情況下,Keras 程式庫會比內建的 TensorFlow 替代方案更方便。但如果 Keras 選項不符合您的使用案例,或者您想要對圖片預先處理進行更低階的控制,您可能需要使用低階 TensorFlow 工具。

KerasCV

如果您剛開始進行 CV 專案,並且不確定需要哪些程式庫和工具,那麼 KerasCV 會是不錯的起點。KerasCV 是以 Keras Core 為基礎建構的模組化 CV 元件程式庫。KerasCV 包含模型、層、指標、回呼和其他工具,可擴充高階 Keras API 以用於 CV 工作。KerasCV API 可協助進行資料擴增、分類、物件偵測、區隔、圖片產生和其他常見的 CV 工作流程。您可以使用 KerasCV 快速組裝生產等級、最先進的訓練和推論管線。

Keras 工具

tf.keras.utils 提供多種高階圖片預先處理工具。例如,tf.keras.utils.image_dataset_from_directory 會從磁碟上的圖片目錄產生 tf.data.Dataset

tf.image

如果 KerasCV 不符合您的使用案例,您可以使用 tf.imagetf.data 來編寫您自己的資料擴增管線或層。

tf.image 模組包含各種圖片處理函式,例如 tf.image.flip_left_righttf.image.rgb_to_grayscaletf.image.adjust_brightnesstf.image.central_croptf.image.stateless_random*

tf.data API 可讓您從簡單、可重複使用的片段建構複雜的輸入管線。

TensorFlow Datasets

TensorFlow Datasets 是可與 TensorFlow 搭配使用的資料集集合。許多資料集 (例如 MNISTFashion-MNISTTF Flowers) 可用於開發及測試電腦視覺演算法。

從哪裡開始

下列資源將協助您開始使用 TensorFlow 和 Keras CV 工具。