暂无描述

zbc f9c7009403 first commit 3 周之前
code f9c7009403 first commit 3 周之前
readme.md f9c7009403 first commit 3 周之前

readme.md

注:新数据应先放入下面路径的total_data中(total_data需新建,在本地导入数据放入;项目名称根据具体项目新建)

code and data location url:192.168.20.250 directory:/data/object_detection

训练模型执行步骤 1、./object_detection.sh /data2/object_detection/data/image/项目名称/

2、训练结束,保存.pt格式:

3、.pt格式模型导出为.pb格式模型 获取pt文件路径:cat export.sh
进入yolov5文件夹 (/data/object_detection/code/yolov5/)后运行 python export.py --weights ./runs/train/helmet_fall_phone/weights/best.pt --include pb #修改项目名称

转换为.pb格式,并保存

4、测试: python detect.py --weights runs/train/safebelt_data/weights/best.pt --source /data2/object_detection/code/yolov5/safebelt_test --data /data2/object_detection/data/image/safebelt_data/yolo/safebelt_data.yaml --device 0