大作业（自用）

改文章默认已经配置好了yolo虚拟环境且 可以运行项目本身给的案例图片

构建自己的数据集：建议使用roboflow直接构建

导出后如图所示

第一步

将文件结构拖成如图所示

第二步

修改data.yaml文件

train: C:\Users\icyler\Desktop\yolov5-code-main\dataset\train\images # 训练集绝对路径
val: C:\Users\icyler\Desktop\yolov5-code-main\dataset\valid\images # 验证集绝对路径
test: C:\Users\icyler\Desktop\yolov5-code-main\dataset\test\images # 测试集绝对路径

names: ['person'] # 训练的标签名

第三步

修改train.py中配置文件的位置

第四步（笔记本才设置）

如果使用的是笔记本需要配置显卡数量

第五步

检验结果

python detect.py --weights runs/train/exp4/weights/best.pt --source dataset/ps.mp4 --view-img