Classification des images : TensorFlow

L'algorithme de classification des images Amazon SageMaker – TensorFlow est un algorithme d'apprentissage supervisé qui prend en charge l'apprentissage par transfert avec de nombreux modèles pré-entraînés provenant du hub TensorFlow. 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 instances Amazon EC2 et des exemples de blocs-notes pour Image Classification - TensorFlow.

Recommandation d’instance Amazon EC2 pour l’algorithme Image Classification - TensorFlow

L’algorithme Image Classification - TensorFlow prend en charge toutes les instances de CPU et de GPU pour l’entraînement, y compris :

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.

Exemples de blocs-notes Image Classification - TensorFlow

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

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