Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
87 lines
3.1 KiB
Markdown
Executable File
87 lines
3.1 KiB
Markdown
Executable File
1,链接webserver
|
||
地址：192.168.1.175
|
||
密码：123
|
||
用户名：xzl
|
||
|
||
2,remmina 远程传数据
|
||
|
||
/home/xzl/yolov5-6.2 训练代码
|
||
/mnt/sas/xzl13/datasets/isa_detect 检测数据集
|
||
/home/xzl/datasets 分类数据集（最新的是 isa_class_1120）
|
||
|
||
|
||
3,vscode
|
||
|
||
/home/xzl/yolov5-6.2 训练工程代码文件夹
|
||
激活训练环境
|
||
conda activate /mnt/sas/xzl13/conda/yolov5s62
|
||
|
||
3.1 检测模型
|
||
3.1.1 数据处理
|
||
标注后的图片和xml标注文件整理到一个文件夹src_data下面。
|
||
所有.jpg图片放到src_data/images,所有xml文件放到src_data/labels
|
||
tar cvf isa_detect_1120.tar.gz src_data
|
||
/mnt/sas/xzl13/datasets 下新建文件夹 isa_detect_1120
|
||
remmina传输文件到下面
|
||
解压
|
||
cd /mnt/sas/xzl13/datasets/isa_detect_1120
|
||
tar xvf isa_detect_1120.tar.gz
|
||
|
||
执行脚本处理数据
|
||
python isa_detect_preprocess.py --src-data ./isa_detect_1124_test/src_data
|
||
|
||
拷贝检测数据到 /mnt/sas/xzl13/datasets/isa_detect
|
||
cd /mnt/sas/xzl13/datasets/isa_detect_1120
|
||
cp -r dst_data/* ../isa_detect/
|
||
|
||
3.1.2 服务器训练检测模型
|
||
检测训练脚本位置：/home/xzl/yolov5-6.2/train.py
|
||
cd /home/xzl/yolov5-6.2/
|
||
nohup python train.py --data isa-detect.yaml --weights '' --cfg yolov5s_isa.yaml --img 640 --batch-size 128 --device 1,2 > train_isa_detect_1124_log &
|
||
|
||
only modify train_isa_detect_1120_log for log
|
||
|
||
3.1.3 导出模型为onnx
|
||
训练完模型保存位置/home/xzl/yolov5-6.2/runs/train/exp6/weights/best.pt
|
||
执行导出转换脚本
|
||
cd /home/xzl/yolov5-6.2/
|
||
python export.py --weights runs/train/exp6/weights/best.pt --include onnx
|
||
导出文件位置：/home/xzl/yolov5-6.2/runs/train/exp6/weights/best.onnx
|
||
|
||
|
||
|
||
|
||
3.2 分类模型
|
||
|
||
3.2.1 从检测数据集roi截图
|
||
cd /mnt/sas/xzl13/datasets/
|
||
python isa_roi_img.py --train-label ./isa_detect_1120/dst_data/labels/train --val-label ./isa_detect_1120/dst_data/labels/val --save-dir ./isa_detect_1120/roi
|
||
roi截图后保存在/mnt/sas/xzl13/datasets/isa_detect_1120/roi下面。下载到本地，用于下一步分类处理数据
|
||
|
||
3.2.2 数据处理
|
||
从webserver下载上次训练数据集src_data到本地
|
||
将欧洲回传的截图或检测数据集roi截图拷贝到src_data相应文件夹下
|
||
|
||
在webserser的/home/xzl/datasets/ 下新建文件夹 isa_class_1120 ，并将本地处理完的src_data数据上传到此目录下。
|
||
|
||
分类数据处理脚本:
|
||
cd home/xzl/datasets/
|
||
python isa_preprocess.py --src-data ./isa_class_1120/src_data
|
||
cp /home/xzl/datasets/isa_class_1120/dst_data/* /home/xzl/datasets/isa_class_1120
|
||
|
||
3.2.3 训练
|
||
/home/xzl/yolov5-6.2/classify/train.py 分类训练脚本
|
||
nohup python classify/train.py --model mobilenet_v3_large --data isa_class_1120 --epochs 80 --img 224 --device 3 > train_isa_clas_1120_log &
|
||
|
||
--data isa_class_1120 指定数据集到/home/xzl/datasets/isa_class_1120
|
||
|
||
|
||
训练完，生成文件在/home/xzl/yolov5-6.2/runs/train-cls/exp5下面。
|
||
模型文件在 /home/xzl/yolov5-6.2/runs/train-cls/exp5/weights/best.pt
|
||
|
||
3.2.4 导出onnx模型
|
||
cd /home/xzl/yolov5-6.2
|
||
python export.py --weights runs/train-cls/isa1120_classify_mobilenetv3_large/weights/best.pt --include onnx --imgsz 224
|
||
|
||
--weights 指定新生成的模型文件
|
||
转生成onnx模型， |