efficientdet-pytorch icon indicating copy to clipboard operation
efficientdet-pytorch copied to clipboard

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

Open RenhangRH opened this issue 3 years ago • 2 comments
trafficstars

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

RenhangRH avatar May 22 '22 13:05 RenhangRH

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

bubbliiiing avatar May 24 '22 13:05 bubbliiiing