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 Amazon SageMaker AI Object Detection es un TensorFlow algoritmo de aprendizaje supervisado que admite el aprendizaje por transferencia con muchos modelos previamente entrenados del 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 EC2 instancias de Amazon y cuadernos de muestra para Object Detection - TensorFlow.

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

El TensorFlow algoritmo de detección de objetos es compatible con todas las instancias de GPU para el entrenamiento, incluidas las siguientes:

  • 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 instancias de SageMaker formación e inferencia en todas AWS las regiones, consulte Amazon SageMaker AI Pricing.

Detección de objetos: TensorFlow ejemplos de cuadernos

Para obtener más información sobre cómo utilizar el TensorFlow algoritmo de detección de objetos mediante SageMaker IA para transferir el aprendizaje en un conjunto de datos personalizado, consulte la introducción al SageMaker TensorFlow cuaderno de detección de objetos.

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.