Clasificación de imágenes - TensorFlow

El algoritmo Amazon SageMaker AI Image Classification 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:

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 imágenes mediante SageMaker IA 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, que puede utilizar para ejecutar el ejemplo en SageMaker IA, consulte. Instancias de Amazon SageMaker Notebook Una vez que haya creado una instancia de bloc de notas y la haya abierto, seleccione la pestaña Ejemplos de SageMaker IA para ver una lista de todos los ejemplos de SageMaker IA. Para abrir un bloc de notas, elija su pestaña Usar y elija Crear copia.