unihai的博客

原创 如何在windows系统下将yolov5的pt模型导出为onnx模型

最近在做本科毕业设计，要求是在树莓派上部署yolo算法来实现火灾检测，在网上查了很多资料，最后选择用yolov5s模型先试着在树莓派上部署，看下效果如何，由于从大佬那里拿到了yolov5火灾检测模型，但想要将它移植到树莓派上第一步要把pt模型转换成onnx模型，原因我想大概是因为树莓派算力不如pc，因此要转换成合适的模型进行部署。