Classification des images - TensorFlow - Amazon SageMaker AI
Recommandation d' EC2 instance Amazon pour l' TensorFlow algorithme de classification des imagesExemples de blocs-notes

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Classification des images - TensorFlow

L'algorithme Amazon SageMaker AI Image Classification est un TensorFlow algorithme d'apprentissage supervisé qui prend en charge l'apprentissage par transfert avec de nombreux modèles préentraînés issus du TensorFlow Hub. Utilisez l'apprentissage par transfert pour affiner l'un des modèles pré-entraînés disponibles sur votre propre jeu de données, même si une grande quantité de données d'image n'est pas disponible. L'algorithme de classification des images prend une image en entrée et génère en sortie une probabilité pour chaque étiquette de classe fournie. Les jeux de données d'entraînement doivent être composés d'images au format .jpg, .jpeg ou .png. Cette page contient des informations sur les recommandations relatives aux EC2 instances Amazon et des exemples de carnets de notes pour la classification des images - TensorFlow.

Rubriques

Recommandation d' EC2 instance Amazon pour l' TensorFlow algorithme de classification des images

L' TensorFlow algorithme de classification des images prend en charge toutes les instances de CPU et de GPU pour l'entraînement, notamment :

  • 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

Nous recommandons d'utiliser les instances de GPU avec davantage de mémoire pour l'entraînement avec de grandes tailles de lot. Les instances de CPU (telles que M5) et de GPU (P2, P3, G4dn ou G5) peuvent être utilisées pour l'inférence.

Classification des images - TensorFlow exemples de carnets

Pour plus d'informations sur l'utilisation de l' TensorFlow algorithme SageMaker AI Image Classification pour l'apprentissage par transfert sur un ensemble de données personnalisé, consultez le bloc-notes Introduction to SageMaker TensorFlow - Classification d'images.

Pour savoir comment créer et accéder à des instances de bloc-notes Jupyter que vous pouvez utiliser pour exécuter l'exemple dans SageMaker AI, consultez. Instances Amazon SageMaker Notebook Après avoir créé une instance de bloc-notes et l'avoir ouverte, sélectionnez l'onglet Exemples d'SageMaker IA pour afficher la liste de tous les exemples d' SageMaker IA. Pour ouvrir un bloc-notes, choisissez son onglet Use (Utiliser), puis Create copy (Créer une copie).