基于yolov5的图像识别

任务完成人脸检测案例实践

一、下载安装yolov5

(这个是参考示例算法,请大家自行选择识别算法框架)
Yolov5官网路径如下:
https://github.com/ultralytics/yolov5

基于yolov5的图像识别_第1张图片

参考第一个例子,下载安装所需库:
https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

二、下载训练集进行模型训练

下载地址:https://github.com/ultralytics/yolov5/releases
基于yolov5的图像识别_第2张图片

yolov5共有四种模型:yolov5s、yolov5m、yolov5l、yolov5x。
不同的预训练模型 ,效果和精度不一样。
下载完成后,对模型进行训练

python detect.py --weights yolov5s.pt  
                           yolov5m.pt
                           yolov5l.pt
                           yolov5x.pt
三、进行图像检测

人脸识别
基于yolov5的图像识别_第3张图片
基于yolov5的图像识别_第4张图片
指定好训练集,以及训练集权重后运行
基于yolov5的图像识别_第5张图片
基于yolov5的图像识别_第6张图片

你可能感兴趣的:(计算机视觉,深度学习,python)