win10系统下paddlepaddle-2.5训练自己数据集并转为onnx进行部署
win10paddledetection训练自己数据集
·
下载paddledetection
https://github.com/PaddlePaddle/PaddleDetection
下载适合 自己python和cuda版本的paddlepaddle
pycharm运行代码，修改paddledetection中模型的配置文件，选择自己想用的模型
修改配置文件中数据集文件的信息，包括数据集路径，训练集，和测试集
开始训练
对训练好的模型进行部署
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.6/deploy/EXPORT_ONNX_MODEL.md
利用export_model.py导出模型，直接点击就可以，配置文件 在ci.py中已经添加，模型的导出路径选择默认
安装paddle2onnx
使用该命令将模型导出为.onnx格式
更新，使用转换命令后出现新的问题，opset需要设置的更高，只需将opset 11 改为16
安装onnxruntime-gpu来支持gpu推理
采用infer_draw进行推理，修改参数包括，配置文件路径，模型路径，图像路径
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)