Clasificación de imágenes - TensorFlow - Amazon SageMaker

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Clasificación de imágenes - TensorFlow

El algoritmo de clasificación de SageMaker imágenes de Amazon es un TensorFlow algoritmo de aprendizaje supervisado que admite el aprendizaje por transferencia con muchos modelos previamente entrenados desde el TensorFlow Hub. Utilice el aprendizaje por transferencia para ajustar uno de los modelos previamente entrenados disponibles en su propio conjunto de datos, aunque no haya una gran cantidad de datos de imagen disponibles. El algoritmo de clasificación de imágenes toma una imagen como entrada y genera una probabilidad para cada etiqueta de clase suministrada. Los conjuntos de datos de entrenamiento deben estar compuestos por imágenes en formato .jpg, .jpeg o .png. Esta página incluye información sobre las recomendaciones de EC2 instancias de Amazon y cuadernos de muestra para Image Classification - TensorFlow.

Recomendación de EC2 instancias de Amazon para el TensorFlow algoritmo de clasificación de imágenes

El TensorFlow algoritmo de clasificación de imágenes es compatible con todas CPU las GPU instancias de formación, incluidas las siguientes:

  • 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 GPU instancias con más memoria para el entrenamiento con lotes de gran tamaño. Se pueden CPU usar tanto las instancias (como la M5) como la GPU (P2, P3, G4dn o G5) para la inferencia.

TensorFlow Clasificación de imágenes: ejemplos de cuadernos

Para obtener más información sobre cómo utilizar el TensorFlow algoritmo de clasificación de SageMaker imágenes para el aprendizaje por transferencia en un conjunto de datos personalizado, consulte la introducción al SageMaker TensorFlow cuaderno de clasificación de imágenes.

Para obtener instrucciones sobre cómo crear instancias de Jupyter Notebook y acceder a ellas en las que puede ejecutar el ejemplo SageMaker, consulte. Instancias de Amazon SageMaker Notebook Tras crear una instancia de bloc de notas y abrirla, seleccione la pestaña SageMakerEjemplos para ver una lista de todos los SageMaker ejemplos. Para abrir un bloc de notas, elija su pestaña Usar y elija Crear copia.