Classificação de imagens - TensorFlow - SageMaker IA da Amazon

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Classificação de imagens - TensorFlow

O algoritmo de classificação de imagens - TensorFlow do Amazon SageMaker é um algoritmo de aprendizado supervisionado que oferece apoio para a 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 instâncias do Amazon EC2 e cadernos de exemplos de Classificação de Imagens - TensorFlow.

Recomendação de instâncias do Amazon EC2 para classificação de imagens - algoritmo TensorFlow

O algoritmo de classificação de imagens - TensorFlow oferece apoio 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.

Caderno de exemplo de classificação de imagens - TensorFlow

Para obter mais informações sobre como usar o algoritmo de classificação de imagens - TensorFlow do SageMaker para transferir aprendizado em um conjunto de dados personalizado, consulte o caderno Introdução à classificação de imagens - TensorFlow do SageMaker.

Para obter instruções sobre como criar e acessar instâncias do caderno Jupyter que você pode utilizar para executar o exemplo no SageMaker AI, consulte Instâncias de SageMaker notebook da Amazon. Depois de criar uma instância de caderno e abri-la, selecione a guia Exemplos do SageMaker AI para ver uma lista de todos os exemplos do SageMaker AI. Para abrir um caderno, escolha a guia Uso e depois escolha Criar cópia.