Yolov5_Efficient

Use yolov5 efficiently(高效地使用Yolo v5)

1.Introduction-介绍

The repository is reconstructed and annotated based on UltralyTICS / YOLOV5, and other functions are added thereto, such as automatic annotation with the Grab Cut, and the pointing center point.

该存储库基于Ultralytics/yolov5进行重构与注释，并且在此基础上加入其他功能，例如自动标注与漫水填充GrabCut，以及绘制中心点进行连线。

2.Performance-效果表现

2.1 Yolo detect+GrabCut

detect预测结果	GrabCut分割结果

demo video

2.2 Yolo center point line 中心点连线

demo1	demo2

2.3 autolabel半自动标注

demo video

3.How to Use-用法

3.1 Train 训练自定义数据集

可参考Ultralytics/yolov5的训练用法：

3.1.1命令行方式：

单独使用：

python yolov5_master/train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt

集成使用：

python yolov5_master/main.py

3.1.2 IDE方式：

3.2 detect 推理预测

3.2.1命令行方式：

单独使用-测试图片

python yolov5_master/detect.py --source ./testfiles/img1.jpg --weights runs/train/bmyolov5s/weights/best.pt

单独使用-测试视频

python yolov5_master/detect.py --source ./testfiles/video.mp4 --weights runs/train/bmyolov5s/weights/best.pt

集成使用

python yolov5_master/main.py

集成方式中包含参数设置与训练、预测的配置选择，可自行更换。

3.3 Yolo detect+GrabCut使用

直接在IDE中修改配置即可。

3.4 Yolo center point line使用