yolov5-face是基于yolov5基础上的人脸识别,
感兴趣的朋友可以先看看yolov5相关,yolov5没有论文,
yolov5项目地址:ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)https://github.com/ultralytics/yolov5
yolo5-face 的论文:
[2105.12931] YOLO5Face: Why Reinventing a Face Detector (arxiv.org)https://arxiv.org/abs/2105.12931yolov5-face的项目地址:
deepcam-cn/yolov5-face: YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931) (github.com)https://github.com/deepcam-cn/yolov5-face
**************************************************************************************************************
YOLOv5-face代码运行
**************************************************************************************************************
第一步:下载yolo5-face
第二步:下载数据集 widerface数据集。
下载网址:
https://drive.google.com/file/d/1tU_IjyOwGQfGNUvZGwWWM4SwxKp2PUQ8/view?usp=sharing
下载后,解压缩位置放到yolov5-face-master项目里data文件夹下的widerface文件夹下。
接着运行train2yolo.py和val2yolo.py文件,这俩个文件是把数据集转成yolo训练用的格式。
直接运行会出错,如图二,
(需要改变文件路径。改变retinaface2yolo.py中103和104行,
如图三,换成自己的存放地址
运行后出现照片和转化后的.txt文件。项目中没有要求做这个,直接要求运行train2yolo.py和val2yolo.py文件,)
在data文件夹下新建widerfaceyolo文件夹,放widerface文件夹对应的yolo所需的文件。
修改train2yolo.py文件:注释掉104-126行,修改读取路径和保存路径,如图4_1,4_2:
运行train2yolo.py后,出现图五: