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