ROBOT_IMAGE_PROCESS_DETECTION_MODEL_TRAIN_v2.0.1

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