Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.
Kerangka Kerja yang Didukung
Amazon SageMaker Neo mendukung kerangka kerja berikut.
Kerangka Kerja | Framework Versi | Versi Model | Model | Format Model (dikemas dalam *.tar.gz) | Toolkit |
---|---|---|---|---|---|
MXNet | 1.8 | Mendukung 1.8 atau yang lebih baru | Klasifikasi Gambar, Deteksi Objek, Segmentasi Semantik, Estimasi Pose, Pengenalan Aktivitas | Satu file simbol (.json) dan satu file parameter (.params) | GluonCV |
ONNX | 1.7 | Mendukung 1.7 atau yang lebih baru | Klasifikasi Citra Citra, SVM | Satu file model (.onnx) | |
Keras | 2.2 | Mendukung 2.2 atau yang lebih baru | Klasifikasi Citra | Satu file definisi model (.h5) | |
PyTorch | 1.7, 1,8 | Mendukung 1.7, 1.8 atau sebelumnya | Klasifikasi Gambar, Deteksi Objek | Satu file definisi model (.pth) | |
TensorFlow | 1.15, 2.4, 2.5 (hanya untuk instans ml.inf1.*) | Mendukung 1.15, 2.4, 2.5 (hanya untuk instans ml.inf1.*) atau sebelumnya | Klasifikasi Gambar, Deteksi Objek | * Untuk model yang disimpan, satu file.pb atau satu file.pbtxt dan direktori variabel yang berisi variabel*Untuk model beku, hanya satu file.pb atau .pbtxt | |
TensorFlow-Ringan | 1.15 | Mendukung 1.15 atau yang lebih baru | Klasifikasi Gambar, Deteksi Objek | Satu model definisi flatbuffer file (.tflite) | |
XGBoost | 1.3 | Mendukung 1.3 atau yang lebih baru | Pohon Keputusan | Satu file model XGBoost (.model) di mana jumlah node dalam pohon kurang dari 2 ^ 31 | |
DARKNET | Klasifikasi Gambar, Deteksi Objek (model Yolo tidak didukung) | Satu file konfigurasi (.cfg) dan satu file bobot (.weights) |