- 博客(4)
- 收藏
- 关注
RSS订阅原创 yolov5模型直接导出torchscript格式、onnx格式
最近在学习训练yolov5模型，但在导出onnx格式文件总是出错，参考了很多文章都是采用直接运行export.py文件，但我确只导出了torchscript格式文件。我运行以上代码，导出的只有 torchscript格式文件，并没有其他人写的onnx格式文件，最后我上github的查看官方的文档，发现导出语句已经进行更改如下。成功解决问题导出了best.onnx文件！
2023-09-23 14:34:00
1642
1
原创 将一个BP神经网络预测模型部署Android studio
由于读研导师的研究方向的需要，刚接触关于模型导入的问题，这次记录一下我的新手历程，文章参考pytorch的官方文档和大佬的文章。
2023-08-23 15:00:21
518
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人