从零部署YOLOv5人脸检测:环境搭建、数据标注到实时应用
1. 环境准备从零搭建YOLOv5开发环境第一次接触YOLOv5时我花了两天时间才把环境配好主要卡在CUDA和PyTorch版本兼容问题上。后来发现其实用conda管理环境特别方便这里分享我的踩坑经验。建议使用Ubuntu 20.04系统实测NVIDIA驱动安装最稳定。先确保你的显卡驱动正常运行nvidia-smi能看到显卡信息。安装Anaconda后创建一个专门的环境conda create -n yolov5 python3.8 conda activate yolov5关键依赖安装顺序很重要先装PyTorch建议1.7.1版本conda install pytorch1.7.1 torchvision0.8.2 torchaudio0.7.2 cudatoolkit11.0 -c pytorch再装其他依赖pip install opencv-python matplotlib pandas tqdm seaborn克隆YOLOv5官方仓库时推荐用国内镜像加速git clone https://gitee.com/mirrors/YOLOv5.git cd yolov5 pip install -r requirements.txt注意如果遇到SSL证书错误可以尝试在pip命令后加--trusted-host pypi.org --trusted-host files.pythonhosted.org2. 数据准备高效标注人脸数据集我用过三种标注工具最终发现LabelImg最适合新手。安装很简单pip install labelImg labelImg打开软件后记得设置两点右下角选择YOLO格式保存目录选labels文件夹标注技巧框选时尽量贴近人脸边缘遇到侧脸时以可见面部区域为准遮挡超过50%的人脸建议不标注数据集划分我建议按7:2:1的比例训练集70%验证集20%测试集10%目录结构示例CASIA-FaceV5/ ├── images │ ├── train │ ├── val │ └── test └── labels ├── train ├── val └── test3. 模型配置定制化YOLOv5s网络在yolov5/models目录下复制yolov5s.yaml重命名为face.yaml。只需修改两个参数nc: 1 # 只有人脸一个类别 names: [face] # 类别名称数据配置文件data/face.yaml示例train: ../CASIA-FaceV5/images/train val: ../CASIA-FaceV5/images/val nc: 1 names: [face]训练参数建议小显存6GB以下batch-size设为8-12中等显存8GBbatch-size 16-24大显存可以尝试324. 模型训练从预训练模型迁移学习启动训练的命令详解python train.py \ --data data/face.yaml \ --cfg models/face.yaml \ --weights yolov5s.pt \ --epochs 100 \ --batch-size 16 \ --img-size 640 \ --device 0 # 使用第一块GPU训练过程常见问题Loss不下降检查学习率默认0.01可能太大CUDA内存不足减小batch-size验证集mAP低检查标注质量实时监控训练进度tensorboard --logdir runs/train5. 模型验证与测试验证模型性能python val.py \ --data data/face.yaml \ --weights runs/train/exp/weights/best.pt \ --batch-size 8 \ --task test # 在测试集上验证测试单张图片python detect.py \ --weights runs/train/exp/weights/best.pt \ --source test.jpg \ --conf 0.5 # 置信度阈值6. 实时人脸检测部署摄像头实时检测命令python detect.py \ --weights best.pt \ --source 0 \ # 0表示默认摄像头 --view-img \ # 实时显示窗口 --conf-thres 0.6 # 调高阈值减少误检性能优化技巧添加--half参数使用半精度推理使用TensorRT加速需导出ONNX对于多摄像头可以用--source rtsp://url7. 常见问题解决方案检测框闪烁问题使用ByteTrack等跟踪算法在detect.py中添加帧间缓存小脸检测效果差修改model.yaml中的anchor尺寸数据增强中添加更多小脸样本侧脸漏检增加侧脸训练数据调整非极大抑制参数iou-thres最后分享一个实用技巧在detect.py中添加以下代码可以保存检测结果if save_img: cv2.imwrite(save_path, im0)

相关新闻

ESP-Drone完全指南:如何快速搭建基于ESP32的开源无人机项目

ESP-Drone完全指南:如何快速搭建基于ESP32的开源无人机项目

ESP-Drone完全指南:如何快速搭建基于ESP32的开源无人机项目 【免费下载链接】esp-drone Mini Drone/Quadcopter Firmware for ESP32 and ESP32-S Series SoCs. 项目地址: https://gitcode.com/GitHub_Trending/es/esp-drone ESP-Drone是一个基于乐鑫ESP32/ES…

2026/6/29 9:08:19阅读更多 →
2026年一键生成论文工具实测认证:5款神器从选题到排版全流程通关秘籍

2026年一键生成论文工具实测认证:5款神器从选题到排版全流程通关秘籍

写论文的焦虑,是每个科研人和学生都无法回避的“成长痛”。选题无从下手,文献检索耗时耗力,逻辑梳理混乱不堪,格式调整反复修改,查重降重更是让人抓耳挠腮。2026年的AI工具,早已不再是冷冰冰的“文字处理机…

2026/6/29 9:08:19阅读更多 →
ArcGIS实战:从零绘制专业中国地图(附完整数据与步骤)

ArcGIS实战:从零绘制专业中国地图(附完整数据与步骤)

1. 环境准备与数据获取 第一次打开ArcGIS时,那种密密麻麻的工具栏确实容易让人发懵。记得我读研时为了完成导师布置的地图作业,整整三天都没搞明白怎么正确设置坐标系。现在回头看,其实只要掌握几个关键步骤,制作专业级中国地图并…

2026/6/29 9:08:19阅读更多 →
APT攻击防御实战:从鱼叉钓鱼到纵深安全体系建设

APT攻击防御实战:从鱼叉钓鱼到纵深安全体系建设

1. 事件背景与APT攻击的行业特殊性最近在分析一些公开的威胁情报报告时,一个针对新闻机构的攻击案例引起了我的注意。这个案例涉及一个被称为“Callisto”的APT组织,其攻击目标直指“无国界记者”这样的国际新闻非政府组织。这并非一个孤立的IT安全事件&…

2026/6/29 13:04:29阅读更多 →
从零上手Scoop:Windows开发者的纯净软件管理指南

从零上手Scoop:Windows开发者的纯净软件管理指南

1. 为什么Windows开发者需要Scoop? 作为一个常年和Windows打交道的开发者,我最头疼的就是软件安装和管理。每次重装系统后,都要像集邮一样逐个下载Python、Node.js、Git这些工具,还得忍受各种安装向导的"下一步"轰炸。…

2026/6/29 13:04:29阅读更多 →
iOS 17 系统设置直达指南:从蓝牙到iCloud的私有路径与官方方案

iOS 17 系统设置直达指南:从蓝牙到iCloud的私有路径与官方方案

1. iOS 17 系统设置跳转的两种核心方法 每次iOS系统更新都会带来一些新的变化,iOS 17也不例外。作为开发者,我们经常需要在App中引导用户去系统设置页面进行某些配置,比如连接蓝牙、切换Wi-Fi网络或者调整键盘设置。这时候就面临一个选择&…

2026/6/29 13:04:29阅读更多 →
5分钟掌握SMUDebugTool:AMD Ryzen处理器底层调试终极指南

5分钟掌握SMUDebugTool:AMD Ryzen处理器底层调试终极指南

5分钟掌握SMUDebugTool:AMD Ryzen处理器底层调试终极指南 【免费下载链接】SMUDebugTool A dedicated tool to help write/read various parameters of Ryzen-based systems, such as manual overclock, SMU, PCI, CPUID, MSR and Power Table. 项目地址: https:/…

2026/6/29 13:04:29阅读更多 →
49.工业级复用 FB 功能块!PLC 电机正反转控制 + 过载保护 + 启停防抖全方案

49.工业级复用 FB 功能块!PLC 电机正反转控制 + 过载保护 + 启停防抖全方案

摘要 可编程逻辑控制器(PLC)是工业自动化领域的核心控制设备。本文从工程师视角出发,系统阐述PLC的硬件架构、扫描周期原理、IEC 61131-3编程标准,并重点以结构化文本(ST)语言为载体,提供一套完整的电机正反转控制与故障诊断程序。文章包含从硬件选型到软件调试的全流程…

2026/6/29 13:04:29阅读更多 →
喜利普厨房空调哪家靠谱

喜利普厨房空调哪家靠谱

在餐饮业中,厨房环境的舒适度直接影响到厨师的工作效率和菜品质量。然而,传统家用空调在高油烟、高温的商用厨房环境中往往表现不佳,导致频繁故障和高额维修费用。针对这一痛点,喜利普商用厨房新风空调以其卓越的产品性能和完善的…

2026/6/29 12:59:29阅读更多 →
AI Coding 六个月真实ROI账本:产品经理的血泪教训,研发的冷静忠告

AI Coding 六个月真实ROI账本:产品经理的血泪教训,研发的冷静忠告

6个月前的2025年12月,Boris Cherny 公开宣布自己卸载了 IDE。一时间,Vibe Coding 成了全行业最热的话题。6个月后,当我们回过头来拉一份真实账本,发现事情远没有"一句话生成一个App"那么浪漫。本文从产品经理和研发两个…

2026/6/29 3:27:55阅读更多 →
审计来了,数据权限全开——审计走了,怎么确保权限全部关掉?

审计来了,数据权限全开——审计走了,怎么确保权限全部关掉?

引言:审计结束三个月了,审计员的权限还没关某城商行每年按照监管要求开展至少一次数据安全审计。审计期间,内审部门需要抽样检查各类业务数据——交易流水、客户信息、员工操作日志、权限配置记录。这些数据分布在不同系统中,审计…

2026/6/29 2:19:08阅读更多 →
如何在3秒内从普通图片生成专业级法线贴图:DeepBump的终极指南

如何在3秒内从普通图片生成专业级法线贴图:DeepBump的终极指南

如何在3秒内从普通图片生成专业级法线贴图:DeepBump的终极指南 【免费下载链接】DeepBump Normal & height maps generation from single pictures 项目地址: https://gitcode.com/gh_mirrors/de/DeepBump 还在为3D建模中的纹理制作而烦恼吗?…

2026/6/29 0:01:47阅读更多 →
OCAuxiliaryTools:终极OpenCore配置工具,让黑苹果安装从未如此简单!

OCAuxiliaryTools:终极OpenCore配置工具,让黑苹果安装从未如此简单!

OCAuxiliaryTools:终极OpenCore配置工具,让黑苹果安装从未如此简单! 【免费下载链接】OCAuxiliaryTools Cross-platform GUI management tools for OpenCore(OCAT) 项目地址: https://gitcode.com/gh_mirrors/oc/OCA…

2026/6/29 0:01:47阅读更多 →
终极Windows 11精简指南:使用tiny11builder快速创建纯净系统镜像

终极Windows 11精简指南:使用tiny11builder快速创建纯净系统镜像

终极Windows 11精简指南:使用tiny11builder快速创建纯净系统镜像 【免费下载链接】tiny11builder Scripts to build a trimmed-down Windows 11 image. 项目地址: https://gitcode.com/GitHub_Trending/ti/tiny11builder 你是否厌倦了Windows 11系统自带的20…

2026/6/29 0:01:47阅读更多 →