yolov5训练的结果转为onnx

1、ubuntu服务器下训练完了,得到best.pb

2、在windows下进行转换为onnx(忘了在ubuntu下转换)

3、方法如下,备份下后修改export.py

yolov5训练的结果转为onnx_第1张图片

4、修改chenexport.py中的模型位置

yolov5训练的结果转为onnx_第2张图片

   parser.add_argument('--weights', type=str, default='../chenresult/best.pt', help='weights path')

 5、运行后生成onnx

yolov5训练的结果转为onnx_第3张图片

你可能感兴趣的:(jeston,nano,深度学习)