# 注：新数据应先放入下面路径的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