Beispiele zum ONNX-Export
Wie erstelle ich ONNX-Dateien?
Im Folgenden werden mehrere Varianten exemplarisch aufgezeigt, wie Sie aus unterschiedlichen Frameworks bestimmte Modelle als ONNX exportieren können. Die Beispiele haben keinen Anspruch auf Vollständigkeit und dienen nur der ersten Orientierung. Für eine ausführliche Dokumentation sei auf jene der entsprechenden Frameworks verwiesen.
Die aufgeführten Beispiele beschränken sich auf das Erstellen einer ONNX-Datei. Beispiele zur Konvertierung, um sie in TwinCAT nutzbar zu machen, finden Sie hier: Konvertieren von ONNX in XML und BML sowie im verlinkten Samples ZIP-Archiv (siehe unten) im Ordner PythonAPI_mllib.
Übersicht verfügbarer Beispiele
Python Package | Modell-Typ | Option | Kommentar | Sample-Link |
---|---|---|---|---|
PyTorch | MLP Regressor |
|
| |
Keras | MLP Regressor |
|
| |
Scikit-learn | MLP Regressor |
|
| |
Scikit-learn | MLP Classifier |
| ONNX-Graph muss adaptiert werden | |
Scikit-learn | SVR |
|
| |
Scikit-learn | SVC | decision_function_shape='ovo' |
| |
Scikit-learn | k-means |
| Meta Key muss im ONNX eingetragen werden. | |
Scikit-learn | PCA |
|
| |
Scikit-learn | Decision Tree Classifier |
|
| |
Scikit-learn | Decision Tree Regressor |
|
| |
Scikit-learn | Extra Tree Classifier |
|
| |
Scikit-learn | Extra Tree Regressor |
|
| |
Scikit-learn | Extra Trees Classifier |
|
| |
Scikit-learn | Extra Trees Regressor |
|
| |
Scikit-learn | Random Forest Classifier |
|
| |
Scikit-learn | Random Forest Regressor |
|
| |
LightGBM | Random Forest Regressor |
| ONNX-Graph muss adaptiert werden | |
Scikit-learn | Gradient Boosting Classifier |
|
| |
Scikit-learn | Gradient Boosting Regressor |
|
| |
Scikit-learn | Hist Gradient Boosting Classifier |
|
| |
Scikit-learn | Hist Gradient Boosting Regressor |
|
| |
XGBoost | XGBClassifier | Nicht alle Konfigurationen lassen einen ONNX-Export zu | Package Version <= 1.5.2 oder >= 1.7.4 notwendig | |
XGBoost | XCBRegressor | Nicht alle Konfigurationen lassen einen ONNX-Export zu | Package Version <= 1.5.2 oder >= 1.7.4 notwendig | |
LightGBM | LGBMRegressor | Nicht alle Konfigurationen lassen einen ONNX-Export zu |
| |
LightGBM | LGBMClassifier |
| ONNX-Graph muss adaptiert werden |
Alle Samples können Sie hier als ZIP-Archiv herunterladen: Beckhoff_ONNX_Samples.zip