各类机器学习框架保存的模型的格式

  • ONNX (.onnx, .pb, .pbtxt)
  • Keras (.h5, .keras)
  • Core ML (.mlmodel)
  • Caffe2 (predict_net.pb, predict_net.pbtxt)
  • MXNet (.model, -symbol.json) 
  • TensorFlow Lite (.tflite). 
  • Caffe (.caffemodel, .prototxt)
  • PyTorch (.pt, .pth)
  • TorchScript (.pt, .pth)
  • Torch (.t7), CNTK (.model, .cntk)
  • PaddlePaddle (__model__)
  • Darknet (.cfg)
  • NCNN (.param)
  • scikit-learn (.pkl)
  • TensorFlow.js (model.json, .pb)
  • TensorFlow (.pb, .meta, .pbtxt)
发布了227 篇原创文章 · 获赞 94 · 访问量 54万+

猜你喜欢

转载自blog.csdn.net/wiborgite/article/details/95724742