项目简介
本项目是基于Python和OpenCV开发的口罩检测与图像标注系统,借助华为Atlas - 200DK边缘计算开发板开展口罩佩戴检测任务。项目涵盖图像检测、摄像头图像捕获和图像标注三个核心功能模块,可应用于实时监控和数据标注场景。
项目的主要特性和功能
- 口罩检测
- 采用YOLOV3 - Resnet18模型检测图像中的口罩佩戴情况。
- 包含图像预处理、模型推理、非极大值抑制(NMS)和后处理等步骤。
- 能计算精度、召回率、准确率和F1分数,并在图像上绘制检测结果。
- 摄像头图像捕获
- 实时捕获摄像头图像并保存到指定目录。
- 可控制捕获图像的数量和间隔时间,避免资源浪费。
- 图像标注
- 读取图像文件并支持鼠标标注。
- 将标注的坐标保存至CSV文件,便于后续数据分析和处理。
安装使用步骤
环境准备
- 确保已安装Python 3.x。
- 安装所需的Python库:
pip install opencv-python numpy
。
运行项目
假设用户已经下载了本项目的源码文件。
1. 运行口罩检测
- 进入项目目录,运行detect_mask.py
文件:
bash
python CV_code/detect_mask.py
- 该脚本会读取指定目录中的图像,进行口罩检测,并输出检测结果和评估指标。
2. 运行摄像头图像捕获
- 进入项目目录,运行picture_capture.py
文件:
bash
python CV_code/picture_capture.py
- 该脚本会实时捕获摄像头图像并保存到指定目录。
3. 运行图像标注
- 进入项目目录,运行tagging.py
文件:
bash
python CV_code/tagging.py
- 该脚本会读取指定目录中的图像,允许用户在图像上进行标注,并将标注结果保存至CSV文件。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】