任务完成人脸检测案例实践
一、下载安装yolov5
(这个是参考示例算法,请大家自行选择识别算法框架)
Yolov5官网路径如下:
https://github.com/ultralytics/yolov5
参考第一个例子,下载安装所需库:
https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
二、下载训练集进行模型训练
下载地址:https://github.com/ultralytics/yolov5/releases
yolov5共有四种模型:yolov5s、yolov5m、yolov5l、yolov5x。
不同的预训练模型 ,效果和精度不一样。
下载完成后,对模型进行训练
python detect.py --weights yolov5s.pt
yolov5m.pt
yolov5l.pt
yolov5x.pt
三、进行图像检测
人脸识别
指定好训练集,以及训练集权重后运行