Detección de objetos - TensorFlow - Amazon SageMaker AI

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.

Detección de objetos - TensorFlow

El algoritmo de detección de objetos - TensorFlow en Amazon SageMaker AI es un algoritmo de aprendizaje supervisado que admite el aprendizaje por transferencia con muchos modelos previamente entrenados desde TensorFlow Model Garden. 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 detección de objetos toma una imagen como entrada y genera una lista de casillas delimitadoras. 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 Object Detection - TensorFlow.

Recomendación de instancias de Amazon EC2 para el algoritmo de detección de objetos - TensorFlow

El algoritmo de detección de objetos - TensorFlow es compatible con todas las instancias de GPU para el entrenamiento, lo que incluye lo siguiente:

  • ml.p2.xlarge

  • ml.p2.16xlarge

  • ml.p3.2xlarge

  • ml.p3.16xlarge

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 o P3). Para obtener una lista completa de las instancias de entrenamiento e inferencia en todas las regiones de AWS, consulte Precios de Amazon SageMaker.

Cuadernos de muestra de detección de objetos - TensorFlow

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

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.