查看原文
其他

行人检测(人体检测)2:YOLOv5实现人体检测(含人体检测数据集和训练代码)

AI吃大瓜 AI吃大瓜 2023-02-19

 这是博文《行人检测(人体检测)2:YOLOv5实现人体检测(含人体检测数据集和训练代码)》项目代码


1.项目介绍

 这是项目《行人检测(人体检测)》系列之《YOLOv5实现人体检测(含人体检测数据集和训练代码)》;项目基于开源YOLOv5项目,实现一个高精度的人体检测(行人检测)算法( Person Detection, Pedestrian Detection);

目前,基于YOLOv5s的人体检测精度平均值mAP_0.5=0.98432,mAP_0.5:0.95=0.84354。为了能部署在手机Android平台上,本人对YOLOv5s进行了简单的模型轻量化,并开发了一个轻量级的版本yolov5s05_416和yolov5s05_320模型;轻量化模型在普通Android手机上可以达到实时的检测效果,CPU(4线程)约30ms左右,GPU约25ms左右 ,基本满足业务的性能需求。下表格给出轻量化模型的计算量和参数量以及其检测精度


整套项目源码内容包含:人体检测数据集 + YOLOv5训练代码和测试代码

(1)人体检测数据集:

  1. VOC人体检测数据集:VOC2007共9963张图片,VOC2012共17125张图片,已经标注了人脸框(face)和人体框(person)

  2. COCO人体检测数据集:总共66697张含有人体的图片,已经标注了人脸框(face)和人体框(person)

  3. MPII人体检测数据集:总共21430张含有人体的图片,已经标注了人脸框(face)和人体框(person)


(2)YOLOv5训练代码和测试代码(Pytorch)

  1. 整套YOLOv5项目工程,含训练代码train.py和测试代码demo.py

  2. 支持高精度版本yolov5s训练和测试

  3. 支持轻量化版本yolov5s05_320和yolov5s05_416训练和测试

  4. 根据本篇博文说明,简单配置即可开始训练:train.py

  5. 测试代码demo.py支持图片,视频和摄像头测试



2.详细说明

    详细说明,请查看原文《行人检测(人体检测)2:YOLOv5实现人体检测(含人体检测数据集和训练代码)》




3.资源下载

    敬请网友尊重原创成果,请勿随意转发资源!谢谢合作;

    关于项目的任何问题,均可在博客评论留言~



    下面是项目下载地址:

《行人检测(人体检测)2:YOLOv5实现人体检测(含人体检测数据集和训练代码)》的项目源码

【源码链接下载】

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存