影像分類 - TensorFlow - Amazon SageMaker AI

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

影像分類 - TensorFlow

Amazon SageMaker AI Image Classification - TensorFlow 演算法是一種監督式學習演算法,支援使用 TensorFlow Hub 中許多預先訓練的模型進行遷移學習。即使沒有大量的影像資料,您也可以使用傳輸學習,在您自有的資料集上從可用的預先訓練模型中擇一來微調。影像分類演算法會將影像做為輸入,並輸出概率給每個所提供的類別標籤。訓練資料集必須包含 .jpg、.jpeg 或 .png 格式的影像。此頁面包含 Amazon EC2 執行個體建議和影像分類範例筆記本 - TensorFlow 的相關資訊。

Amazon EC2 執行個體推薦服務針對影像分類 - TensorFlow 演算法

影像分類 - TensorFlow 演算法支援所有 CPU 和 GPU 執行個體進行訓練,包括:

  • ml.p2.xlarge

  • ml.p2.16xlarge

  • ml.p3.2xlarge

  • ml.p3.16xlarge

  • ml.g4dn.xlarge

  • ml.g4dn.16.xlarge

  • ml.g5.xlarge

  • ml.g5.48xlarge

我們推薦以記憶體容量較大的 GPU 執行個體用於大批次訓練。CPU (例如 M5) 和 GPU (P2、P3、G4dn 或 G5) 執行個體都可用於推論。

影像分類 - TensorFlow 範例筆記本

如需如何使用 SageMaker AI Image Classification - TensorFlow 演算法在自訂資料集上傳輸學習的詳細資訊,請參閱 SageMaker TensorFlow - Image Classification notebook 簡介

如需如何建立和存取可用於在 SageMaker AI 中執行範例的 Jupyter 筆記本執行個體的說明,請參閱 Amazon SageMaker 筆記本執行個體。在您建立並開啟筆記本執行個體之後,請選取 SageMaker AI 範例索引標籤以查看所有 SageMaker AI 範例的清單。若要開啟筆記本,請選擇其使用標籤,然後選擇建立複本