yolov5-pytorch模型转onnx模型再转tensorflow savedmode模型进行tfserving部署
pytorch框架下的yolov5模型部署到tensorflow serving,需要将pytorch的pt模型转换为onnx模型,再将onnx模型转换为tfserving的savedmode模型。1. pytorch的pt模型转onnx模型使用yolov5中自带的 export.py 脚本(在models下)可以将模型导出为 TorchScript, ONNX, CoreML。环境: yolov5-5.0的requirements.txt dependencies, including Python
原创
2021-05-15 10:26:48 ·
4177 阅读 ·
1 评论