Classificação de imagens - TensorFlow

O algoritmo Amazon SageMaker AI Image Classification - é um TensorFlow algoritmo de aprendizado supervisionado que oferece suporte ao aprendizado por transferência com muitos modelos pré-treinados do TensorFlow Hub. Use transferir aprendizado para ajustar um dos modelos pré-treinados disponíveis em seu próprio conjunto de dados, mesmo que uma grande quantidade de dados de imagem não esteja disponível. O algoritmo de classificação de imagens usa uma imagem como entrada e gera uma probabilidade para cada um dos rótulos de classe. Os conjuntos de dados de treinamento devem consistir em imagens no formato .jpg, .jpeg ou .png. Esta página inclui informações sobre recomendações de EC2 instâncias da Amazon e exemplos de notebooks para classificação de imagens - TensorFlow.

Tópicos

• Como usar o TensorFlow algoritmo de classificação de imagens por SageMaker IA

• Interface de entrada e saída para o TensorFlow algoritmo de classificação de imagens

• Recomendação de EC2 instância da Amazon para o TensorFlow algoritmo de classificação de imagens

• Classificação de imagens - TensorFlow exemplos de cadernos

• Como TensorFlow funciona a classificação de imagens

• TensorFlow Modelos de hub

• Classificação de imagens - TensorFlow Hiperparâmetros

• Ajustar uma classificação de imagens - TensorFlow modelo

Recomendação de EC2 instância da Amazon para o TensorFlow algoritmo de classificação de imagens

O TensorFlow algoritmo de classificação de imagens oferece suporte a todas as instâncias de CPU e GPU para treinamento, incluindo:

• 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

Recomendamos o uso de instâncias de GPU com mais memória para treinamento com grandes tamanhos de lote. Tanto as instâncias de CPU (como M5) quanto as de GPU (P2, P3, G4dn ou G5) podem ser usadas para inferência.

Classificação de imagens - TensorFlow exemplos de cadernos

Para obter mais informações sobre como usar o TensorFlow algoritmo SageMaker AI Image Classification para transferir o aprendizado em um conjunto de dados personalizado, consulte o caderno Introdução à SageMaker TensorFlow Classificação de Imagens.

Para obter instruções sobre como criar e acessar instâncias do notebook Jupyter que você pode usar para executar o exemplo em SageMaker IA, consulte. Instâncias do Amazon SageMaker Notebook Depois de criar uma instância do notebook e abri-la, selecione a guia Exemplos de SageMaker IA para ver uma lista de todas as amostras de SageMaker IA. Para abrir um caderno, escolha a guia Uso e depois escolha Criar cópia.