YOLOV5训练

简介

   YOLOv4还没有退热，YOLOv5就已经来了！

      6月9日，Ultralytics公司开源了YOLOv5，离上一次YOLOv4发布不到50天，不过这一次的YOLOv5是基于PyTorch实现的，而不是之前版本的darknet！

      根据官方给出的数字，YOLOv5的速度最快可以达到每秒140帧（FPS），但是权重文件只有YOLOv4的1/9，而且准确度更高。本次的发布的YOLOv5并不是一个单独的模型，而是一个模型家族，包括了YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x，要求Python 3.7和PyTorch 1.5以上版本。

1、环境

  Ubuntu

https://github.com/ultralytics/yolov5

2、数据标注---制作自己的数据集

工具labelImag ，Yolo为Txt格式

 https://github.com/jaccen2007/labelImg

3、数据集,可以使用已经标注过的公共数据集

https://public.roboflow.com

4 以口罩为例

yolov5l.yaml中80修改为2， 主要是2个分类，戴口罩与不戴口罩。

漫长的等待后

5、测试

      python detect.py --weight weights/best.pt --source  ../mask/test/images/***.jpg