Image Classification - TensorFlow - Amazon SageMaker AI
Recomendación de instancias de Amazon EC2 para el algoritmo Image Classification - TensorFlowCuadernos de ejemplo

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.

Image Classification - TensorFlow

El algoritmo Image Classification - TensorFlow en Amazon SageMaker es un algoritmo de aprendizaje supervisado que admite el aprendizaje por transferencia con muchos modelos previamente entrenados desde 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 instancias de Amazon EC2 y cuadernos de muestra para Image Classification - TensorFlow.

Temas

Recomendación de instancias de Amazon EC2 para el algoritmo Image Classification - TensorFlow

El algoritmo Image Classification - TensorFlow es compatible con todas las instancias de CPU y GPU para el entrenamiento, lo que incluye lo siguiente:

  • 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

Para el entrenamiento con lotes grandes, recomendamos utilizar instancias de GPU con más memoria. Para la inferencia, se pueden usar instancias de CPU (como M5) o de GPU (P2, P3, G4dn o G5).

Cuadernos de ejemplo de Image Classification - TensorFlow

Para obtener más información sobre cómo utilizar el algoritmo Image Classification - TensorFlow en SageMaker para el aprendizaje por transferencia en un conjunto de datos personalizado, consulte el cuaderno Introduction to SageMaker TensorFlow - Image Classification.

Para obtener instrucciones sobre cómo crear instancias de cuadernos de Jupyter que pueda utilizar para ejecutar el ejemplo en SageMaker AI, consulte Instancias de Amazon SageMaker Notebook. Después de crear y abrir una instancia de cuaderno, seleccione la pestaña Ejemplos de SageMaker AI para ver una lista de todas las muestras de SageMaker AI. Para abrir un cuaderno, elija su pestaña Usar y elija Crear copia.