search

bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码，可以用于训练自己的模型。
MIT License
305 stars 61 forks source link

如何将训练好的.pth权重和模型转换为 .onnx or .pb通用模型文件 #40

Open RenhangRH opened 2 years ago

RenhangRH commented 2 years ago

非常感谢博主视频和博文，已经跟了博主学习了一段时间，但是无法将训练好的权重转换为onnx或pb模型，建议能否出一期视频或博文，专门讲如何把训练好的权重与模型的各种格式文件(如.pt .pth.h5)转换为.onnx或.pb后缀的通用模型文件，以便于其他平台部署和推理。这有利于将视觉深度学习在工业环境中得到应用，非常感谢。

bubbliiiing commented 2 years ago

这个库貌似自定义的有点多，转不了onnx