爱吃肉的鹏 2021-03-20 12:33 采纳率: 25%
浏览 405

pytorch训练好的模型如何工业部署

在pytorch环境下，基于yolov4训练好的模型。现在想要试试工业部署，按网上教程将模型pth文件转化成了onnx文件，然后又将onnx文件转成了caffe2下的文件，文件格式为pb格式。接下来我应该再如何去做呢？

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2021-03-21 22:03
    关注

    看你生产环境是什么，c#，.net,c/c++还是java？使用的框架又是什么，用的什么推理方式，opencv还是openvino还是tenseRT？这些东西先确定下在说后面的部署，不然你部署个tensrt的，然后告诉你生产环境不支持这框架，那也是白费

    评论

报告相同问题？