yolov5

1、下载代码YOLOV5-mask-42: 基于YOLOV5的口罩检测系统-提供教学视频

2、安装requirement.txt里面需要的软件

3、python detect.py --source data/images/bus.jpg --weights pretrained/yolov5s.pt 运行此句检查效果

4、pip install labelimg 安装标记软件。标记分成voc格式的xml格式和yolo的txt格式。此处用yolo格式生成txt。

5、数据集划分，将数据集下载到代码目录下，运行 list_dataset_file.py 划分数据集为train.txt和valid.txt。

6、创建mask_data.yaml修改里面的train和val的地址信息和mask_yolov5.yaml里面的信息。

7、运行python train.py --data mask_data.yaml --cfg mask_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-size 4 --device cpu

也可以直接在train.py文件中直接修改参数值。

8、

9、查看对应的结果。