yolov5系列-yolov5模型部署到安卓手机、nvidia-jetson

---

1. 训练安全帽yolov5s模型

略

2. 安全帽模型部署到安卓app

2.1 导出需要的onnx文件

1. 修改export.py文件
   export_helmet_yolov5s_my.py文件由export.py复制而来,修改data、weights

# data：训练模型的data
parser.add_argument('--data', type=str, default=ROOT / 'data/mydata_helmet/mydata.yaml', help='dataset.yaml path')
# 训练好的模型:第一步得到安全帽的yolov5s模型
parser.add_argument('--weights', type=str, default=ROOT / 'weights/my_models/helmet/yolov5s/train/exp2/weights/best.pt', help='weights path')

2. 生成 onnx 文件
执行脚本会在模型文件夹下生成best.onnx、best.torchscript.pt两个文件

python export_yolov5n6.py --