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.
Amazon SageMaker Neo prend en charge les frameworks suivants.
| Framework | Version du cadre | Version de modèle | Modèles | Formats de modèle (packagés dans *.tar.gz) | Boîtes à outils |
|---|---|---|---|---|---|
| MXNet | 1.8 | Prend en charge la version 1.8 ou antérieure | classification d'images, détection d'objets, segmentation sémantique, estimation de pose, reconnaissance d'activités | Un fichier de symboles (.json) et un fichier de paramètres (.params) | GluonCV v0.8.0 |
| ONNX | 1,7 | Prend en charge la version 1.7 ou antérieure | Classification d'images, SVM | Un fichier de modèle (.onnx) | |
| Keras | 2.2 | Prend en charge la version 2.2 ou antérieure | Classification d’images | Un fichier de définition de modèle (.h5) | |
| PyTorch | 1,7, 1,8 | Prend en charge la version 1.7, 1.8 ou antérieure | Classification d'images, détection d'objets | Un fichier de définition de modèle (.pth) | |
| TensorFlow | 1.15, 2.4, 2.5 (uniquement pour les instances ml.inf1.*) | Prend en charge les versions 1.15, 2.4, 2.5 (uniquement pour les instances ml.inf1.*) ou antérieures | Classification d'images, détection d'objets | *Pour les modèles enregistrés, un fichier .pb ou .pbtxt et un répertoire de variables contenant des variables *Pour les modèles figés, un seul fichier .pb ou .pbtxt | |
| TensorFlow-Léger | 1.15 | Prend en charge la version 1.15 ou antérieure | Classification d'images, détection d'objets | Un fichier de tampon plat de définition de modèle (.tflite) | |
| XGBoost | 1.3 | Prend en charge la version 1.3 ou antérieure | Arbres de décision | Un fichier de XGBoost modèle (.model) où le nombre de nœuds dans une arborescence est inférieur à 2^31 | |
| DARKNET | Classification des images, détection d'objets (le modèle Yolo n'est pas pris en charge) | Un fichier de configuration (.cfg) et un fichier de poids (.weights) |
