YoloV5 Android APP开发--ONNX和NCNN安装(二)

模型训练，根据不同的模型5l、5s等调整训练值。

python3 train.py --img 640 --batch 16 --epochs 77 --data lamp.yaml --weights yolov5s.pt --resume

python3 train.py --img 640 --batch 6 --epochs 60 --data lamp.yaml --weights yolov5l.pt

GTX1060显卡下，5l –batch设为6，5s设为16，迭代次数300(5s),60(5l),加—resume参数从上次训练结束的模型后继续训练。

训练生成的结果yolov5->run->train->weights下best.pt及last.pt

一、ONNX安装

Open Neural Network Exchange（ONNX，开放神经网络交换）格式，是一个用于表示深度学习模型的标准，可使模型在不同框架之间进行转移。