0
点赞
收藏
分享

微信扫一扫

【YOLO学习】yolo5-face论文里代码实现运行:持续更新中

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后,出现图五:

举报

相关推荐

0 条评论