| 模型格式 | 主要来源/用途 | 对应加载与执行库 | 支持语言/平台 |
|---|---|---|---|
PyTorch .pt / .pth / .ptl | PyTorch 保存的模型或权重 | 🔸 torch.load()(训练)🔸 torch.jit.load()(TorchScript 推理) | Python / C++ |
TorchScript (.pt) | PyTorch 的可执行格式(图模式) | libtorch、torch.jit | C++、Python |
ONNX .onnx | 通用开放格式,可由 PyTorch、TensorFlow、MXNet 等导出 | 🔸 onnxruntime(微软)🔸 TensorRT(NVIDIA)🔸 OpenVINO(Intel)🔸 MNN(阿里)🔸 TFLite(转换后) | 多平台(CPU/GPU/NPU) |
TensorFlow SavedModel / .pb | TensorFlow 模型保存格式 | 🔸 tensorflow(官方)🔸 tf-serving(服务端)🔸 tflite(移动端) | Python / C++ / Java |
Keras .h5 / .keras | Keras 训练的模型 | 🔸 keras.models.load_model() | Python |
TensorFlow Lite .tflite | 移动端 / 边缘设备优化格式 | 🔸 tflite.Interpreter | Android / iOS / Raspberry Pi |
CoreML .mlmodel | 苹果生态模型格式 | 🔸 coremltools / iOS CoreML Runtime | macOS / iOS |
TensorRT .plan / .engine | NVIDIA GPU 优化格式 | 🔸 tensorrt Runtime / C++ API | Linux / Windows(GPU) |
OpenVINO IR .xml + .bin | Intel 推理优化格式 | 🔸 openvino.runtime.Core() | Intel CPU / GPU / VPU |
MXNet .params + .json | Apache MXNet 模型 | 🔸 mxnet.model.load_checkpoint() | Python / C++ |
Caffe .prototxt + .caffemodel | Caffe 模型 | 🔸 caffe.Net() | C++ / Python |
PaddlePaddle .pdmodel + .pdparams | 百度飞桨 | 🔸 paddle.inference.create_predictor() | Python / C++ |
MindSpore .mindir | 华为 MindSpore | 🔸 mindspore.load_checkpoint() / load_mindir() | Python |
| gguf / ggml / safetensors | LLM(大语言模型)高效格式 | 🔸 llama.cpp、transformers、ctransformers、ggml | C++ / Python |
常见深度学习模型格式与对应加载执行库
最新推荐文章于 2025-12-02 20:45:02 发布
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
6万+

被折叠的 条评论
为什么被折叠?



